fix property access after change

techstormpc · Oct 14, 2022 · da8965c · da8965c
1 parent 28870f1
commit da8965c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/partner_tools/cli/controller.py b/partner_tools/cli/controller.py
@@ -40,7 +40,7 @@ def run_register_device():
         'hardwareHash': '',
         'productKey': ''
     }
-    client.customer.enroll_devices([device], customer['id'], DEVICE_BATCH_NAME)
+    client.customer.enroll_devices([device], customer.id, DEVICE_BATCH_NAME)
 
     print()
     print('Device was added')
@@ -60,7 +60,7 @@ def run_get_devices():
 
     customer = get_customer(customers)
 
-    devices = client.customer.get_devices(customer['id'], DEVICE_BATCH_NAME)
+    devices = client.customer.get_devices(customer.id, DEVICE_BATCH_NAME)
     print(devices)
 
 

diff --git a/partner_tools/cli/interactive/common_args.py b/partner_tools/cli/interactive/common_args.py
@@ -5,13 +5,13 @@
 
 
 def get_customer(customers: List[Customer]):
-    choices = [f'{customer["companyProfile"]["companyName"]} - ({customer["id"]})'
+    choices = [f'{customer.companyProfile.companyName} - ({customer.id})'
                for customer in customers]
 
     resp = ask(QuestionType.select,
                "Please select a customer:",
                choices=choices)
 
     customer = [x for x in customers
-                if f'{x["companyProfile"]["companyName"]} - ({x["id"]})' == resp][0]
+                if f'{x.companyProfile.companyName} - ({x.id})' == resp][0]
     return customer
diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
 
 setup(
     name='ms-partner-tools',
-    version='0.1.0',
+    version='0.1.1',
     author='TechStorm PC',
     license='MIT',
     classifiers=[