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

Try utf-16 if utf-8 fails

Browse Source
This commit is contained in:
Dustin Pianalto 2018-05-09 16:02:42 -08:00
parent a99abb7cc7
commit e3dc53ee7d
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
exts/imports/process_files.py
View File

@ -84,33 +84,35 @@ def process_files(z) -> (ConfigParser, ConfigParser, list):
dino_data = dict() dino_data = dict()
game_config = ConfigParser() game_config = ConfigParser()
mods = list() mods = list()
z.extractall(path='submissions_temp/tmp/') path = 'submissions_temp/tmp/'
for filename in os.listdir('submissions_temp/tmp/'): z.extractall(path=path)
for filename in os.listdir(path):
if filename.endswith('.ini'): if filename.endswith('.ini'):
# ignore any files that don't end with .ini # ignore any files that don't end with .ini
if filename.lower() == 'game.ini': if filename.lower() == 'game.ini':
# Clean the Game.ini file, removing unnecessary lines # Clean the Game.ini file, removing unnecessary lines
try: try:
with open(filename, encoding='utf-8') as file: with open(f'{path}{filename}', encoding='utf-8') as file:
game_config = process_file(file, 'game.ini', 'utf-8') game_config = process_file(file, 'game.ini', 'utf-8')
mods = check_for_mods(file)
except UnicodeDecodeError as e: except UnicodeDecodeError as e:
print(e) print(e)
try: try:
with open(filename, encoding='utf-16-le') as file: with open(f'{path}{filename}', encoding='utf-16-le') as file:
game_config = process_file(file, 'game.ini', 'utf-16-le') game_config = process_file(file, 'game.ini', 'utf-16-le')
mods = check_for_mods(file)
except UnicodeDecodeError as e: except UnicodeDecodeError as e:
print(e) print(e)
return 0, 0, 0 return 0, 0, 0
mods = check_for_mods(z.open(filename))
elif 'DinoExport' in filename: elif 'DinoExport' in filename:
# Get the contents of all DinoExport_*.ini files loaded into a dict # Get the contents of all DinoExport_*.ini files loaded into a dict
try: try:
with open(filename, encoding='utf-8') as file: with open(f'{path}{filename}', encoding='utf-8') as file:
dino_data[filename] = process_file(file, 'dino.ini', 'utf-8') dino_data[filename] = process_file(file, 'dino.ini', 'utf-8')
except UnicodeDecodeError as e: except UnicodeDecodeError as e:
print(e) print(e)
try: try:
with open(filename, encoding='utf-16-le') as file: with open(f'{path}{filename}', encoding='utf-16-le') as file:
dino_data[filename] = process_file(file, 'dino.ini', 'utf-16-le') dino_data[filename] = process_file(file, 'dino.ini', 'utf-16-le')
except UnicodeDecodeError as e: except UnicodeDecodeError as e:
print(e) print(e)