def worker(self, queue): ThreadedBase.worker(self, queue) res = { "error": None, } try: if not self.xml: save_manager = SaveManager(self.savedata.savefile, self.savedata.gluid) data, save_number = save_manager.load(self.savedata.legacy) if not self.savedata.legacy: data = decompress_data(data) res["save_manager"] = save_manager res["save_number"] = save_number else: with open(self.xml, 'rb') as f: data = f.read() except Exception as e: res["error"] = e if not res["error"]: xml_handle = XmlHandler(data) xml_handle.pre_load() res["xml_handle"] = xml_handle res["done"] = True queue.put(res)
def worker(self, queue): ThreadedBase.worker(self, queue) res = { "error": None, } try: if not self.xml: save_manager = SaveManager(self.savedata.savefile, self.savedata.gluid) data, save_number = save_manager.load(self.savedata.legacy) if not self.savedata.legacy: data = decompress_data(data) res["save_manager"] = save_manager res["save_number"] = save_number else: with open(self.xml, 'rb') as f: data = f.read() except Exception as e: res["error"] = e if not res["error"]: xml_handle = XmlHandler(data) xml_handle.pre_load() res["xml_handle"] = xml_handle res["done"] = True queue.put(res)
class PonyShell(Cmd, object): prompt = 'ponyshell> ' def __init__(self, savedata): Cmd.__init__(self) self._save_manager = SaveManager(savedata.savefile, savedata.gluid) data, self.save_number = self._save_manager.load(savedata.legacy) if not savedata.legacy: data = decompress_data(data) self._xml_handle = XmlHandler(data) self._xml_handle.pre_load() self.savedata = savedata self._show_functions = { 'player': show_player, 'currencies': show_currencies, 'currency': show_currency, 'ponies': show_ponies, 'pony': show_pony, 'zones': show_zones, 'zone': show_zone, 'quests': show_quests, } self._set_functions = { 'player': set_player, 'currency': set_currency, 'ponies': set_ponies, 'pony': set_pony, 'inventory': set_inventory, 'zones': set_zones, 'zone': set_zone, 'quests': set_quests, } def cmdloop(self, intro=""): try: Cmd.cmdloop(self, intro) self.postloop() except KeyboardInterrupt: print("^C") self.cmdloop() @docopt_cmd def do_show(self, args): """Show what you requested. Usage: show player [<player_data>] show currencies show currency <currency_id>... show ponies [-i|-o] show pony <pony_id>... show zones show zone <zone_id>... show quests Arguments: player_data Id of a player_data item. Can be retrieved with "show player". currency_id Id of a currency. Can be retrieved with "show currencies". pony_id Id of a pony. Can be retrieved with "show ponies". zone_id Id of a zone. Can be retrieved with "show zones". Options: -i Displays ponies in inventory. -o Displays not owned ponies. -h --help Show this help.""" for i in self._show_functions: if args[i]: self._show_functions[i](self._xml_handle, args) break @docopt_cmd def do_set(self, args): """Set what you requested. Usage: set player <player_data> <value> set currency <value> <currency_id>... set ponies (level|shards) (up|down) set ponies (level|shards) <value> set ponies reset_game_timer set pony (level|shards) (up|down) <pony_id>... set pony (level|shards|next_game) <value> <pony_id>... set pony reset_game_timer <pony_id>... set inventory add <not_owned_pony_id>... set zones clear [clearables|foes] set zones reset_shops_timer set zone clear [clearables|foes] <zone_id>... set zone reset_shops_timer <zone_id>... set quests complete Arguments: player_data Id of a player_data item. Can be retrieved with "show player". currency_id Id of a currency. Can be retrieved with "show currencies". pony_id Id of a pony. Can be retrieved with "show ponies". zone_id Id of a zone. Can be retrieved with "show zones". not_owned_pony_id Id of a not owned pony. Can be retrieved with "show ponies -o". level value An integer between 0 and 5. shards value An integer between 0 and 10. next_game One of Ball, Apple or Book. Options: -h --help Show this help.""" for i in self._set_functions: if args[i]: self._set_functions[i](self._xml_handle, args) break @docopt_cmd def do_dump_xml(self, args): """Dump the actual XML tree. Usage: dump_xml <file> Arguments: file Write to specified file. Options: -h --help Show this help.""" try: with open(args['<file>'], 'wb') as f: f.write(self._xml_handle.prettify()) except Exception as e: print("Was unable to write to file, reason: {}".format(str(e))) @docopt_cmd def do_import_xml(self, args): """Import an XML tree. Use with caution. Usage: import_xml <file> Arguments: file Path to a file containing an XML tree. Options: -h --help Show this help.""" try: with open(args['<file>'], 'rb') as f: xml_data = f.read() new_xml_handle = XmlHandler(xml_data) new_xml_handle.pre_load() except Exception as e: print("Was unable to load from file, reason: {}".format(str(e))) else: self._xml_handle = new_xml_handle @docopt_cmd def do_write_save(self, args): """Write the current state of the save into a file Usage: write_save <file> [<gluid>] write_save -l <file> Arguments: file Path to the new save file. gluid GLUID used to encrypt the new save file. Must be base64 encoded. Options: -l --legacy Write a legacy save file (1.8.x version). -h --help Show this help.""" if args['<gluid>'] is not None: try: args['<gluid>'] = binascii.a2b_base64( args['<gluid>'].encode('utf-8')) except binascii.Error: print("Invalid encryption key") return try: self._save_manager.save( compress_data(self._xml_handle.to_string()), args['<file>'], self.save_number, args['<gluid>'], (self.savedata.legacy or args['--legacy']) and args['<gluid>'] is None) except Exception as e: print("Was unable to write to file, reason: {}".format(str(e))) @docopt_cmd def do_check_version(self, args): """Check online for a newer release Usage: check_version Options: -h --help Show this help.""" print("Querying github API...") ver = check_version(True) if ver["error"]: print(ver["error"]) elif ver["up_to_date"]: print("You're running the latest version") else: print("A newer version is available!") print("You can download it using this link:") print(ver["download_url"]) def do_bye(self, args): print("Exit") return True do_exit = do_bye do_quit = do_bye do_EOF = do_bye def emptyline(self): pass def default(self, line): print("{}: command not found".format(line))
class PonyShell(Cmd, object): prompt = 'ponyshell> ' def __init__(self, savedata): Cmd.__init__(self) self._save_manager = SaveManager(savedata.savefile, savedata.gluid) data, self.save_number = self._save_manager.load(savedata.legacy) if not savedata.legacy: data = decompress_data(data) self._xml_handle = XmlHandler(data) self._xml_handle.pre_load() self.savedata = savedata self._show_functions = { 'currencies': show_currencies, 'currency': show_currency, 'ponies': show_ponies, 'pony': show_pony, 'zones': show_zones, 'zone': show_zone, } self._set_functions = { 'currency': set_currency, 'ponies': set_ponies, 'pony': set_pony, 'zones': set_zones, 'zone': set_zone, 'inventory': set_inventory, } def cmdloop(self, intro=""): try: Cmd.cmdloop(self, intro) self.postloop() except KeyboardInterrupt: print("^C") self.cmdloop() @docopt_cmd def do_show(self, args): """Show what you requested. Usage: show currencies show currency <currency_id>... show ponies [-i|-o] show pony <pony_id>... show zones show zone <zone_id>... Arguments: currency_id Id of a currency. Can be retrieved with "show currencies". pony_id Id of a pony. Can be retrieved with "show ponies". zone_id Id of a zone. Can be retrieved with "show zones". Options: -i Displays ponies in inventory. -o Displays not owned ponies. -h --help Show this help.""" for i in self._show_functions: if args[i]: self._show_functions[i](self._xml_handle, args) break @docopt_cmd def do_set(self, args): """Set what you requested. Usage: set currency <value> <currency_id>... set ponies (level|shards) (up|down) set ponies (level|shards) <value> set ponies reset_game_timer set pony (level|shards) (up|down) <pony_id>... set pony (level|shards|next_game) <value> <pony_id>... set pony reset_game_timer <pony_id>... set zones clear [clearables|foes] set zones reset_shops_timer set zone clear [clearables|foes] <zone_id>... set zone reset_shops_timer <zone_id>... set inventory add <not_owned_pony_id>... Arguments: currency_id Id of a currency. Can be retrieved with "show currencies". pony_id Id of a pony. Can be retrieved with "show ponies". zone_id Id of a zone. Can be retrieved with "show zones". not_owned_pony_id Id of a not owned pony. Can be retrieved with "show ponies -o". level value An integer between 0 and 5. shards value An integer between 0 and 10. next_game One of Ball, Apple or Book. Options: -h --help Show this help.""" for i in self._set_functions: if args[i]: self._set_functions[i](self._xml_handle, args) break @docopt_cmd def do_dump_xml(self, args): """Dump the actual XML tree. Usage: dump_xml <file> Arguments: file Write to specified file. Options: -h --help Show this help.""" try: with open(args['<file>'], 'wb') as f: f.write(self._xml_handle.prettify()) except Exception as e: print("Was unable to write to file, reason: {}".format(str(e))) @docopt_cmd def do_import_xml(self, args): """Import an XML tree. Use with caution. Usage: import_xml <file> Arguments: file Path to a file containing an XML tree. Options: -h --help Show this help.""" try: with open(args['<file>'], 'rb') as f: xml_data = f.read() new_xml_handle = XmlHandler(xml_data) new_xml_handle.pre_load() except Exception as e: print("Was unable to load from file, reason: {}".format(str(e))) else: self._xml_handle = new_xml_handle @docopt_cmd def do_write_save(self, args): """Write the current state of the save into a file Usage: write_save <file> [<gluid>] write_save -l <file> Arguments: file Path to the new save file. gluid GLUID used to encrypt the new save file. Must be base64 encoded. Options: -l --legacy Write a legacy save file (1.8.x version). -h --help Show this help.""" if args['<gluid>'] is not None: try: args['<gluid>'] = binascii.a2b_base64(args['<gluid>'].encode('utf-8')) except binascii.Error: print("Invalid encryption key") return try: self._save_manager.save(compress_data(self._xml_handle.to_string()), args['<file>'], self.save_number, args['<gluid>'], (self.savedata.legacy or args['--legacy']) and args['<gluid>'] is None) except Exception as e: print("Was unable to write to file, reason: {}".format(str(e))) @docopt_cmd def do_check_version(self, args): """Check online for a newer release Usage: check_version Options: -h --help Show this help.""" print("Querying github API...") ver = check_version(True) if ver["error"]: print(ver["error"]) elif ver["up_to_date"]: print("You're running the latest version") else: print("A newer version is available!") print("You can download it using this link:") print(ver["download_url"]) def do_bye(self, args): print("Exit") return True do_exit = do_bye do_quit = do_bye do_EOF = do_bye def emptyline(self): pass def default(self, line): print("{}: command not found".format(line))