mcm-cli 1.2.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

{mcm_cli-1.2.0.dist-info → mcm_cli-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mcm-cli
-Version: 1.2.0
+Version: 1.3.1
 Summary: A command-line interface for Moloco Commerde Media
 Home-page: https://github.com/moloco-mcm/mcm-cli
 Author: Moloco MCM Team

{mcm_cli-1.2.0.dist-info → mcm_cli-1.3.1.dist-info}/RECORD

@@ -1,13 +1,13 @@
 mcmcli/__init__.py,sha256=-U6lMZ9_99IXAKwnqnYXYr6NcO6TSmG-kxewgvJjU4k,575
-mcmcli/__main__.py,sha256=G9B6mZCD2JJpk8nHDdexuKfNigm8WZeFJXhutUEmIUE,1616
+mcmcli/__main__.py,sha256=Rnx3sLJc0OObJic70Z0I6h0-pjTaKK-qYqj42nzcskw,1616
 mcmcli/logging.py,sha256=xjRS5ey1ONx_d34qB1Fetb_SwPysoh2hzNDuNAaYYWQ,1739
-mcmcli/requests.py,sha256=ZoQULPpKdAvGtki25Jr6K2Xq2v213XLinzPHUeBi9wo,2601
+mcmcli/requests.py,sha256=IuySBQ8P_GoGF3f_TRysfgQNOhi2n9M84WK_eRXnoEU,2945
 mcmcli/command/account.py,sha256=kxJbBKYrw6OyCdaNZ0K0BEvKgZEAjj5azO-lNSeYLTM,25107
 mcmcli/command/admin.py,sha256=nJ7rm0nm0jHPobg0PjNHIWaWURTQu6QEUEUKIo__GO0,3010
 mcmcli/command/auth.py,sha256=QLdr_XFW5BVw9r4a7Kjj5BTBXpSux3AWI9eI03S8aiA,2480
 mcmcli/command/config.py,sha256=sdzge-l_Yi2P_TlTgSLqShcGyPCzpW3QJzctpIvc-g4,4195
 mcmcli/command/decision.py,sha256=Zjbmt71OVU-oL8Itt9O-SvwT9Lbxw-PAgRZaIgiXi-E,8411
-mcmcli/command/wallet.py,sha256=0MprkN5hJqJCTbMpMnFLRcLKnnc5PNYpx5Sjhys6XTY,9242
+mcmcli/command/wallet.py,sha256=giPDktndwh-NSIZbCvBzCQ6_MttZsqOkCW1wJ-fFJ8Y,11708
 mcmcli/data/account.py,sha256=pe7tPapP6vlUD5D5L5Nh5k2bkWdYOK01Mpt5fBYFnJs,1782
 mcmcli/data/account_user.py,sha256=27nQp52nMma5a3QszSJGqgq5Z0ivIb-nMZcZuhEgbEg,1328
 mcmcli/data/decision.py,sha256=bQ5j_PbPRSFa0sY5g9UVqdNzl_2epchcz1lHoDVuV90,2880
@@ -15,11 +15,11 @@ mcmcli/data/error.py,sha256=d6xa_jTXumlA0EzXy2PJQ86ajBb0Pm90fss9R3LuHUc,1094
 mcmcli/data/item.py,sha256=Z2xTRhU8T4vyJADO0l6-XPyQXvb9DX_OAjExhSXpW2A,1091
 mcmcli/data/seller.py,sha256=40SA7QekM3a3svDrDYLo_QYJ68VUxDO0KeGejJMp4k4,1004
 mcmcli/data/token.py,sha256=11wtyLHCAZHu0LVbNDPa-yipcL6lenxoYIKEI58VzFs,1744
-mcmcli/data/wallet.py,sha256=W-CksF9SPqiv3jZg07Wy8ehVUP5Ot1Gbq2LEGNQCOC8,1906
-mcm_cli-1.2.0.dist-info/LICENSE,sha256=RFhQPdSOiMTguUX7JSoIuTxA7HVzCbj_p8WU36HjUQQ,10947
-mcm_cli-1.2.0.dist-info/METADATA,sha256=7ewr0UPyJ6IqNvXOSL7Gci2BryiET3dEqfnuM1BYOxc,3018
-mcm_cli-1.2.0.dist-info/NOTICE,sha256=Ldnl2MjRaXPxcldUdbI2NTybq60XAa2LowRhFrRTuiI,76
-mcm_cli-1.2.0.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-mcm_cli-1.2.0.dist-info/entry_points.txt,sha256=qTHAWZ-ejSiU4t11RYwtAU8ScqhQPDeMVTG9y4wMVLg,60
-mcm_cli-1.2.0.dist-info/top_level.txt,sha256=sh7oqIaqLQlMtKHlxHHgpV2xGMrBMPFWpSp0C6nvJ_Y,7
-mcm_cli-1.2.0.dist-info/RECORD,,
+mcmcli/data/wallet.py,sha256=eMUi8N0vJdg_E10TPhSPoZkZtmIG7gHyqgabQ8C5Lg8,3217
+mcm_cli-1.3.1.dist-info/LICENSE,sha256=RFhQPdSOiMTguUX7JSoIuTxA7HVzCbj_p8WU36HjUQQ,10947
+mcm_cli-1.3.1.dist-info/METADATA,sha256=GMtyMAiGd5DxGm7v7Id5AcSgtW5tcq5paDD1XrXlaag,3018
+mcm_cli-1.3.1.dist-info/NOTICE,sha256=Ldnl2MjRaXPxcldUdbI2NTybq60XAa2LowRhFrRTuiI,76
+mcm_cli-1.3.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mcm_cli-1.3.1.dist-info/entry_points.txt,sha256=qTHAWZ-ejSiU4t11RYwtAU8ScqhQPDeMVTG9y4wMVLg,60
+mcm_cli-1.3.1.dist-info/top_level.txt,sha256=sh7oqIaqLQlMtKHlxHHgpV2xGMrBMPFWpSp0C6nvJ_Y,7
+mcm_cli-1.3.1.dist-info/RECORD,,

{mcm_cli-1.2.0.dist-info → mcm_cli-1.3.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

mcmcli/__main__.py

@@ -35,7 +35,7 @@ def version():
 	"""
 	Show the tool version
 	"""
-	typer.echo(f"Version: mcm-cli v1.2.0")
+	typer.echo(f"Version: mcm-cli v1.3.1")
 
 app.add_typer(mcmcli.command.account.app, name="account", help="Ad account management")
 app.add_typer(mcmcli.command.admin.app, name="admin", help="Platform administration")

mcmcli/command/wallet.py

@@ -12,9 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from datetime import datetime
 from enum import Enum
 from mcmcli.data.error import Error
-from mcmcli.data.wallet import Wallet, WalletsWrapper
+from mcmcli.data.wallet import Wallet, WalletsWrapper, PlatformWalletsWrapper
 from mcmcli.requests import CurlString, api_request
 
 import json
@@ -36,20 +37,68 @@ class OperationType(Enum):
     DEPOSIT = "deposit"
     WITHDRAW = "withdraw"
 
+def _get_wallet_balance(wallet: Wallet):
+    wa0 = wallet.accounts[0]
+    wa1 = wallet.accounts[1]
+
+    credits_amount_micro  = wa0.balance.amount_micro if wa0.type == 'CREDITS' else wa1.balance.amount_micro
+    pre_paid_amount_micro = wa0.balance.amount_micro if wa0.type == 'PRE_PAID' else wa1.balance.amount_micro
+    credit_balance = float(credits_amount_micro) / float(1000000)
+    prepaid_balance = float(pre_paid_amount_micro) / float(1000000)
+
+    return (credit_balance, prepaid_balance)
+
+
+@app.command()
+def platform_balance(
+    to_curl: bool = typer.Option(False, help="Generate the curl command instead of executing it."),
+    profile: str = typer.Option("default", help="profile name of the MCM CLI."),
+):
+    """
+    Get the current balances of all ad accounts in CSV format.
+    """
+    auth = mcmcli.command.auth.AuthCommand(profile)
+    curl, error, token = auth.get_token()
+    if error:
+        print(f"ERROR: {error.message}", file=sys.stderr, flush=True)
+        return
+
+    # Get current UTC timestamp and format it as an ISO 8601 string with microseconds
+    current_timestamp = datetime.utcnow().isoformat(timespec='microseconds') + "Z"
+
+    wc = WalletCommand(profile, auth, token.token)
+    curl, error, platform_wallets = wc.get_platform_balance(to_curl)
+    if to_curl:
+        print(curl)
+        return
+    if error:
+        print(f"ERROR: {error.message}", file=sys.stderr, flush=True)
+        return
+    if platform_wallets is None:
+        print(f"ERROR: Cannot find the wallets", file=sys.stderr, flush=True)
+        return
+
+    print("ad_account_id,credit_balance,prepaid_balance,got_balance_info_at_utc")
+    for account_id, account_data in platform_wallets.items():
+        wallet = account_data.wallets[0]
+        credit_balance, prepaid_balance = _get_wallet_balance(wallet)
+        print(f'{account_id},{credit_balance},{prepaid_balance},{current_timestamp}')
+    
+
 @app.command()
 def balance(
     account_id: str = typer.Option(help="Ad account ID"), 
     to_curl: bool = typer.Option(False, help="Generate the curl command instead of executing it."),
     to_json: bool = typer.Option(False, help="Print raw output in json"),
-    profile: str = "default",
-    ):
+    profile: str = typer.Option("default", help="profile name of the MCM CLI."),
+):
     """
     Retrive the current balance of the given ad account's wallet.
     """
     auth = mcmcli.command.auth.AuthCommand(profile)
     curl, error, token = auth.get_token()
     if error:
-        print(f"ERROR: {error.message}")
+        print(f"ERROR: {error.message}", file=sys.stderr, flush=True)
         return
 
     wc = WalletCommand(profile, auth, token.token)
@@ -58,24 +107,20 @@ def balance(
         print(curl)
         return
     if error:
-        print(f"ERROR: {error.message}")
+        print(f"ERROR: {error.message}", file=sys.stderr, flush=True)
+        return
+    if wallet is None:
+        print(f"ERROR: Wallet does not exist.", file=sys.stderr, flush=True)
         return
     if to_json:
         print(wallet.model_dump_json())
         return
     
-    wa0 = wallet.accounts[0]
-    wa1 = wallet.accounts[1]
-
-    credits_amount_micro  = wa0.balance.amount_micro if wa0.type == 'CREDITS' else wa1.balance.amount_micro
-    pre_paid_amount_micro = wa0.balance.amount_micro if wa0.type == 'PRE_PAID' else wa1.balance.amount_micro
-    credit_amount = float(credits_amount_micro) / float(1000000)
-    money_amount = float(pre_paid_amount_micro) / float(1000000)
+    credit_balance, prepaid_balance = _get_wallet_balance(wallet)
 
     print(f"Ad account ID = {account_id}")
-    print(f"Wallet ID = {wallet.id}")
-    print(f"PRE_PAID balance = {money_amount}")
-    print(f"CREDITS balance = {credit_amount}")
+    print(f"CREDITS balance = {credit_balance}")
+    print(f"PRE_PAID balance = {prepaid_balance}")
     return
 
 
@@ -86,8 +131,8 @@ def deposit(
     fund_amount: float = typer.Option(help="The fund amount to deposit"), 
     to_curl: bool = typer.Option(False, help="Generate the curl command instead of executing it."),
     to_json: bool = typer.Option(False, help="Print raw output in json"),
-    profile: str = "default",
-    ):
+    profile: str = typer.Option("default", help="profile name of the MCM CLI."),
+):
     """
     Add or top up the money amount to the current balance of the given ad account's wallet.
     """
@@ -137,8 +182,8 @@ def withdraw(
     fund_amount: float = typer.Option(help="The amount of credit to add"), 
     to_curl: bool = typer.Option(False, help="Generate the curl command instead of executing it."),
     to_json: bool = typer.Option(False, help="Print raw output in json"),
-    profile: str = "default",
-    ):    
+    profile: str = typer.Option("default", help="profile name of the MCM CLI."),
+):    
     """
     Withdraws the money amount from the current balance of the given ad account's wallet.
     """
