wayfinderak/ASB_submission_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

Browse Source 
# Conflicts:
#	submissions_temp/Sandmann.zip
#	submissions_temp/coldino_hhta.zip
#	submissions_temp/coldino_sp.zip
#	submissions_temp/donlexa.zip
This commit is contained in:
Dustin Pianalto
2018-05-10 16:53:24 -08:00
parent 72b7c88297 e4db9af2be
commit 9ba68474e7
22 changed files with 798 additions and 274 deletions
Download Patch File Download Diff File Expand all files Collapse all files
exts/imports/process_files.py
+19 -4
View File
@@ -45,6 +45,13 @@ def rename_section(cfg, sec, sec_new):
return cfg
def get_server_guid(server_file):
server_file = server_file.encode()
# p = s.pack((int.from_bytes(hash.encode(), byteorder='little') >> 64) & max_int,
# int.from_bytes(hash.encode(), byteorder='little') & max_int)
def process_file(in_file, file_type) -> ConfigParser:
with open(f'{config_dir}{bot_config_file}') as f:
bot_config = json.load(f)
@@ -97,19 +104,27 @@ def process_files(z) -> (ConfigParser, ConfigParser, list):
try:
with open(f'{path}{filename}', encoding='utf-8') as file:
game_config = process_file(file, 'game.ini')
file.seek(0)
mods = check_for_mods(file)
file.seek(0)
server_file = file.read()
except UnicodeDecodeError as e:
print(e)
try:
with open(f'{path}{filename}', 'w+b') as file:
with open(f'{path}{filename}', 'rb') as file:
contents = file.read()
file.write(contents.decode('utf-16-le').encode('utf-8'))
with open(f'{path}{filename}', encoding='utf-8') as file:
with open(f'{path}utf8{filename}', 'wb') as f:
f.write(contents.decode('utf-16-le').replace('\uFEFF', '').encode('utf-8'))
with open(f'{path}utf8{filename}', encoding='utf-8') as file:
game_config = process_file(file, 'game.ini')
file.seek(0)
mods = check_for_mods(file)
file.seek(0)
server_file = file.read()
except UnicodeDecodeError as e:
print(e)
return 0, 0, 0
server_guid = get_server_guid(server_file)
elif 'DinoExport' in filename:
# Get the contents of all DinoExport_*.ini files loaded into a dict
print(filename)
@@ -154,7 +169,7 @@ def generate_dino_files(dino_data, directory):
def generate_files(storage_dir, ctx, filename, game_ini, dinos_data, mods):
if not os.path.isdir(f'{storage_dir}/{ctx.author.id}'):
os.mkdir(f'{storage_dir}/{ctx.author.id}')
directory = f'{storage_dir}/{ctx.author.id}/{filename}_' \
directory = f'{storage_dir}/{ctx.author.id}/{filename.replace(".zip", "")}_' \
f'{ctx.message.created_at.strftime("%Y%m%dT%H%M%S")}'
os.mkdir(directory)
generate_game_ini(game_ini, mods, directory)