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))