@@ -202,6 +247,25 @@ class WalletCommand:
             "Authorization": f"Bearer {token}"
         }
 
+    def get_platform_balance(
+        self,
+        to_curl: bool,
+    ) -> tuple[
+        None | CurlString,
+        None | Error,
+        None | dict[str, WalletsWrapper]
+    ]:
+        _api_url = f"{self.api_base_url}/wallets-bulk/read"
+
+        curl, error, json_obj = api_request('POST', to_curl, _api_url, self.headers)
+        if curl:
+            return curl, None, None
+        if error:
+            return None, error, None
+
+        platform_wallets_wrapper = PlatformWalletsWrapper(**json_obj)
+        return None, None, platform_wallets_wrapper.result
+
 
     def get_balance(
         self,

mcmcli/data/wallet.py

@@ -67,3 +67,46 @@ class Wallet(BaseModel):
 
 class WalletsWrapper(BaseModel):
     wallets: list[Wallet]
+
+
+#
+# API response dataclasses for the platform balance API responses
+#
+#
+# {
+#     "result": {
+#         "10054": {
+#             "wallets": [
+#                 {
+#                     "id": "wRydQ9MhQx4ZyJED", 
+#                     "title": "", 
+#                     "ad_account_id": "10054", 
+#                     "type": "PRE_PAYMENT", 
+#                     "accounts": [
+#                         {
+#                             "type": "CREDITS", 
+#                             "balance": {
+#                                 "currency": "USD", 
+#                                 "amount_micro": "1890020000"
+#                             }
+#                         },
+#                         {
+#                             "type": "PRE_PAID", 
+#                             "balance": {
+#                                 "currency": "USD", 
+#                                 "amount_micro": "0"
+#                             }
+#                         }
+#                     ], 
+#                     "created_at": "2024-09-24T17:26:09.437580Z", 
+#                     "updated_at": "2024-10-20T22:00:21.412482Z"
+#                 }
+#             ]
+#         }
+#     }
+# }
+
+class PlatformWalletsWrapper(BaseModel):
+    result: dict[str, WalletsWrapper] # Dictionary with ad_account_id as keys
+
+

mcmcli/requests.py

@@ -38,6 +38,8 @@ def get(url, headers):
     try:
         res = requests.get(url, headers=headers)
         return None, json.loads(res.text)
+    except json.JSONDecodeError:
+        return Error(code=0, message=res.text), None
     except Exception as e:
         return e, None
 
@@ -46,6 +48,8 @@ def delete(url, headers):
     try:
         res = requests.delete(url, headers=headers)
         return None, json.loads(res.text)
+    except json.JSONDecodeError:
+        return Error(code=0, message=res.text), None
     except Exception as e:
         return e, None
 
@@ -54,6 +58,8 @@ def post(url, headers, payload):
     try:
         res = requests.post(url, headers=headers, json=payload)
         return None, json.loads(res.text)
+    except json.JSONDecodeError:
+        return Error(code=0, message=res.text), None
     except Exception as e:
         return  e, None
 
@@ -61,6 +67,8 @@ def put(url, headers, payload):
     try:
         res = requests.put(url, headers=headers, json=payload)
         return None, json.loads(res.text)
+    except json.JSONDecodeError:
+        return Error(code=0, message=res.text), None
     except Exception as e:
         return e, None