Fix creation of python server file

DistriNet · Nov 15, 2024 · cf43c7d · cf43c7d
1 parent 301da7e
commit cf43c7d
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/bci/evaluations/custom/custom_evaluation.py b/bci/evaluations/custom/custom_evaluation.py
@@ -221,11 +221,17 @@ def add_page(self, project: str, poc: str, domain: str, path: str, file_type: st
         domain_path = os.path.join(Global.custom_page_folder, project, poc, domain)
         if not os.path.exists(domain_path):
             os.makedirs(domain_path)
-        page_path = os.path.join(domain_path, path)
-        if not os.path.exists(page_path):
-            os.makedirs(page_path)
-        new_file_name = f'index.{file_type}'
-        file_path = os.path.join(page_path, new_file_name)
+
+        if file_type == 'py':
+            file_name = path if path.endswith('.py') else path + '.py'
+            file_path = os.path.join(domain_path, file_name)
+        else:
+            page_path = os.path.join(domain_path, path)
+            if not os.path.exists(page_path):
+                os.makedirs(page_path)
+            new_file_name = f'index.{file_type}'
+            file_path = os.path.join(page_path, new_file_name)
+
         if os.path.exists(file_path):
             return False
         with open(file_path, 'w') as file: