Merge pull request #14 from diffgram/fix-permissions-on-file-exists

fix condition

Author: anthony-chaudhary

sdk/diffgram/core/core.py

@@ -17,6 +17,7 @@
 from diffgram.brain.train import Train
 from diffgram.export.export import Export
 from diffgram.task.task import Task
+from requests.auth import HTTPBasicAuth
 
 
 class Project():
@@ -73,6 +74,9 @@ def get_member_list(self):
 		data = response.json()
 		return data['project']['member_list']
 
+	def get_http_auth(self):
+		return HTTPBasicAuth(self.client_id, self.client_secret)
+
 	def get_label(
 			self,
 			name=None,

sdk/diffgram/core/diffgram_dataset_iterator.py

@@ -87,7 +87,9 @@ def __next__(self):
     def __validate_file_ids(self):
         if not self.diffgram_file_id_list:
             return
-        result = self.project.file.file_list_exists(self.diffgram_file_id_list, use_session = False)
+        result = self.project.file.file_list_exists(
+            self.diffgram_file_id_list, 
+            use_session = False)
         if not result:
             raise Exception(
                 'Some file IDs do not belong to the project. Please provide only files from the same project.')

sdk/diffgram/file/file_constructor.py

@@ -5,7 +5,6 @@
 import json
 import os
 import requests
-from requests.auth import HTTPDigestAuth
 
 class FileConstructor():
     """
@@ -422,7 +421,7 @@ def file_list_exists(self, id_list, use_session = True):
             response = requests.post(
                 url = self.client.host + url,
                 json = spec_dict,
-                auth = HTTPDigestAuth(self.client.client_id, self.client.client_secret)
+                auth = self.client.get_http_auth()
             )
         self.client.handle_errors(response)
 
@@ -466,7 +465,7 @@ def get_by_id(self,
             # Add Auth
             response = requests.post(self.client.host + endpoint,
                           json = spec_dict,
-                          auth = HTTPDigestAuth(self.client.client_id, self.client.client_secret))
+                          auth = self.client.get_http_auth())
 
         self.client.handle_errors(response)
 

sdk/diffgram/file/view.py

@@ -1,5 +1,4 @@
 import requests
-from requests.auth import HTTPDigestAuth
 
 
 def get_file_id():
@@ -49,7 +48,7 @@ def get_label_file_dict(self, use_session = True):
         # Add Auth
         response = requests.get(self.host + endpoint,
                                 headers = {'directory_id': str(self.directory_id)},
-                                auth = HTTPDigestAuth(self.client_id, self.client_secret))
+                                auth = self.client.get_http_auth())
 
     self.handle_errors(response)