Handle '.' in repo name (#158)

potpie-ai · Nov 4, 2024 · 3935edb · 3935edb
1 parent 6b76511
commit 3935edb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/app/modules/parsing/graph_construction/parsing_helper.py b/app/modules/parsing/graph_construction/parsing_helper.py
@@ -150,15 +150,14 @@ async def download_and_extract_tarball(
         except requests.exceptions.RequestException as e:
             logger.error(f"Error fetching tarball: {e}")
             return e
-
         tarball_path = os.path.join(
             target_dir,
-            f"{repo.full_name.replace('/', '-')}-{branch.replace('/', '-')}.tar.gz",
+            f"{repo.full_name.replace('/', '-').replace('.', '-')}-{branch.replace('/', '-').replace('.', '-')}.tar.gz",
         )
 
         final_dir = os.path.join(
             target_dir,
-            f"{repo.full_name.replace('/', '-')}-{branch.replace('/', '-')}-{user_id}",
+            f"{repo.full_name.replace('/', '-').replace('.', '-')}-{branch.replace('/', '-').replace('.', '-')}-{user_id}",
         )
 
         try:
@@ -171,6 +170,8 @@ async def download_and_extract_tarball(
                 extracted_dir = os.path.join(temp_dir, os.listdir(temp_dir)[0])
                 for root, dirs, files in os.walk(extracted_dir):
                     for file in files:
+                        if file.startswith("."):
+                            continue
                         file_path = os.path.join(root, file)
                         if self.is_text_file(file_path):
                             try: