mcm-cli 1.0.1__tar.gz → 1.1.0__tar.gz

{mcm_cli-1.0.1/mcm_cli.egg-info → mcm_cli-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mcm-cli
-Version: 1.0.1
+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.1 → mcm_cli-1.1.0/mcm_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mcm-cli
-Version: 1.0.1
+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.1 → mcm_cli-1.1.0}/mcm_cli.egg-info/SOURCES.txt

@@ -13,6 +13,7 @@ mcmcli/__main__.py
 mcmcli/logging.py
 mcmcli/requests.py
 mcmcli/command/account.py
+mcmcli/command/admin.py
 mcmcli/command/auth.py
 mcmcli/command/config.py
 mcmcli/command/decision.py

{mcm_cli-1.0.1 → mcm_cli-1.1.0}/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.1")
+	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")

mcm_cli-1.1.0/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}')
+

{mcm_cli-1.0.1 → mcm_cli-1.1.0}/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)
@@ -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)
@@ -182,10 +182,19 @@ def withdraw(
     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,

{mcm_cli-1.0.1 → mcm_cli-1.1.0}/setup.py

@@ -18,7 +18,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='mcm-cli',
-    version='1.0.1',
+    version='1.1.0',
     description='A command-line interface for Moloco Commerde Media',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',