Update message to send SKU by platform

planetarium · Aug 1, 2024 · 986018f · 986018f
1 parent 632f29e
commit 986018f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/iap/api/purchase.py b/iap/api/purchase.py
@@ -279,6 +279,7 @@ def request_product(receipt_data: ReceiptSchema,
                                 "agent_addr": receipt.agent_addr,
                                 "avatar_addr": receipt.avatar_addr,
                                 "planet_id": receipt_data.planetId.decode(),
+                                "package_name": receipt.package_name,
                             }))
     logger.info(f"Voucher message: {resp['MessageId']}")
 

diff --git a/worker/worker/handler.py b/worker/worker/handler.py
@@ -13,7 +13,7 @@
 from common import logger
 from common._crypto import Account
 from common._graphql import GQL
-from common.enums import TxStatus
+from common.enums import TxStatus, PackageName
 from common.models.product import Product
 from common.models.receipt import Receipt
 from common.utils.actions import create_unload_my_garages_action_plain_value
@@ -106,7 +106,12 @@ def process(sess: Session, message: SQSMessageRecord, nonce: int = None) -> Tupl
     planet_id: PlanetID = PlanetID(bytes(message.body["planet_id"], 'utf-8'))
     agent_address = message.body.get("agent_addr")
     avatar_address = message.body.get("avatar_addr")
-    memo = json.dumps({"iap": {"g_sku": product.google_sku, "a_sku": product.apple_sku}})
+    package_name = PackageName(message.body.get("package_name"))
+    memo = json.dumps({"iap":
+                           {"g_sku": product.google_sku,
+                            "a_sku": product.apple_sku_k if package_name == PackageName.NINE_CHRONICLES_K
+                            else product.apple_sku}
+                       })
     # Through bridge
     if planet_id != CURRENT_PLANET:
         agent_address = planet_dict[planet_id]["agent"]