Merge pull request #91 from goldlabelapps/staging

This pull request refactors the CSV import functionality for the queue API by introducing a dedicated `csv` module, adds a new (template) Apollo CSV import endpoint, and cleans up test files.

Author: goldlabelapps

app/api/queue/__init__.py

@@ -9,8 +9,7 @@
 from .routes.create import router as create_router
 from .routes.delete import router as delete_router
 
-from .routes.import_linkedin import router as import_linkedin_router
-
+from .csv import linkedin as linkedin_import_router
 from .routes.alter import router as alter_router
 from .routes.rename_column import router as rename_router
 
@@ -20,6 +19,10 @@
 router.include_router(get_router)
 router.include_router(create_router)
 router.include_router(delete_router)
-router.include_router(import_linkedin_router)
+router.include_router(linkedin_import_router.router)
+
+# Register Apollo CSV import route
+from .csv import apollo as apollo_import_router
+router.include_router(apollo_import_router.router)
 router.include_router(alter_router)
 router.include_router(rename_router)

app/api/queue/csv/README.md

@@ -0,0 +1 @@
+# Placeholder for csv import routes

app/api/queue/csv/__init__.py

@@ -0,0 +1 @@
+# Makes this directory a Python package

app/api/queue/csv/apollo.py

@@ -0,0 +1,29 @@
+import os
+import csv
+import time
+from fastapi import APIRouter, HTTPException
+from app.utils.make_meta import make_meta
+from app.utils.db import get_db_connection_direct
+
+router = APIRouter()
+
+@router.post("/queue/csv/apollo")
+def import_apollo_csv() -> dict:
+    """POST /queue/csv/apollo: Import data from apollo.csv into the queue table (template)."""
+    csv_path = os.path.join(os.path.dirname(__file__), "../csv/apollo/seed.csv")
+    if not os.path.exists(csv_path):
+        raise HTTPException(status_code=404, detail="seed.csv not found")
+    try:
+        conn = get_db_connection_direct()
+        cursor = conn.cursor()
+        # TODO: Implement CSV parsing and DB insertion logic for Apollo format
+        # Example placeholder for batch import logic:
+        # with open(csv_path, newline='', encoding='utf-8') as csvfile:
+        #     reader = csv.DictReader(csvfile)
+        #     for row in reader:
+        #         pass  # Process each row
+        conn.commit()
+        conn.close()
+        return {"meta": make_meta("success", "Apollo CSV import template executed"), "imported": 0}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

app/api/queue/routes/import_linkedin.py app/api/queue/csv/linkedin.pyapp/api/queue/routes/import_linkedin.py renamed to app/api/queue/csv/linkedin.py

@@ -7,9 +7,9 @@
 
 router = APIRouter()
 
-@router.post("/queue/import/linkedin")
+@router.post("/queue/csv/linkedin")
 def import_linkedin_csv() -> dict:
-    """POST /queue/import/linkedin: Import data from linkedin.csv into the queue table, robust for large files."""
+    """POST /queue/csv/linkedin: Import data from linkedin.csv into the queue table, robust for large files."""
     csv_path = os.path.join(os.path.dirname(__file__), "../csv/linkedin/linkedin.csv")
     if not os.path.exists(csv_path):
         raise HTTPException(status_code=404, detail="linkedin.csv not found")