Apple-2-Tools/atrcopy
1
0
Fork 0
mirror of https://github.com/robmcmullen/atrcopy.git synced 2024-11-29 11:51:14 +00:00
Code Issues Projects Releases Wiki Activity

Moved main script into run() function

Browse Source
This commit is contained in:
Rob McMullen 2016-02-09 10:32:02 -08:00
parent 7b3c009e63
commit 59d5f40e10
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
atrcopy.py
View File

@ -776,7 +776,7 @@ def process(image, dirent, options):
else:
print dirent
if __name__ == "__main__":
def run():
import sys
import argparse
@ -800,8 +800,8 @@ if __name__ == "__main__":
data = to_numpy(data)
try:
header = AtrHeader(data[0:16])
for format in [KBootImage, AtariDosDiskImage, BootDiskImage]:
print "trying", format.__name__
for format in [KBootImage, AtariDosDiskImage]:
if options.verbose: print "trying", format.__name__
try:
image = format(data, filename)
print "%s: %s" % (filename, image)
@ -818,12 +818,11 @@ if __name__ == "__main__":
raise
#pass
except:
raise
print "%s: Doesn't look like a supported disk image" % filename
if options.verbose: print "%s: Doesn't look like a supported disk image" % filename
try:
image = AtariDosFile(data, filename)
image = AtariDosFile(data)
except InvalidBinaryFile:
print "%s: Doesn't look like an XEX either" % filename
if options.verbose: print "%s: Doesn't look like an XEX either" % filename
continue
if image is None:
image = BootDiskImage(data, filename)
@ -839,3 +838,5 @@ if __name__ == "__main__":
except ByteNotInFile166:
print "Invalid sector for: %s" % str(dirent)
if __name__ == "__main__":
run()