mcm-cli 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

{mcm_cli-1.0.0.dist-info → mcm_cli-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mcm-cli
-Version: 1.0.0
+Version: 1.1.0
 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.0.0.dist-info → mcm_cli-1.1.0.dist-info}/RECORD

@@ -1,12 +1,13 @@
 mcmcli/__init__.py,sha256=-U6lMZ9_99IXAKwnqnYXYr6NcO6TSmG-kxewgvJjU4k,575
-mcmcli/__main__.py,sha256=6XmesgSjMyL2ui_SpQpl0RMSRr2CQ7DWVx8a-ZCKZxw,1473
+mcmcli/__main__.py,sha256=9eAfYXz2reT3cOAgkqgCAuk169yB8KSla_rIF4oak6E,1616
 mcmcli/logging.py,sha256=xjRS5ey1ONx_d34qB1Fetb_SwPysoh2hzNDuNAaYYWQ,1739
 mcmcli/requests.py,sha256=50N_LiWIWr8-3EPs_yR9f4uEQf8rQ68s_FoMYRhgjzI,2343
 mcmcli/command/account.py,sha256=EtcYZsC5LL5ue-hKsZZl3_0oZ09nJdVi774yqiTykl8,14712
+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=TB3WrmniREsDk0ui20p3ha6OZMwA2wAJaxQ9QDGtKnA,8968
+mcmcli/command/wallet.py,sha256=0MprkN5hJqJCTbMpMnFLRcLKnnc5PNYpx5Sjhys6XTY,9242
 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
@@ -14,10 +15,10 @@ mcmcli/data/error.py,sha256=d6xa_jTXumlA0EzXy2PJQ86ajBb0Pm90fss9R3LuHUc,1094
 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.0.0.dist-info/LICENSE,sha256=RFhQPdSOiMTguUX7JSoIuTxA7HVzCbj_p8WU36HjUQQ,10947
-mcm_cli-1.0.0.dist-info/METADATA,sha256=NJGEah01TQz9RJrMtgM_Y1-a8Mz9DvZ6y4YB4mseCdo,3018
-mcm_cli-1.0.0.dist-info/NOTICE,sha256=Ldnl2MjRaXPxcldUdbI2NTybq60XAa2LowRhFrRTuiI,76
-mcm_cli-1.0.0.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-mcm_cli-1.0.0.dist-info/entry_points.txt,sha256=qTHAWZ-ejSiU4t11RYwtAU8ScqhQPDeMVTG9y4wMVLg,60
-mcm_cli-1.0.0.dist-info/top_level.txt,sha256=sh7oqIaqLQlMtKHlxHHgpV2xGMrBMPFWpSp0C6nvJ_Y,7
-mcm_cli-1.0.0.dist-info/RECORD,,
+mcm_cli-1.1.0.dist-info/LICENSE,sha256=RFhQPdSOiMTguUX7JSoIuTxA7HVzCbj_p8WU36HjUQQ,10947
+mcm_cli-1.1.0.dist-info/METADATA,sha256=zXrJD8pf59IjAeaqxJed60ROfk6-5eFQprXggZZSqOM,3018
+mcm_cli-1.1.0.dist-info/NOTICE,sha256=Ldnl2MjRaXPxcldUdbI2NTybq60XAa2LowRhFrRTuiI,76
+mcm_cli-1.1.0.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
+mcm_cli-1.1.0.dist-info/entry_points.txt,sha256=qTHAWZ-ejSiU4t11RYwtAU8ScqhQPDeMVTG9y4wMVLg,60
+mcm_cli-1.1.0.dist-info/top_level.txt,sha256=sh7oqIaqLQlMtKHlxHHgpV2xGMrBMPFWpSp0C6nvJ_Y,7
+mcm_cli-1.1.0.dist-info/RECORD,,

mcmcli/__main__.py

@@ -17,7 +17,10 @@
 """mcm cli entry point script."""
 # mcmcli/__main__.py
 
+from typing import Optional
+
 import mcmcli.command.account
+import mcmcli.command.admin
 import mcmcli.command.auth
 import mcmcli.command.config
 import mcmcli.command.decision
@@ -32,9 +35,10 @@ def version():
 	"""
 	Show the tool version
 	"""
-	typer.echo(f"Version: mcm-cli v1.0.0")
+	typer.echo(f"Version: mcm-cli v1.1.0")
 
 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")
 app.add_typer(mcmcli.command.auth.app, name="auth", help="Authentication management")
 app.add_typer(mcmcli.command.config.app, name="config", help="Configurations")
 app.add_typer(mcmcli.command.decision.app, name="decision", help="Decision command")

mcmcli/command/admin.py

