Refactor to use new .get(), and better support Default dataset

Author: anthony-chaudhary

sdk/diffgram/core/core.py

@@ -21,6 +21,7 @@
 class Project():
     default_directory: Directory
     last_response_header: None
+    directory_list: list
 
     def __init__(
             self,
@@ -34,6 +35,7 @@ def __init__(
             refresh_local_label_dict = True
 
     ):
+        self.directory_list = []
 
         self.session = requests.Session()
         adapter = requests.adapters.HTTPAdapter(pool_connections = 30, pool_maxsize = 30)
@@ -84,7 +86,6 @@ def __init__(
 
         self.label_schema_list = self.get_label_schema_list()
 
-        self.directory_list = None
 
 
     def get_member_list(self):
@@ -314,8 +315,11 @@ def set_default_directory(self,
         if directory is not None:
             self.directory_id = directory.id
             self.default_directory = directory
-        if not hasattr(self, 'directory_list'):
-            self.directory_list = self.directory.get_directory_list()
+
+        if self.directory_id is None:
+            default_directory = self.directory.get()
+            self.directory_id = default_directory.id
+            self.default_directory = default_directory
 
         self.session.headers.update(
             {'directory_id': str(self.directory_id)})

sdk/diffgram/core/directory.py

@@ -299,7 +299,7 @@ def list_files(
             return file_list
 
     def get(self,
-            name: str):
+            name: str = None):
 
         """
 
@@ -310,8 +310,6 @@ def get(self,
          TODO refactor set_directory_by_name() to use this
 
         """
-        if name is None:
-            raise Exception("No name provided.")
 
         # Cache
         if not self.client.directory_list: