Fixed a bug where stubs for 'training' or 'pricing' were imported ins…

…tead of stubs for service

snet/sdk/__init__.py

@@ -95,8 +95,8 @@ def create_service_client(self, org_id: str, service_id: str, group_name=None,
             lib_generator.generate_client_library()
         else:
             path_to_pb_files = self.get_path_to_pb_files(org_id, service_id)
-            pb_2_file_name = find_file_by_keyword(path_to_pb_files, keyword="pb2.py")
-            pb_2_grpc_file_name = find_file_by_keyword(path_to_pb_files, keyword="pb2_grpc.py")
+            pb_2_file_name = find_file_by_keyword(path_to_pb_files, keyword="pb2.py", exclude=['training', 'pricing'])
+            pb_2_grpc_file_name = find_file_by_keyword(path_to_pb_files, keyword="pb2_grpc.py", exclude=['training', 'pricing'])
             if not pb_2_file_name or not pb_2_grpc_file_name:
                 lib_generator.generate_client_library()
 
@@ -145,8 +145,10 @@ def get_path_to_pb_files(self, org_id: str, service_id: str) -> str:
 
     def get_module_by_keyword(self, org_id: str, service_id: str, keyword: str) -> ModuleName:
         path_to_pb_files = self.get_path_to_pb_files(org_id, service_id)
-        file_name = find_file_by_keyword(path_to_pb_files, keyword)
+        file_name = find_file_by_keyword(path_to_pb_files, keyword, exclude=['training', 'pricing'])
+        # print(f"File name: {file_name}")
         module_name = os.path.splitext(file_name)[0]
+        # print(f"Module name: {module_name}")
         return ModuleName(module_name)
 
     def get_service_metadata(self, org_id, service_id):

snet/sdk/service_client.py

@@ -201,7 +201,7 @@ def get_path_to_pb_files(self, org_id: str, service_id: str) -> str:
     def get_services_and_messages_info(self):
         # Get proto file filepath and open it
         path_to_pb_files = self.get_path_to_pb_files(self.org_id, self.service_id)
-        proto_file_name = find_file_by_keyword(path_to_pb_files, keyword=".proto")
+        proto_file_name = find_file_by_keyword(path_to_pb_files, keyword=".proto", exclude=['training', 'pricing'])
         proto_filepath = os.path.join(path_to_pb_files, proto_file_name)
         with open(proto_filepath, 'r') as file:
             proto_content = file.read()

snet/sdk/utils/utils.py

@@ -142,10 +142,12 @@ def __exit__(self, exc_type, exc_value, traceback):
             pass
 
 
-def find_file_by_keyword(directory, keyword):
+def find_file_by_keyword(directory, keyword, exclude=None):
+    if exclude is None:
+        exclude = []
     for root, dirs, files in os.walk(directory):
         for file in files:
-            if keyword in file:
+            if keyword in file and all(e not in file for e in exclude):
                 return file