passport.py/passport.py

46 lines
984 B
Python
Executable File

#!/usr/bin/env python3
from passport import eddimage, wozimage
from passport import DefaultLogger, DebugLogger
from passport import Crack, Verify, EDDToWoz
import os.path
import sys
def usage(error_code):
exe = sys.argv[0]
print(STRINGS["header"])
print("""usage: {exe} crack image.woz
{exe} verify image.woz
{exe} convert image.edd""".format(**locals()))
sys.exit(error_code)
args = len(sys.argv)
if args < 3:
usage(0)
cmd, inputfile = sys.argv[1:3]
if cmd == "crack":
processor = Crack
elif cmd == "verify":
processor = Verify
elif cmd == "convert":
processor = EDDToWoz
else:
print("unrecognized command")
usage(1)
base, ext = os.path.splitext(inputfile)
ext = ext.lower()
if ext == ".woz":
reader = wozimage.WozReader
elif ext == ".edd":
reader = eddimage.EDDReader
else:
print("unrecognized file type")
usage(1)
logger = DefaultLogger # TODO add flag to change this
processor(reader(inputfile), logger)