tools: tu-deck-grabber: support buyback command

dsuchka · Jul 28, 2024 · 8e3fde1 · 8e3fde1
1 parent e01bc0a
commit 8e3fde1
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/tools/tu-deck-grabber.py b/tools/tu-deck-grabber.py
@@ -571,6 +571,13 @@ def salvageCard(self, cid):
         })
         return self.__run_api_req_with_retries(rd.getUrlParamMessage(), rd)
 
+    def buybackCard(self, cid, count: int = 1):
+        rd = self.__mkRequestData('buybackCard', {
+            'card_id': str(cid),
+            'number': str(count),
+        })
+        return self.__run_api_req_with_retries(rd.getUrlParamMessage(), rd)
+
     def salvageL1CommonCards(self):
         rd = self.__mkRequestData('salvageL1CommonCards', {})
         return self.__run_api_req_with_retries(rd.getUrlParamMessage(), rd)
@@ -891,6 +898,7 @@ def _up(x_id):
             client.fuseCard(target_cid)
             return target_cid
     up_or_fuse(c_id, [c_id])
+    print('SP: {}'.format(rsp['user_data']['salvage']))
     return
 
 def cmd_salvage(client, args):
@@ -906,8 +914,17 @@ def cmd_salvage(client, args):
         count = 1 if (len(args) < 3) else int(args[2])
         for i in range(0, count):
             rsp = client.salvageCard(cid)
-    sp = rsp['user_data']['salvage']
-    print('SP: {}'.format(sp))
+    print('SP: {}'.format(rsp['user_data']['salvage']))
+    return
+
+def cmd_buyback(client, args):
+    if (len(args) < 2):
+        print('USAGE: buyback <CARD_ID> [COUNT]')
+        return
+    cid = int(args[1])
+    count = 1 if (len(args) < 3) else int(args[2])
+    rsp = client.buybackCard(cid, count)
+    print('SP: {}'.format(rsp['user_data']['salvage']))
     return
 
 def cmd_buy20(client, args) -> int:
@@ -946,6 +963,7 @@ def _collect_rarity_to_count(resp):
         for cid, cnt in cid2cnt.items():
             cname = id_to_cards[cid]['full_name']
             print(f'     * {cnt} x {cname}')
+    print('SP: {}'.format(rsp['user_data']['salvage']))
     return total_count
 
 def cmd_card(client, args):
@@ -1091,6 +1109,8 @@ def write_dump_safe(name, data, append = False):
                 cmd_fuse(client, args)
             elif (args[0] == 'salvage'):
                 cmd_salvage(client, args)
+            elif (args[0] == 'buyback'):
+                cmd_buyback(client, args)
             elif (args[0] == 'buy20'):
                 cmd_buy20(client, args)
             elif (args[0] == 'card'):