Get disks, images and snapshots

azurekit/manager.py

@@ -1,3 +1,4 @@
+from os import environ
 from .services import *
 
 class AzureManager(object):

azurekit/services/base.py

@@ -1,10 +1,17 @@
+from os import environ
+import sys
 from azure.common.client_factory import get_client_from_auth_file
 
 class BaseService(object):
 
     def set_client(self, service):
+
         self.service = service
-        self.client = get_client_from_auth_file(service)
+        try:
+            self.client = get_client_from_auth_file(service)
+        except (FileNotFoundError, KeyError):
+            print ('[!] Credentials file not found. Set path in "AZURE_AUTH_LOCATION" environment variable')
+            sys.exit(0)
 
     def __init__(self, service):
         self.set_client(service)

azurekit/services/compute.py

@@ -4,9 +4,18 @@
 
 class ComputeService(BaseService):
 
-    def get_instances(self):
+    def get_machines(self):
         return self.client.virtual_machines.list_all()
 
+    def get_disks(self):
+        return self.client.disks.list()
+
+    def get_snapshots(self):
+        return self.client.snapshots.list()
+
+    def get_images(self):
+        return self.client.images.list()
+
 
     def __init__(self):
         BaseService.__init__(self, ComputeManagementClient)