def main():
argument = sys.argv[1]
if argument.endswith('.vm'):
asm_filename = argument.replace('.vm', '.asm')
asm_file = open(asm_filename, 'a')
code_writer = CodeWriter(asm_file)
open_vm_file(argument, code_writer)
code_writer.close()
else:
if argument.endswith('/'):
argument = argument[0:-1]
foldername = os.path.basename(argument)
asm_file = open('%s/%s.asm' % (argument, foldername), 'a')
code_writer = CodeWriter(asm_file)
code_writer.write_comment('write init')
code_writer.write_init()
files = glob.glob('%s/*.vm' % argument)
for file in files:
open_vm_file(file, code_writer)
code_writer.close()
