def main(script, command): d = Disk() if command == 'upload': d.upload() elif command == 'download': d.download()