You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.5 KiB
78 lines
2.5 KiB
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import argparse
|
|
import json
|
|
|
|
import _pathfix
|
|
|
|
from mhapi.db import MHDB
|
|
from mhapi.model import ModelJSONEncoder
|
|
|
|
|
|
def parse_args(argv):
|
|
parser = argparse.ArgumentParser(description=
|
|
"Determine the different paths for making a weapons and the cost"
|
|
" and parts required for each path."
|
|
)
|
|
parser.add_argument("-j", "--json", action="store_true",
|
|
default=False,
|
|
help="output as json instead of plaintext")
|
|
parser.add_argument("weapon",
|
|
help="Full name of weapon")
|
|
return parser.parse_args(argv)
|
|
|
|
|
|
def get_costs(db, weapon):
|
|
"""
|
|
Get a list of alternative ways of making a weapon, as a list of dicts
|
|
containing item counts. The dicts also contain special keys _zenny
|
|
for the total zenny needed, and _path for a list of weapons that
|
|
make up the upgrade path.
|
|
"""
|
|
costs = []
|
|
if weapon.parent_id:
|
|
parent_weapon = db.get_weapon(weapon.parent_id, True)
|
|
costs = get_costs(db, parent_weapon)
|
|
for cost in costs:
|
|
for item in weapon.upgrade_components:
|
|
if item.type == "Weapon":
|
|
continue
|
|
if item.name not in cost["components"]:
|
|
cost["components"][item.name] = 0
|
|
cost["components"][item.name] += item.quantity
|
|
cost["zenny"] += weapon.upgrade_cost
|
|
cost["path"] += [weapon]
|
|
if weapon.creation_cost:
|
|
create_cost = dict(zenny=weapon.creation_cost,
|
|
path=[weapon],
|
|
components={})
|
|
for item in weapon.create_components:
|
|
create_cost["components"][item.name] = item.quantity
|
|
costs = [create_cost] + costs
|
|
return costs
|
|
|
|
|
|
if __name__ == '__main__':
|
|
args = parse_args(None)
|
|
|
|
db = MHDB()
|
|
|
|
weapon = db.get_weapon_by_name(args.weapon, True)
|
|
if not weapon:
|
|
raise ValueError("Weapon '%s' not found" % name)
|
|
costs = get_costs(db, weapon)
|
|
if args.json:
|
|
for cost in costs:
|
|
cost["path"] = [dict(name=w.name, id=w.id)
|
|
for w in cost["path"]]
|
|
json.dump(costs, sys.stdout, cls=ModelJSONEncoder, indent=2)
|
|
else:
|
|
for cost in costs:
|
|
components = cost["components"]
|
|
print "=", ", ".join([w.name for w in cost["path"]])
|
|
print " Zenny", cost["zenny"]
|
|
for item_name in sorted(components.iterkeys()):
|
|
print "%20s %2d" % (item_name, components[item_name])
|
|
print
|