-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathamiberry_xml_builder.py
712 lines (569 loc) · 28.2 KB
/
amiberry_xml_builder.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
import argparse
import datetime
import hashlib
import lhafile
import math
import openretroid
import os
from pathlib import Path
import platform
from slave_lha.parse_lha.read_lha import LhaSlaveArchive
import sys
import tempfile
from utils import text_utils
from whdload import whdload_slave
import xml.etree.ElementTree as etree
# =======================================
# Functions
# =======================================
def sha1(fname):
hash_sha1 = hashlib.sha1()
with open(str(fname), "rb") as f:
for chunk in iter(lambda: f.read(4096), b""):
hash_sha1.update(chunk)
return hash_sha1.hexdigest()
# Get a value from a file
def value_list(in_file, game_name):
file_name = "settings/" + in_file
if os.path.isfile(file_name) is False:
return ""
with open(file_name) as f:
content = f.readlines()
content = [x.strip() for x in content]
f.close()
answer = ""
for this_line in content:
if not this_line == "":
this_word = this_line.split()
if this_word[0] == game_name:
answer = this_word[1]
break
return answer
# Ensure a game package is set within a file
def check_list(in_file, game_name):
temp_game = game_name
if text_utils.right(temp_game.lower(),4) == ".iso" or text_utils.right(temp_game.lower(),4) == ".cue":
temp_game = text_utils.left(temp_game,len(temp_game)-4)
if text_utils.right(temp_game.lower(),4) == ".adf" or text_utils.right(temp_game.lower(),4) == ".hdf":
temp_game = text_utils.left(temp_game,len(temp_game)-4)
file_name = "settings/" + in_file
if os.path.isfile(file_name) is False:
return False
with open(file_name) as f:
content = f.readlines()
content = [x.strip() for x in content]
f.close()
answer = False
for this_line in content:
if this_line == temp_game:
answer = True
break
return answer
# =======================================
# Variables definition
# =======================================
print()
print(
text_utils.FontColours.BOLD + text_utils.FontColours.OKBLUE + "HoraceAndTheSpider and osvaldolove" + text_utils.FontColours.ENDC + "'s " + text_utils.FontColours.BOLD +
"Amiberry XML Builder" + text_utils.FontColours.ENDC + text_utils.FontColours.OKGREEN + " (0.6)" + text_utils.FontColours.ENDC + " | " + "" +
text_utils.FontColours.FAIL + "www.ultimateamiga.co.uk" + text_utils.FontColours.ENDC)
print()
parser = argparse.ArgumentParser(description='Create Amiberry XML for WHDLoad Packs.')
parser.add_argument('--scandir', '-s', # command line argument
help='Directories to Scan',
default='/home/pi/RetroPie/roms/amiga/' # Default directory if none supplied
)
parser.add_argument('--refresh', '-n', # command line argument
action="store_true", # if argument present, store value as True otherwise False
help="Full XML refresh"
)
parser.add_argument('--forceinput', '-f', # command line argument
action="store_true", # if argument present, store value as True otherwise False
help="Force -S to be used on OSX, and ignore timecheck"
)
# Parse all command line arguments
args = parser.parse_args()
# Get the directories to scan (or default)
# yeah, i shouldnt do this, but i'm lazy, so i will.
if platform.system() == "Darwin" and args.forceinput != True:
input_directory = "/Volumes/Macintosh HD/Users/horaceandthespider/Google Drive/Geek/Shared Pi Games/amiga/_Standard/"
else:
input_directory = args.scandir
# set the name of the db file
whdbfile = 'whdload_db.xml'
# Setup Bool Constant for xml refresh
FULL_REFRESH = args.refresh
hash_algorithm = 'SHA1'
count = 1
XML_HEADER= '<?xml version="1.0" encoding="UTF-8"?>' + chr(10)
XML_HEADER = XML_HEADER + '<whdbooter timestamp="' + datetime.datetime.now().strftime("%Y-%m-%d at %H:%M:%S") + '">' + chr(10)
XML_OLD = ""
if FULL_REFRESH == False:
text_file = open(whdbfile, "r")
XML_OLD = text_file.read()
text_file.close()
a = XML_OLD.find("<whdbooter")
for b in range(a , a+100):
if XML_OLD[b]==">":
break
c = XML_OLD.find("</whdbooter")
XML_OLD = XML_OLD[b+2:c]
XML = ""
XML_FOOTER = "</whdbooter>" + chr(10)
# Reports
ERROR_MSG = 'Problem file log: ' + datetime.datetime.now().strftime("%Y-%m-%d at %H:%M:%S") + '' + chr(10)
COMPLETE_MSG = 'Scanned file log: ' + datetime.datetime.now().strftime("%Y-%m-%d at %H:%M:%S") + '' + chr(10)
# =======================================
# main section starts here...
# =======================================
for file2 in Path(input_directory + "/").glob('**/*.lha'):
archive_path = str(file2)
this_file = os.path.basename(archive_path)
# check for a skip, and that its value for skipping
if (FULL_REFRESH==False and XML_OLD.find('<game filename="' + this_file[0:len(this_file)-4].replace("&", "&") + '"')>-1):
print("Skipping: " + text_utils.FontColours.OKBLUE + text_utils.FontColours.BOLD + this_file + text_utils.FontColours.ENDC)
COMPLETE_MSG = COMPLETE_MSG + "Skipped: " + this_file + chr(10)
elif text_utils.left(this_file,2)=="._":
...
count = count - 1
else:
print()
print("Processing: " + text_utils.FontColours.FAIL + text_utils.FontColours.BOLD + this_file + text_utils.FontColours.ENDC)
try:
slave_archive = LhaSlaveArchive(archive_path, hash_algorithm)
file_details = openretroid.parse_file(archive_path)
slave_archive.read_lha()
ArchiveSHA = sha1(file2)
# Defaults
HW_CHIPSET = "ECS"
HW_PROCESSOR = "68020"
HW_SPEED = "7"
HW_CHIP = .5
HW_FAST = 0
HW_Z3 = 0
hardware = ""
SLAVE_XML=""
first_slave=""
UUID = ""
n=1
default_slave = ""
default_slave_found = False
UUID = file_details['uuid']
print("Openretro URL: " + text_utils.FontColours.UNDERLINE + text_utils.FontColours.OKBLUE + "http://www.openretro.org/game/{}".format(UUID) + text_utils.FontColours.ENDC)
print()
# default slave
def_msg = ""
default_slave = value_list("WHD_DefaultSlave.txt", text_utils.left(this_file,len(this_file) - 4))
if default_slave != "":
def_msg = " (Lookup from list using File Name)"
# From here, we need to process WHDLoad header information out of the slave files!
for slave in slave_archive.slaves:
slave.get_hash()
print(text_utils.FontColours.BOLD + ' Slave Found: ', end='')
print(text_utils.FontColours.OKBLUE + slave.name + text_utils.FontColours.ENDC)
if default_slave != "":
if slave.name.find(default_slave) >0:
default_slave_found = True
# extract the slave as a temp file
fp = tempfile.NamedTemporaryFile()
fp.write(slave.data)
fp.seek(0)
this_slave = whdload_slave.whdload_factory(fp.name)
fp.close()
# we could work something out here later... but maybe it doesnt even matter here
# we can use the 'sub path' of slave.name to get the old UAE Config Maker folder name
slave_path = os.path.dirname(slave.name)
sub_path = text_utils.left(slave.name,len(slave_path) - len(slave.name))
full_game_name = text_utils.make_full_name(sub_path)
if first_slave == "":
first_slave = slave.name.replace(slave_path +"/","")
# Extract H/W settings from the slaves
for slave_flag in this_slave.flags:
if slave_flag == "Req68020":
HW_PROCESSOR = "68020"
if slave_flag == "ReqAGA":
HW_CHIPSET = "AGA"
HW_SPEED = "14"
# where we have multiple slaves, we will set the requirements as the highest ones found
# e.g. the one needing most memory etc
# round up any wierd chipram values
temp_chip_ram = this_slave.base_mem_size/1048576
for i in range(0, 2):
low_ram = int(math.pow(2, i-1))
high_ram = int(math.pow(2, i ))
if temp_chip_ram > low_ram and temp_chip_ram < high_ram:
temp_chip_ram = high_ram
# update the value if the highest slave requirement
if temp_chip_ram > HW_CHIP:
whd_chip_ram = temp_chip_ram
# round up any wierd fastram values
temp_fast_ram = this_slave.exp_mem/1048576
for i in range(0, 5):
low_ram = int(math.pow(2, i-1))
high_ram = int(math.pow(2, i ))
if temp_fast_ram > low_ram and temp_fast_ram < high_ram:
temp_fast_ram = high_ram
# update the value if the highest slave requirement
whd_fast_ram = 0
if temp_fast_ram > HW_FAST:
whd_fast_ram = temp_fast_ram
# we use the name of the 'last' slave, if there is only one
last_slave = slave.name.replace(slave_path +"/","")
SLAVE_XML += chr(9)+ chr(9)+ '<slave number="' + str(n) + '">' + chr(10)
SLAVE_XML += chr(9)+ chr(9)+ chr(9) + '<filename>' + (slave.name.replace(slave_path +"/","")).replace("&", "&") + '</filename>' + chr(10)
SLAVE_XML += chr(9)+ chr(9)+ chr(9) + '<datapath>' + (this_slave.current_dir).replace("&", "&") + '</datapath>' + chr(10)
if (this_slave.config) is not None:
SLAVE_XML += chr(9)+ chr(9)+ chr(9) + '<custom>' + chr(10)
for configs in this_slave.config:
if configs is not None:
SLAVE_XML += chr(9)+ chr(9)+ chr(9) + ((configs.replace("<","")).replace(">","")).replace("&", "&") + chr(10)
SLAVE_XML += chr(9)+ chr(9)+ chr(9) + '</custom>' + chr(10)
SLAVE_XML += chr(9)+ chr(9)+ '</slave>' + chr(10)
n=n+1
# end of slave checking
print()
print("Game name: " + text_utils.FontColours.HEADER + full_game_name + text_utils.FontColours.ENDC)
print("Lookup Name: " + text_utils.FontColours.HEADER + sub_path + text_utils.FontColours.ENDC)
# return of the default slave!
if default_slave_found == False:
default_slave = ""
if len(slave_archive.slaves) == 1 and default_slave=="":
default_slave = last_slave
def_msg = " (Only slave in archive search)"
elif default_slave=="":
default_slave = first_slave
def_msg = " (First slave in archive search)"
print("Default Slave: " + text_utils.FontColours.HEADER + default_slave + text_utils.FontColours.WARNING + def_msg + text_utils.FontColours.ENDC)
# =======================================
# DISPLAY SETTINGS
# Amiberry 3.2+: HEIGHT can be any value yet let's stick to the following
# values to keep things tidy and easy to maintain.
# Possible values: 400, 432, 480, 512, 524, 540, 568
# default: AUTOHEIGHT
listheights = ['400', '432', '480', '512', '524', '540', '568']
HW_HEIGHT = ''
for possibleheight in listheights:
if check_list('Screen_Height_'+possibleheight+'.txt', sub_path) is True:
HW_HEIGHT = possibleheight
break
# Amiberry 3.2+: WIDTH can be any value yet let's stick to the following
# values to keep things tidy and easy to maintain.
# Possible values: 640, 704, 720
# default: 720
listwidths = ['640', '704']
HW_WIDTH = '720'
for possiblewidth in listwidths:
if check_list('Screen_Width_'+possiblewidth+'.txt', sub_path) is True:
HW_WIDTH = possiblewidth
break
# centering
# default: enabled
HW_H_CENTER = 'SMART'
if check_list('Screen_NoCenter_H.txt', sub_path) is True:
HW_H_CENTER = 'NONE'
HW_V_CENTER = 'SMART'
if check_list('Screen_NoCenter_V.txt', sub_path) is True:
HW_V_CENTER = 'NONE'
# offset
# default: 0 (for both horizontal and vertical)
offset_h = value_list("Screen_Offset_H.txt", sub_path)
offset_v = value_list("Screen_Offset_V.txt", sub_path)
min_offset_h = -60
max_offset_h = 60
if offset_h.lstrip('-').isnumeric():
HW_H_OFFSET = int(offset_h)
if min_offset_h <= HW_H_OFFSET <= max_offset_h:
pass
elif HW_H_OFFSET < min_offset_h:
HW_H_OFFSET = min_offset_h
elif HW_H_OFFSET > max_offset_h:
HW_H_OFFSET = max_offset_h
else:
HW_H_OFFSET = ''
min_offset_v = -20
max_offset_v = 20
if offset_v.lstrip('-').isnumeric():
HW_V_OFFSET = int(offset_v)
if min_offset_v <= HW_V_OFFSET <= max_offset_v:
pass
elif HW_V_OFFSET < min_offset_v:
HW_V_OFFSET = min_offset_v
elif HW_V_OFFSET > max_offset_v:
HW_V_OFFSET = max_offset_v
else:
HW_V_OFFSET = ''
# NTSC
HW_NTSC = ""
if check_list("Chipset_ForceNTSC.txt", sub_path) is True:
HW_NTSC = "TRUE"
elif this_file.find("NTSC") > -1:
HW_NTSC = "TRUE"
# =======================================
# CONTROL SETTINGS
# mouse / mouse 2 / CD32
use_mouse1 = check_list("Control_Port0_Mouse.txt", sub_path)
use_mouse2 = check_list("Control_Port1_Mouse.txt", sub_path)
use_cd32_pad = check_list("Control_CD32.txt", sub_path)
# =======================================
# MEMORY SETTINGS
# Let's limit possible Z3 values to 128Mb.
# Amiberry 5+: 8Mb of fast RAM/Z2 set as default
# Default: 2Mb Chip / 8Mb Z2 / 0Mb Z3
for i in range(0, 8):
z3_ram = int(math.pow(2, i))
if check_list("Memory_Z3Ram_" + str(z3_ram) + ".txt", sub_path) is True:
HW_24BIT = "FALSE"
break
else:
z3_ram = 0
HW_24BIT = ""
# =======================================
# CHIPSET SETTINGS
# sprite collisions
# Default: Playfield
# can't find a single case requiring value different than default.
# blitter
# Default: Wait for Blitter
HW_BLITS = ""
if check_list("Chipset_ImmediateBlitter.txt", sub_path) is True:
HW_BLITS = "IMMEDIATE"
# fast copper
# Default: False
HW_FASTCOPPER = ""
if check_list("Chipset_FastCopper.txt", sub_path) is True:
HW_FASTCOPPER = "TRUE"
# =======================================
# CPU SETTINGS
# clock speed (MHz)
# Default: 7 for non-AGA / 14 for AGA
HW_SPEED = ""
if check_list("CPU_ClockSpeed_14.txt", sub_path) is True:
HW_SPEED = "14"
if check_list("CPU_ClockSpeed_25.txt", sub_path) is True:
HW_SPEED = "25"
if check_list("CPU_ClockSpeed_Max.txt", sub_path) is True:
HW_SPEED = "MAX"
# cpu model
# Default: 68000 for non-AGA / 68020 for AGA
# 24 bit addressing
# Default: True / you can set Z3 separately
# compatible CPU
# Default: True
HW_CPUCOMP = ''
if check_list('CPU_NoCompatible.txt', sub_path) is True:
HW_CPUCOMP = 'FALSE'
# CPU cycle exact
# available only for 68000 CPU
# Default: False
HW_CPUEXACT = ""
if check_list("CPU_CycleExact.txt", sub_path) is True:
HW_CPUEXACT = "TRUE"
# JIT Cache
# Default: False
HW_JIT = ""
if check_list("CPU_ForceJIT.txt",sub_path) == True:
HW_JIT = "TRUE"
# CHIPSET
HW_CHIPSET = ""
if this_file.find("_AGA") > -1:
HW_CHIPSET = "AGA"
if this_file.find("_CD32") > -1:
HW_CHIPSET = "AGA"
use_cd32_pad = True
if check_list('Chipset_AGA.txt',sub_path) is True:
HW_CHIPSET = 'AGA'
# ================================
# building hardware section
if HW_BLITS != '':
hardware += chr(10) + ('BLITTER=') + HW_BLITS
if HW_CHIPSET != '':
hardware += chr(10) + ('CHIPSET=') + HW_CHIPSET
if HW_SPEED != '':
hardware += chr(10) + ('CLOCK=') + HW_SPEED
if HW_24BIT != '':
hardware += chr(10) + ('CPU_24BITADDRESSING=') + HW_24BIT
if HW_CPUCOMP != '':
hardware += chr(10) + ('CPU_COMPATIBLE=') + HW_CPUCOMP
if HW_CPUEXACT != '':
hardware += chr(10) + ('CPU_EXACT=') + HW_CPUEXACT
if HW_FASTCOPPER != '':
hardware += chr(10) + ('FAST_COPPER=') + HW_FASTCOPPER
if HW_JIT != '':
hardware += chr(10) + ('JIT=') + HW_JIT
if HW_NTSC != '':
hardware += chr(10) + ('NTSC=') + HW_NTSC
if use_mouse1 == True:
hardware += chr(10) + ('PRIMARY_CONTROL=MOUSE')
else:
hardware += chr(10) + ('PRIMARY_CONTROL=JOYSTICK')
if use_mouse1 == True:
hardware += chr(10) + ('PORT0=MOUSE')
elif use_cd32_pad == True:
hardware += chr(10) + ('PORT0=CD32')
else:
hardware += chr(10) + ('PORT0=JOY')
if use_mouse2 == True:
hardware += chr(10) + ('PORT1=MOUSE')
elif use_cd32_pad == True:
hardware += chr(10) + ('PORT1=CD32')
else:
hardware += chr(10) + ('PORT1=JOY')
# SCREEN OPTIONS
# Screen: size, auto-height/crop
# Disable AUTOHEIGHT and set HEIGHT only when there's HEIGHT
if HW_HEIGHT != '':
HW_AUTO_HEIGHT = 'FALSE'
hardware += chr(10) + ('SCREEN_AUTOHEIGHT=') + HW_AUTO_HEIGHT
hardware += chr(10) + ('SCREEN_HEIGHT=') + HW_HEIGHT
else:
HW_AUTO_HEIGHT = 'TRUE'
hardware += chr(10) + ('SCREEN_AUTOHEIGHT=') + HW_AUTO_HEIGHT
if HW_WIDTH != '720' or HW_AUTO_HEIGHT == 'FALSE':
hardware += chr(10) + ('SCREEN_WIDTH=') + HW_WIDTH
# H_CENTER only if there's no H_OFFSET
if HW_H_CENTER == 'SMART' and HW_H_OFFSET != '':
HW_H_CENTER = 'NONE'
hardware += chr(10) + ('SCREEN_CENTERH=') + HW_H_CENTER
# V_CENTER only if there's no V_OFFSET
if HW_V_CENTER == 'SMART' and HW_V_OFFSET != '':
HW_V_CENTER = 'NONE'
hardware += chr(10) + ('SCREEN_CENTERV=') + HW_V_CENTER
if HW_H_OFFSET != '':
hardware += chr(10) + ('SCREEN_OFFSETH=') + str(HW_H_OFFSET)
if HW_V_OFFSET != '':
hardware += chr(10) + ('SCREEN_OFFSETV=') + str(HW_V_OFFSET)
if z3_ram != 0:
hardware += chr(10) + ('Z3_RAM=') + str(z3_ram)
# custom controls
custom_file = 'customcontrols/' + sub_path
custom_text = ''
# remove any items which are not amiberry custom settings
if os.path.isfile(custom_file) == True:
with open(custom_file) as f:
customsettings_content = f.readlines()
f.close()
for this_line in customsettings_content:
if this_line.find('amiberry_custom') > -1 and '\n' in this_line:
custom_text += chr(9) + chr(9) + chr(9) + this_line
elif this_line.find('amiberry_custom') > -1 and not '\n' in this_line:
custom_text += chr(9) + chr(9) + chr(9) + this_line + chr(10)
# external libraries (eg. xpk, required for Dungeon Master)
extra_libs = 'False'
if check_list('WHD_Libraries.txt', sub_path) is True:
extra_libs = 'True'
# generate XML
XML += chr(10) + chr(9)+ '<game filename="' + text_utils.left(this_file,len(this_file) - 4).replace("&", "&") + '" sha1="' + ArchiveSHA + '">' + chr(10)
XML += chr(9) + chr(9) + '<name>' + full_game_name.replace("&", "&") + '</name>' + chr(10)
XML += chr(9) + chr(9) + '<subpath>' + sub_path.replace("&", "&") + '</subpath>' + chr(10)
XML += chr(9) + chr(9) + '<variant_uuid>' + UUID + '</variant_uuid>' + chr(10)
XML += chr(9) + chr(9) + '<slave_count>' + str(len(slave_archive.slaves)) + '</slave_count>' + chr(10)
XML += chr(9) + chr(9) + '<slave_default>' + default_slave.replace("&", "&") + '</slave_default>' + chr(10)
XML += chr(9) + chr(9) + '<slave_libraries>' + extra_libs + '</slave_libraries>' + chr(10)
XML += SLAVE_XML
XML += chr(9) + chr(9) + '<hardware>'
XML += hardware.replace(chr(10), chr(10) + chr(9) + chr(9) + chr(9))
XML += chr(10) + chr(9) + chr(9) + '</hardware>' + chr(10)
if len(custom_text) > 0:
XML += chr(9) + chr(9) + '<custom_controls>' + chr(10) + custom_text + chr(9) + chr(9) + '</custom_controls>' + chr(10)
XML += chr(9)+ '</game>'
COMPLETE_MSG += "Scanned: " + full_game_name + chr(10)
except FileNotFoundError:
print("Could not find LHA archive: {}".format(archive_path))
ERROR_MSG += "Could not find LHA archive: {}".format(this_file) + chr(10)
except lhafile.BadLhafile:
print("Could not read LHA archive: {}".format(archive_path))
ERROR_MSG += "Could not read LHA archive: {}".format(this_file) + chr(10)
except KeyboardInterrupt:
print()
print("User Abort")
break
except:
print("Something went wrong with LHA archive: {}".format(archive_path))
ERROR_MSG += "Could not read LHA archive: {}".format(this_file) + chr(10)
# limit to a certian number of archives (for testing)
if count >= 99999:
break
count = count + 1
# =======================================
# XML snippets
# for recent games
# =======================================
print()
print('Adding external XML snippets')
snippet_dir = 'snippets/'
snippet_text = ''
# add fake root to 'old' XML to get valid xml
xml_old_snippet = '<root>' + XML_OLD + '</root>'
snipoldroot = etree.fromstring(xml_old_snippet)
for snippet_file in os.listdir(snippet_dir):
xmlsnip = os.path.join(snippet_dir,snippet_file)
with open(xmlsnip, 'r') as snippets_content:
# add fake root to 'snippets' to get valid xml
xml_snippet = '<root>' + snippets_content.read() + '</root>'
sniproot = etree.fromstring(xml_snippet)
# check if snippet's element has no duplicate then update it
for snipgame in sniproot.findall('game'):
snippet_filename = snipgame.get('filename')
snippet_sha1 = snipgame.get('sha1')
# check for sha1 in new packages or snippets
if snippet_sha1 in XML or snippet_sha1 in snippet_text:
COMPLETE_MSG += 'Snippet skipped: ' + snippet_filename + chr(10)
continue
# check for sha1 in 'old' XML
# if found delete it - it will be re-added below
# this way we would always get the latest version
if snippet_sha1 in XML_OLD:
for old_elem in snipoldroot.findall('.//game[@sha1="{value}"]'.format(value=snippet_sha1)):
snipoldroot.remove(old_elem)
# add element from snippet
snipstr = etree.tostring(snipgame).decode().strip()
if snipstr.startswith('\t'):
snippet_text += snipstr + chr(10)
else:
snippet_text += chr(9) + snipstr + chr(10)
# add to report
COMPLETE_MSG += 'Snippet added: ' + snippet_filename + chr(10)
# return a string and delete the fake root
XML_OLD = etree.tostring(snipoldroot).decode()
XML_OLD = XML_OLD.replace('<root>', '').replace('</root>', '').replace('<root />', '')
if len(snippet_text) > 0:
XML += chr(10) + snippet_text + chr(9)
# =======================================
# XML is complete, let's put it all together
# =======================================
XML = XML_HEADER + XML_OLD + XML + XML_FOOTER
# =======================================
# write down XML file
# =======================================
print("Generating XML File")
text_file = open(whdbfile, "w+")
text_file.write(XML)
text_file.close()
# =======================================
# Line Squasher
# * line(s) matching specified characters will be deleted
# * ensure only ASCII characters are written
#
# eg.:
# offtext = ['FAST_RAM=8', '\t\t\n']
# =======================================
offtext = []
with open(whdbfile, 'r') as deloffset:
olines = deloffset.readlines()
with open(whdbfile, 'w') as nomoreoffset:
for line in olines:
# remove blank lines
if line.rstrip():
# ensure only ASCII characters
if not any(offset in line for offset in offtext) and all(ord(ch) < 128 for ch in line):
nomoreoffset.write(line)
# =======================================
# Reports
# =======================================
text_file = open("files_scanned.txt", "w+")
text_file.write(COMPLETE_MSG)
text_file.close()
##if ERROR_MSG != "":
text_file = open("files_failed.txt", "w+")
text_file.write(ERROR_MSG)
text_file.close()