Text Extraction Done, UI Fixes Done, Extracted Template Layout Saved in DB
This commit is contained in:
53
backend/api/templates.py
Normal file
53
backend/api/templates.py
Normal file
@@ -0,0 +1,53 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException, status
|
||||
from sqlalchemy.orm import Session
|
||||
from typing import List
|
||||
|
||||
from database import get_db
|
||||
from schemas.template_schemas import (
|
||||
TemplateCreate, TemplateSchema,
|
||||
TemplateFieldCreate, TemplateFieldSchema,
|
||||
TemplateFieldMappingCreate, TemplateFieldMappingSchema
|
||||
)
|
||||
from services.template_service import TemplateService
|
||||
|
||||
router = APIRouter(prefix="/api/templates", tags=["Templates"])
|
||||
|
||||
@router.post("", response_model=TemplateSchema, status_code=status.HTTP_201_CREATED)
|
||||
def create_template(template_data: TemplateCreate, db: Session = Depends(get_db)):
|
||||
service = TemplateService(db)
|
||||
return service.create_template(template_data)
|
||||
|
||||
@router.get("", response_model=List[TemplateSchema])
|
||||
def get_templates(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
|
||||
service = TemplateService(db)
|
||||
return service.get_all_templates(skip, limit)
|
||||
|
||||
@router.get("/{template_id}", response_model=TemplateSchema)
|
||||
def get_template(template_id: int, db: Session = Depends(get_db)):
|
||||
service = TemplateService(db)
|
||||
template = service.get_template(template_id)
|
||||
if not template:
|
||||
raise HTTPException(status_code=404, detail="Template not found")
|
||||
return template
|
||||
|
||||
@router.delete("/{template_id}", status_code=status.HTTP_204_NO_CONTENT)
|
||||
def delete_template(template_id: int, db: Session = Depends(get_db)):
|
||||
service = TemplateService(db)
|
||||
if not service.delete_template(template_id):
|
||||
raise HTTPException(status_code=404, detail="Template not found")
|
||||
return None
|
||||
|
||||
@router.post("/{template_id}/fields", response_model=TemplateFieldSchema, status_code=status.HTTP_201_CREATED)
|
||||
def add_template_field(template_id: int, field_data: TemplateFieldCreate, db: Session = Depends(get_db)):
|
||||
service = TemplateService(db)
|
||||
# Check if template exists
|
||||
if not service.get_template(template_id):
|
||||
raise HTTPException(status_code=404, detail="Template not found")
|
||||
return service.add_template_field(template_id, field_data)
|
||||
|
||||
@router.post("/{template_id}/mappings/save", response_model=List[TemplateFieldMappingSchema])
|
||||
def save_mappings(template_id: int, mappings: List[TemplateFieldMappingCreate], db: Session = Depends(get_db)):
|
||||
service = TemplateService(db)
|
||||
if not service.get_template(template_id):
|
||||
raise HTTPException(status_code=404, detail="Template not found")
|
||||
return service.save_mapping(template_id, mappings)
|
||||
Reference in New Issue
Block a user