@@ -0,0 +1,88 @@
+# Copyright 2023 Moloco, Inc
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import mcmcli.command.account
+import mcmcli.command.auth
+import mcmcli.command.config
+import mcmcli.command.wallet
+import mcmcli.requests
+import sys
+import typer
+
+app = typer.Typer(add_completion=False)
+
+def _create_admin_command(profile):
+    auth = mcmcli.command.auth.AuthCommand(profile)
+    _, error, token = auth.get_token()
+    if error:
+        print(f"ERROR: {error.message}")
+        return None
+    return AdminCommand(profile, auth, token.token)
+
+@app.command()
+def list_wallet_balances(
+    profile: str = typer.Option("default", help="profile name of the MCM CLI."),
+):
+    """
+    List the wallet balances of all of the ad accounts
+    """
+    admin = _create_admin_command(profile)
+    if admin is None:
+        return
+    admin.list_wallet_balances()
+
+
+class AdminCommand:
+    def __init__(
+        self,
+        profile,
+        auth_command: mcmcli.command.auth.AuthCommand,
+        token
+    ):
+        self.profile = profile
+        self.auth_command = auth_command
+        self.config = mcmcli.command.config.get_config(profile)
+        mcmcli.command.config.assert_config_exists(self.config)
+
+        self.token = token
+        self.api_base_url = f"{self.config['management_api_hostname']}/rmp/mgmt/v1/platforms/{self.config['platform_id']}"
+        self.headers = {
+            "accept": "application/json",
+            "content-type": "application/json",
+            "Authorization": f"Bearer {token}"
+        }
+
+    def list_wallet_balances(
+        self
+    ):
+        ac = mcmcli.command.account.AccountCommand(self.profile, self.auth_command, self.token)
+        wc = mcmcli.command.wallet.WalletCommand(self.profile, self.auth_command, self.token)
+        _, error, accounts = ac.list_accounts()
+        if error:
+            print(error, file=sys.stderr, flush=True)
+            return
+
+        print("ad_account_title, ad_account_id, credit_balance, prepaid_balance")
+        for id in accounts:
+            _, error, wallet = wc.get_balance(id, to_curl=False)
+            if error:
+                continue
+            w0 = wallet.accounts[0]
+            w1 = wallet.accounts[1]
+            credits = w0 if w0.type == 'CREDITS' else w1
+            prepaid = w1 if w1.type == 'PRE_PAID' else w0
+            credits = int(credits.balance.amount_micro) / 1000000
+            prepaid = int(prepaid.balance.amount_micro) / 1000000
+            print(f'"{accounts[id].title}", {id}, {credits}, {prepaid}')
+

mcmcli/command/wallet.py

@@ -52,7 +52,7 @@ def balance(
         print(f"ERROR: {error.message}")
         return
 
-    wc = WalletCommand(profile, token.token)
+    wc = WalletCommand(profile, auth, token.token)
     curl, error, wallet = wc.get_balance(account_id, to_curl)
     if to_curl:
         print(curl)
@@ -97,7 +97,7 @@ def deposit(
         print(f"ERROR: {error.message}")
         return
 
-    wc = WalletCommand(profile, token.token)
+    wc = WalletCommand(profile, auth, token.token)
 
     # Check the wallet first
     curl, error, wallet = wc.get_balance(account_id, to_curl=False)
@@ -127,7 +127,7 @@ def deposit(
     credits_amount  = float(credits_amount_micro)  / float(1000000)
     pre_paid_amount = float(pre_paid_amount_micro) / float(1000000)
 
-    print(f"Deposited to the wallet. Currence balance of the ad account id {account_id} is {pre_paid_amount} for PRE_PAID and {credits_amount} for CREDITS.")
+    print(f"Funds have been deposited into the wallet. The current balance for ad account ID {account_id} is {pre_paid_amount} in PRE_PAID and {credits_amount} in CREDITS.")
     return
 
 @app.command()
@@ -148,7 +148,7 @@ def withdraw(
         print(f"ERROR: {error.message}")
         return
 
-    wc = WalletCommand(profile, token.token)
+    wc = WalletCommand(profile, auth, token.token)
 
     # Check the wallet first
     curl, error, wallet = wc.get_balance(account_id, to_curl=False)
@@ -178,14 +178,23 @@ def withdraw(
     credits_amount  = float(credits_amount_micro)  / float(1000000)
     pre_paid_amount = float(pre_paid_amount_micro) / float(1000000)
 
-    print(f"Withdrew the funds from to the wallet. Currence balance of the ad account id {account_id} is {pre_paid_amount} for PRE_PAID and {credits_amount} for CREDITS.")
+    print(f"Funds were withdrawn out of the wallet. The current balance of ad account ID {account_id} is {pre_paid_amount} for PRE_PAID and {credits_amount} for CREDITS.")
     return
 
 class WalletCommand:
-    def __init__(self, profile, token):
+    def __init__(
+        self,
+        profile,
+        auth_command: mcmcli.command.auth.AuthCommand,
+        token
+    ):
         self.config = mcmcli.command.config.get_config(profile)
-        mcmcli.command.config.assert_config_exists(self.config)
+        if (self.config is None):
+            print(f"ERROR: Failed to load the CLI profile", file=sys.stderr, flush=True)
+            sys.exit()
 
+        self.profile = profile
+        self.auth_command = auth_command
         self.api_base_url = f"{self.config['management_api_hostname']}/rmp/mgmt/v1/platforms/{self.config['platform_id']}"
         self.headers = {
             "accept": "application/json",
@@ -193,6 +202,7 @@ class WalletCommand:
             "Authorization": f"Bearer {token}"
         }
 
+
     def get_balance(
         self,
         account_id: str,