From 57f7060794f6352496ee6c2b2acdbb83d541e8f8 Mon Sep 17 00:00:00 2001
From: Bryce Allen <bdallen@uchicago.edu>
Date: Wed, 10 Mar 2021 18:15:21 -0500
Subject: [PATCH] 4u: add shop weapons to planner

---
 db/mh4u.db              | Bin 3406848 -> 3407872 bytes
 db/mh4u/weapon_shop.csv |  38 ++++++++++++++++++++++++++++++++++++++
 db/set_weapon_buy.py    |  39 +++++++++++++++++++++++++++++++++++++++
 mhapi/model.py          |   9 +++++++++
 4 files changed, 86 insertions(+)
 create mode 100644 db/mh4u/weapon_shop.csv
 create mode 100755 db/set_weapon_buy.py

diff --git a/db/mh4u.db b/db/mh4u.db
index 5022c7b2ed02c2c54524fb8926875390ce90bf82..dd8053a8a39d56d92853b9d96cbc43e16f872653 100644
GIT binary patch
delta 2170
zcmXw4dr(y86+hp1c}7_7WkL4xypO$Emvvb$>lV;hLWvUzpc0xGjRqgL4r44R0)oop
z60aIs5~!M+rc58HEoua+34A(}sWX|h(+4&E!$(QdscD-@oe|AUXPivOK+ngTojEf*
zclUSB`JMCoo!kG_N2I_1uVgqcu@N90MwCkYQc^o&TaHB>i#nFzSP71m=vYaPmF!q4
zj+N?Ik2%&7$4YaobjR{IR)%AVjwLx(rek>>E6cI69c!s$<v5nlv1G@}b*wzc%6F^+
z$0~HJqA5#xJI61Bu!WfP2ArW!f=ng=z)l`Zmmm-hhC^#)oup9r1E136%^I*%$CgN7
zgw3#7ChMvA!);kPl$rYA&{(Pjs)3M4&+q3dFVn<N-6EfI(p@bpr`;EGm7{D+o^l~y
z>;z#K{gSrR6zC+U$dZ#%|8~ii#=av#(5xskL-C@=DwF}5HQ;uYD_^seRm%Kx8Hn6R
zZ_qw6OEyC%xK(E|c8Gh=HcFsl@sKR5ys3x7$~IoSYglP#n@5y!8Y~2%4+;gK-Q*To
z4R0gM#g7eVpGT^ZkQpnIRZi98D*I{i@0of^`3GgTt2p_Y+dw!he4Ea~Hrhn~Mw($3
zTV5-6pA!*l7y;9a7jB(a_RzF{k2SrgP&PZQgx97IfzSd&^g4Z-Oc0k`6${)Iqxwr?
z#`zRvYeb^aGC9nfzBjM*W7F*Qhsp-ZS{D@8=RxoZI!9k1mxvBSShHKQGtZYx5DAAv
zku`DP&wc7ESd{6q`manr+ZOsi+5??bAbl{$XDha!-^ABr6a~#V##f?#kBZH#Yq>g1
zSWk)iYE61O2v5;7v=G|K4++9{NOsCQ-<2R_Moq(vulrZsYBz7Pvr8=^tmu&X1<7mx
zXb=w2yYwV|f?Oakz%2w<NH%>3iw5*iAXpvW=TM(|fER5%q6UTOel^J_RsvKC9rQE0
ziOi67a21h3vF8GgCZrmXZ^ZPYpQtUIVbz;zmGA}v$C5_XI?oXhzE5XpHaP)DU>>2(
zV&8ZMCmJzPpE94S?~b~UOQH8a>L5FDNA0DVH2^nJ3^jC~o}-oI8mU33DcPyxUhFrh
zM`AUy%$0EJo_Z)VYhd5FBtd~53Wm&h&Y#^^chJ-atp2Y0E!4|>)t>%RP0IHi1fiM!
zmmVZvlWtgWw`V2ioK*VjBs*bz1MWH;js>tQo_R1$>)?}ENYj!xWK@FSz)gCO?j|2Y
zC2^(qvx_C$JMkidqGrIXmJJU2-B#@=_h7~ZpKCZ8jIEIk&M~n`<H~WXw`y(dnQhv6
z!ki|}&Jt@u*oi*tfLii*+{-7(GbY(-lLE&_g3(9#uR63I-c5bG_7-6so!Us1Xo2uD
zy+fZR*Wq)tnuXByV((-Nx?eX@5ONN$+;v%NMfdNY+>29JjZmm8{=Dsqwv%`C#bxa<
zyM9G0rKl326Llif%QQe<C;2djDJ#Ur$$b*!73Ad=gybB~vh=F<6q=qjzOQ{mS?5)4
zI4!ddgkK01^g~q1XJj|rhC0Nqlx*=g**GFxcC0#Hw`oSRsa(o#&uEX=ddfiv!3{b>
z7RVkbM~Gjt<%@+9L}SrdtV}N8_3C~4pjR%nrHi?XQ;9{Z<pR#_T&_OArGCq&e?Zvs
zJpJSR^fMsz(z|#m>&X<F*(ol^1#iY(U3?Z58$mNa0ynSHkMfcJx?I0R*xV{zUh63a
zp^i?{rKBB3p?IWfsn~jHJt|ApgGO2Wemv2yZ>PyWWuyK29>KE=gyVt$%V-PvC)wbN
zX<1p4oqVYQcMvcmc$ydQ)eny8U3_50NA(8E5{~Os&1ePTHNi`V$UNBvt&3-Ji+JE&
zuH(EwfFn7>g|oVaO}hKf>b-2&IsKO{;-Z&(@l1A+2V^Jw8Hcu7Jop~Asp>`;uSb9b
z-^}wjb3Zov{4rrV-=E-5X-D@0q`u@T8`G~O{z>?dtaJbHq`yBQwTAVv^;Q0En!b)5
z#=ktxI`~VDU9a|+F7up5oskkaO+Tm4liy-FTe7pKp2Jm`vB)<9-_+ti<`tXmv?)AC
zXznoH%thtjYw`DRC;oV^e}b^Rt^T*RicKKwrVn8bnrItcLfSBAjbuMI-HACv!60|j
zBKg1g4|}tHHkt0goIxXi=XCLh!E63DnzE8vGyXxs9$xb&)f+A(K@JFQ^d{t>C;muQ
zfD1jW3)L6EwU^0Wp0c^nXh}%?6u=c%=Da$x(SuajVU$}m6pPfz-bEAV8^^dXY72~X
z!fqBCB_u6@OaYk^cF<4gMpsmf#smN|_bxh)xN^QRdb~k1FdoMUU@XH3ViaS9FqUJ4
pF-kBZ7%MPLj8fL1MJH#)bX-gV_2Qco_xc+3Vq$@^U5YVB{tp8LMWO%z

delta 1983
zcmXw33s6+o89on%6_({(0olv)xaZ!zuFJA+uREI+Q)|d3ZJ9(s%u5R~%1MSAK_q}E
z&oy45woRZqIks_<N!k&kB1(W`I?lxDw4F4iwbNoU#L2`=$45FCooPDFbQF6ot2^`U
z-Pt?e`Ty_xzyI|2{|)wkdl3%#7S{sczBjC`lRv`O(v!?)&}PVH*k+4tw%BHAHcPje
z%VtY#mSM9@o4IY4WiyY>vTa6err2z$&2ns(YqLC?Ewfp^&Ac{KZB}42pUnzwR%Ekc
zo0Uv5?cIE}1mMl+3$z(tgloWAPy!x&_-R!;L>7-^DIgFIhC^#r1G=(!saNadj}&Nc
zko1vE1(;zgY{gY$iF`Hb)%NmB1)81NIkH3n)Req9|FS^K;nk{EFE;zMQ-z*RfE&<l
z)P`KJ9YRhn<T0LEuH99V*&~lCAZV4xtx&4gz-n!uaIMkipLSR+=o0FKvv3obOR=sF
zX>DZt@r?>FWT}v<Qm2apL$cJSVQm!oD8PMaUXs3r&w$?o`aqg|igyodUm(vjQd}ft
z#Y$A_h>IN7c8NZ&4HS|s0C(Z<p;@#AUWLztuQ_q$R*uY0Xo&27k4Tzn1}rO8vS~uw
zE@r2+@Z%Y;1Kf;0M+rE^$wxFP^t;-6l$mwHC6$?xXf)2R%xn4l%pI*+G~d&Xc-@jQ
zh;E|i;5almxt-IaB*}@@3W$Wmp~zY_>}Yz+OJ5a+O8>dkF`^m$5A{g<UV|A6Mu?kb
zbO@5x6VE!Pn<YUj#X3+%Ulctn=}#)%9RNRueuIkPVTkw(jp@q{Y5i4GduRzHX(yjj
zK*)+(rj;6_-AVU|;=S|+BpU(lMz>HS%7*9Q3NXV-Zthy~X*UgGcM_bG^#VpH5KQpC
zKDvy*(oMHb_0u%3rvl)8=)b5zPOA!h%*mPDu=wx@t-z$~lyo9Q%}DTmXsVCU<~NSh
zMPkuWik@`$0^EeAQ63zGE>2kP4nBW_9wa@Z*^ah|C9|eFj{NKfZ4rZC(O$H4J;1MH
z4|)|LI0_Ts?_m8Qr9UPrZ_<a5l1v}XkzEFjNUTy-of&VQqkDPZEt(|FqlyBGj8HIS
zr4GF|N4HMhrfG$qJpebNf1};-F5Cm|adLk7i%JrWR?8`d!?A#D!Z~Wtt+$JNZaux4
zBmlP2H>e(7f|XzbBxDO={^NSRl_;lwBw0}_U?o)3Vb#>>`^9JV`bnrHvrao6hNHpQ
zTGe!PU9Z#og}p^j<`SuEn_QH3cpX-Qj~7(M#pO2rSCW@8h8<oc7`@M{Ytwr~dxt(;
zK^g&m8GVIn;SAgcu7gI|N-gOgYnIDuSaQy4zEh~}oW7H9`atg@JI8iNZ>bpy#Z#~T
zb57qTW<JnYAtwMGSVw<A4E_T?3_eN)VC1}h5s?jJEeh}z`FuqoHQy1o^}PO===e|{
zaxbj{_#Ir1evh7oU&5Eb6;LI+ct}Zl&g4lm<QQX#RIU1HJt?kC>&u^Te;42oIs)h6
z4)7@_-{hAn$^5g$3W&y{u~=L!a!OQsje}ylYWxn8?6U<67Q~_nwa8I)&1>uvA)j%%
z&~psnUNk3*pMx4WCQJW9JX~&!L!z9mmGO(nMZ8a%Ty5+Zb8C!($30~LKZ(ZBGS~)=
zg0kTXFKHUDk^!Mc(2S=ReNDgdf;if5Y{#S&-~o)$UU@rg0)GUh!xg#W{r$!UL>7;4
zkX8q*h`cfvZk6)=MyJ372Bw}}0KbNF(9fmH25^g$iM)D#^|;YXcD?Up(H96fBu7o(
zaf6A5_l#dRd*l!XaT*##0QQ1^IU`sn{_OR~FlnEV4N}7l%aso}tgT*u<5Z!4k>B--
z+&lo5JkPb7tZDw)BmS1*8keZ4@b5-hKM?Km)el9t{H_!;3ID1U8NYN=^EvuHyue9!
z;ZMbdCjTi&aweaaCR?${gM1#_;XfebH!1H7xsBmyJQdq@P5vJ7@=pJob#4amOX!X?
z=4;3}dCFfaTL0odC^PGt>X2uJf<b2yoMv05{avJQs$CuxGz0RkTljZ=+TSYf{MDbf
zJ{N#=paR^AK9Ses417W^mdeMe`F2K*=#s&RhGG$Zftp49xgv9?_*0RY1v9!tr;J!Y
z4w=^%n_gaCXx52Oip`bKoemlR-ij`x8b_sOZWKB5(VK9fMmHsvO9UiVNCYKHB|;Kq
W5@Csz5)p~-NLUi9M2#LjGy8ubR3jz;

diff --git a/db/mh4u/weapon_shop.csv b/db/mh4u/weapon_shop.csv
new file mode 100644
index 0000000..c848d36
--- /dev/null
+++ b/db/mh4u/weapon_shop.csv
@@ -0,0 +1,38 @@
+Iron Sword,1500
+Buster Sword,2550
+Giant Jawblade,5250
+Ravager Blade,5250
+Iron Katana,1500
+Canine Katana,3300
+Eager Cleaver,10950
+Hunter's Knife,1500
+Assassin's Dagger,3300
+Chief Kris,4200
+Matched Slicers,1500
+Chief's Scythes,3300
+Dual Hatchets,4200
+War Hammer,1500
+War Mace,2550
+Bone Bludgeon+,4200
+Iron Striker+,5250
+Metal Bagpipe,1500
+Hunter's Horn,4200
+Heavy Bagpipe+,5250
+Iron Lance,1500
+Knight Lance,2550
+Rampart,5250
+Spiked Javelin,6450
+Iron Gunlance,1500
+Jaggid Gunlance,4200
+Defender's Gunlance,6450
+Bone Axe,1950
+Wild Axe,4200
+Power Gasher,5250
+Elite Blad,1950
+Bone Staff,3300
+Cross Bowgun,2550
+Cross Bowgun+,4200
+Bone Shooter,2100
+Bone Shooter+,5400
+Hunter's Bow I,1500
+Hunter's Stoutbow I,4200
diff --git a/db/set_weapon_buy.py b/db/set_weapon_buy.py
new file mode 100755
index 0000000..84bb496
--- /dev/null
+++ b/db/set_weapon_buy.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python2
+
+import os.path
+import codecs
+import csv
+
+import _pathfix
+
+from mhapi.db import MHDB
+
+
+def set_buy(db, item_id, buy):
+    print "buy", item_id, buy
+    cur = db.cursor()
+    cur.execute("""UPDATE items SET
+                   buy=? WHERE _id=?""",
+                (buy, item_id))
+
+def set_buy_by_name(db, name, buy):
+    cur = db.cursor()
+    cur.execute("""UPDATE items SET
+                   buy=? WHERE name=?""",
+                (buy, name))
+    rowid = cur.lastrowid
+    print "buy", rowid, name, buy
+
+if __name__ == '__main__':
+    db = MHDB(game="4u")
+    delta_file_path = os.path.join(_pathfix.db_path, "mh4u", "weapon_shop.csv")
+
+    with open(delta_file_path) as f:
+        reader = csv.reader(f)
+        for row in reader:
+            name = row[0]
+            value = int(row[1])
+            set_buy_by_name(db, name, value)
+
+    db.commit()
+    db.close()
diff --git a/mhapi/model.py b/mhapi/model.py
index 451dc84..f132372 100644
--- a/mhapi/model.py
+++ b/mhapi/model.py
@@ -641,6 +641,11 @@ def get_costs(db, weapon):
         for item in weapon.create_components:
             create_cost["components"][item.name] = item.quantity
         costs = [create_cost] + costs
+    if weapon.buy:
+        buy_cost = dict(zenny=int(weapon.buy),
+                        path=[weapon],
+                        components={})
+        costs = [buy_cost] + costs
     return costs
 
 
@@ -678,6 +683,10 @@ class ItemStars(object):
         costs = get_costs(self.db, weapon)
         # find least 'expensive' path
         for c in costs:
+            # don't calculate stars from buy cost (buys aren't available
+            # until after item is craftable, sometimes much later)
+            if not c["components"]:
+                continue
             current_stars = self._get_component_stars(c)
             for k, v in current_stars.iteritems():
                 if v is None: