Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions app/api/queue/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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)
1 change: 1 addition & 0 deletions app/api/queue/csv/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
# Placeholder for csv import routes
1 change: 1 addition & 0 deletions app/api/queue/csv/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
# Makes this directory a Python package
29 changes: 29 additions & 0 deletions app/api/queue/csv/apollo.py
Original file line number Diff line number Diff line change
@@ -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))
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down
70 changes: 0 additions & 70 deletions tests/test_prompt.py

This file was deleted.

Loading