bda/mhapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: separate calculations and print

Browse Source
This commit is contained in:
Bryce Allen
2015-03-22 12:03:55 -05:00
parent 8af6a3a577
commit a449087aba
4 changed files with 147 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bin/mhrewards.py
View File

@@ -34,4 +34,7 @@ if __name__ == '__main__':
item_row = rewards.find_item(db, item_name, err_out)
if item_row is None:
sys.exit(os.EX_DATAERR)
rewards.print_quests_and_rewards(db, item_row, out)
ir = rewards.ItemRewards(db, item_row)
ir.print_recommended_hunts(out)
ir.print_monsters(out)
ir.print_quests(out)

2
mhapi/db.py
View File

@@ -10,7 +10,7 @@ import sqlite3
class Quest(object):
def __init__(self, quest_row, quest_rewards):
self._row = quest_row
self._rewards = quest_rewards
self.rewards = quest_rewards
self._template = string.Template(
"$name ($hub $stars* $rank)"

142
mhapi/rewards.py
View File

@@ -97,8 +97,9 @@ class QuestItemExpectedValue(object):
@param quest_rewards: list of rows from quest_rewards table for a single
quest
"""
def __init__(self, item_id, quest_rewards):
def __init__(self, item_id, quest):
self.item_id = item_id
self.quest = quest
self.fixed_rewards = dict(A=0, B=0, Sub=0)
self.total_reward_p = dict(A=0, B=0, Sub=0)
@@ -108,7 +109,7 @@ class QuestItemExpectedValue(object):
self.slot_rewards = dict(A=[], B=[], Sub=[])
self.total_expected_values = [0, 0, 0]
self._set_rewards(quest_rewards)
self._set_rewards(quest.rewards)
def is_sub(self):
"""Item is available from sub quest"""
@@ -305,9 +306,12 @@ class RankAndSkills(object):
Helper to track the best strategy with a given set of skills and hunter
rank.
"""
def __init__(self, rank, cap_skill=stats.CAP_SKILL_NONE,
def __init__(self, rank="G",
luck_skill=stats.LUCK_SKILL_NONE,
cap_skill=stats.CAP_SKILL_NONE,
carving_skill=stats.CARVING_SKILL_NONE):
self.rank = rank
self.luck_skill = luck_skill
self.cap_skill = cap_skill
self.carving_skill = carving_skill
self.best = None
@@ -328,6 +332,12 @@ class RankAndSkills(object):
return True
return False
def add_quest_item(self, quest_item):
if self.rank == "LR" and quest_item.rank != "LR":
return False
if self.rank == "HR" and quest_item.monster_rank == "G":
return False
class HuntItemStrategy(object):
"""
@@ -345,8 +355,8 @@ class HuntItemStrategy(object):
cap_skill=cap_skill)
def print(self, out):
out.write("%s %s %s (%5.2f)\n"
% (self.hunt_item.monster_name,
out.write("%s %s %s (%5.2f)\n" %
(self.hunt_item.monster_name,
self.hunt_item.monster_rank,
self.strat, self.ev))
@@ -369,7 +379,6 @@ class HuntItemExpectedValue(object):
@param hunt_rewards: list of rows from hunt_rewards table for a single
monster and rank
"""
def __init__(self, item_id, monster_name, monster_rank, hunt_rewards):
self.item_id = item_id
self.monster_name = monster_name
@@ -416,27 +425,65 @@ class HuntItemExpectedValue(object):
self.matching_rewards.append(reward)
def print_monsters_and_rewards(db, item_row, out):
item_id = item_row["_id"]
monsters = db.get_item_monsters(item_id)
class ItemRewards(object):
def __init__(self, db, item_row):
self.db = db
self.item_row = item_row
self.item_id = item_row["_id"]
skill_sets = OrderedDict([
("No skills" , RankAndSkills("G")),
("Capture God" , RankAndSkills("G", cap_skill=stats.CAP_SKILL_GOD)),
("Carving God" , RankAndSkills("G",
carving_skill=stats.CARVING_SKILL_GOD)),
("Low Rank" , RankAndSkills("LR")),
("High Rank" , RankAndSkills("HR")),
self. skill_sets = OrderedDict([
("No skills", RankAndSkills("G")),
("Capture God", RankAndSkills("G", cap_skill=stats.CAP_SKILL_GOD)),
("Carving God",
RankAndSkills("G", carving_skill=stats.CARVING_SKILL_GOD)),
("Great Luck",
RankAndSkills("G", luck_skill=stats.LUCK_SKILL_GREAT)),
("Low Rank", RankAndSkills("LR")),
("High Rank", RankAndSkills("HR")),
])
self._hunt_items = OrderedDict()
self._quest_items = OrderedDict()
self._find_hunt_items()
self._find_quest_items()
def _find_hunt_items(self):
monsters = self.db.get_item_monsters(self.item_id)
for m in monsters:
mid = m["monster_id"]
rank = m["rank"]
monster = db.get_monster(mid)
reward_rows = db.get_monster_rewards(mid, rank)
hunt_item = HuntItemExpectedValue(item_id, monster["name"], rank,
reward_rows)
monster = self.db.get_monster(mid)
reward_rows = self.db.get_monster_rewards(mid, rank)
hunt_item = HuntItemExpectedValue(self.item_id, monster["name"],
rank, reward_rows)
key = (mid, rank)
self._hunt_items[key] = hunt_item
out.write("%s %s\n" % (monster["name"], rank))
for s in self.skill_sets.values():
s.add_hunt_item(hunt_item)
def get_hunt_item(self, monster_id, monster_rank):
key = (monster_id, monster_rank)
return self._hunt_items.get(key)
def _find_quest_items(self):
"""
Get a list of the quests for acquiring a given item and the probability
of getting the item, depending on cap or kill and luck skills.
"""
quests = self.db.get_item_quest_objects(self.item_id)
if not quests:
return
for q in quests:
quest_item = QuestItemExpectedValue(self.item_id, q)
self._quest_items[q.id] = quest_item
def print_monsters(self, out):
for hunt_item in self._hunt_items.itervalues():
out.write("%s %s\n"
% (hunt_item.monster_name, hunt_item.monster_rank))
hunt_item.print(out, indent=2)
kill_ev = [0, 0]
@@ -457,54 +504,44 @@ def print_monsters_and_rewards(db, item_row, out):
out.write(" %20s %5.2f / 100\n" % ("Shiny", shiny_ev))
out.write("\n")
for s in skill_sets.values():
s.add_hunt_item(hunt_item)
print("*** Poogie Recommends ***", file=out)
no_skill_best = skill_sets["No skills"].best
for name, skill_set in skill_sets.iteritems():
def print_recommended_hunts(self, out):
out.write("*** Poogie Recommends ***\n")
no_skill_best = self.skill_sets["No skills"].best
for name, skill_set in self.skill_sets.iteritems():
if skill_set.best is None:
# Don't print out a rank with no options
continue
if name != "No skills" and skill_set.best.is_same_strat(no_skill_best):
if (name != "No skills"
and skill_set.best.is_same_strat(no_skill_best)):
# Don't print out a skill set/rank that doesn't differ from
# no skills any rank
continue
out.write("[%-11s] " % name)
skill_set.best.print(out)
out.write("\n")
print(file=out)
def print_quests_and_rewards(db, item_row, out):
def print_quests(self, out):
"""
Get a list of the quests for acquiring a given item and the probability
of getting the item, depending on cap or kill and luck skills.
"""
item_id = item_row["_id"]
quests = db.get_item_quest_objects(item_id)
print_monsters_and_rewards(db, item_row, out)
if not quests:
return
for q in quests:
out.write(unicode(q) + "\n")
for quest_item in self._quest_items.itervalues():
out.write(unicode(quest_item.quest) + "\n")
out.write(" %20s" % "= Quest\n")
quest = QuestItemExpectedValue(item_id, q._rewards)
quest.check_totals(out)
quest.print(out, indent=2)
quest_item.check_totals(out)
quest_item.print(out, indent=2)
monsters = db.get_quest_monsters(q.id)
quest_monsters = self.db.get_quest_monsters(quest_item.quest.id)
quest_ev = quest.expected_value()
quest_ev = quest_item.expected_value()
cap_ev = [quest_ev, quest_ev]
kill_ev = [quest_ev, quest_ev]
shiny_ev = 0
for m in monsters:
for m in quest_monsters:
mid = m["monster_id"]
monster = db.get_monster(mid)
reward_rows = db.get_monster_rewards(mid, q.rank)
hunt_item = HuntItemExpectedValue(item_id, monster["name"],
q.rank, reward_rows)
hunt_item = self.get_hunt_item(mid, quest_item.quest.rank)
kill_ev[0] += hunt_item.expected_value(STRAT_KILL)
kill_ev[1] += hunt_item.expected_value(STRAT_KILL,
@@ -517,7 +554,9 @@ def print_quests_and_rewards(db, item_row, out):
if kill_ev[0] == 0 and cap_ev[0] == 0 and shiny_ev == 0:
continue
out.write(" %20s\n" % ("= " + monster["name"] + " " + q.rank))
out.write(" %20s\n"
% ("= " + hunt_item.monster_name
+ " " + hunt_item.monster_rank))
hunt_item.print(out, indent=2)
@@ -529,4 +568,3 @@ def print_quests_and_rewards(db, item_row, out):
if shiny_ev:
out.write(" %20s %5.2f / 100\n" % ("Shiny", shiny_ev))
out.write("\n")

6
mhapi/web/wsgi.py
View File

@@ -81,8 +81,10 @@ class App(object):
else:
item_row = rewards.find_item(self.db, item_name, resp.body_file)
if item_row is not None:
rewards.print_quests_and_rewards(self.db, item_row,
resp.body_file)
ir = rewards.ItemRewards(self.db, item_row)
ir.print_recommended_hunts(resp.body_file)
ir.print_monsters(resp.body_file)
ir.print_quests(resp.body_file)
return resp
def get_all_names(self, req, resp):