| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274 |
- import logging
- import os
- import uuid
- from datetime import datetime
- from pathlib import Path
- from fastapi import APIRouter, Depends, File, HTTPException, UploadFile
- from fastapi.responses import FileResponse
- from sqlalchemy import case, func, select
- from sqlalchemy.ext.asyncio import AsyncSession
- from sqlalchemy.orm import selectinload
- from backend.app.core.config import settings
- from backend.app.core.database import get_db
- from backend.app.models.archive import PrintArchive
- from backend.app.models.print_queue import PrintQueueItem
- from backend.app.models.project import Project
- from backend.app.models.project_bom import ProjectBOMItem
- from backend.app.schemas.project import (
- ArchivePreview,
- BatchAddArchives,
- BatchAddQueueItems,
- BOMItemCreate,
- BOMItemResponse,
- BOMItemUpdate,
- ProjectChildPreview,
- ProjectCreate,
- ProjectListResponse,
- ProjectResponse,
- ProjectStats,
- ProjectUpdate,
- TimelineEvent,
- )
- logger = logging.getLogger(__name__)
- router = APIRouter(prefix="/projects", tags=["projects"])
- async def compute_project_stats(db: AsyncSession, project_id: int, target_count: int | None = None) -> ProjectStats:
- """Compute statistics for a project."""
- # Count total archives
- total_result = await db.execute(select(func.count(PrintArchive.id)).where(PrintArchive.project_id == project_id))
- total_archives = total_result.scalar() or 0
- # Count completed archives
- completed_result = await db.execute(
- select(func.count(PrintArchive.id)).where(
- PrintArchive.project_id == project_id, PrintArchive.status == "completed"
- )
- )
- completed_prints = completed_result.scalar() or 0
- # Count failed archives
- failed_result = await db.execute(
- select(func.count(PrintArchive.id)).where(
- PrintArchive.project_id == project_id, PrintArchive.status == "failed"
- )
- )
- failed_prints = failed_result.scalar() or 0
- # Sum print time, filament, and energy
- sums_result = await db.execute(
- select(
- func.coalesce(func.sum(PrintArchive.print_time_seconds), 0).label("total_time"),
- func.coalesce(func.sum(PrintArchive.filament_used_grams), 0).label("total_filament"),
- func.coalesce(func.sum(PrintArchive.cost), 0).label("total_filament_cost"),
- func.coalesce(func.sum(PrintArchive.energy_kwh), 0).label("total_energy"),
- func.coalesce(func.sum(PrintArchive.energy_cost), 0).label("total_energy_cost"),
- ).where(PrintArchive.project_id == project_id)
- )
- sums = sums_result.first()
- # Count queued items
- queued_result = await db.execute(
- select(func.count(PrintQueueItem.id)).where(
- PrintQueueItem.project_id == project_id, PrintQueueItem.status == "pending"
- )
- )
- queued_prints = queued_result.scalar() or 0
- # Count in-progress items
- in_progress_result = await db.execute(
- select(func.count(PrintQueueItem.id)).where(
- PrintQueueItem.project_id == project_id, PrintQueueItem.status == "printing"
- )
- )
- in_progress_prints = in_progress_result.scalar() or 0
- # Calculate progress
- progress_percent = None
- remaining_prints = None
- if target_count and target_count > 0:
- progress_percent = round((completed_prints / target_count) * 100, 1)
- remaining_prints = max(0, target_count - completed_prints)
- # BOM stats
- bom_result = await db.execute(
- select(
- func.count(ProjectBOMItem.id).label("total"),
- func.sum(case((ProjectBOMItem.quantity_acquired >= ProjectBOMItem.quantity_needed, 1), else_=0)).label(
- "completed"
- ),
- ).where(ProjectBOMItem.project_id == project_id)
- )
- bom_stats = bom_result.first()
- return ProjectStats(
- total_archives=total_archives,
- completed_prints=completed_prints,
- failed_prints=failed_prints,
- queued_prints=queued_prints,
- in_progress_prints=in_progress_prints,
- total_print_time_hours=round((sums.total_time or 0) / 3600, 2),
- total_filament_grams=round(sums.total_filament or 0, 2),
- progress_percent=progress_percent,
- estimated_cost=round((sums.total_filament_cost or 0), 2),
- total_energy_kwh=round((sums.total_energy or 0), 3),
- total_energy_cost=round((sums.total_energy_cost or 0), 2),
- remaining_prints=remaining_prints,
- bom_total_items=bom_stats.total or 0,
- bom_completed_items=int(bom_stats.completed or 0),
- )
- @router.get("", response_model=list[ProjectListResponse])
- @router.get("/", response_model=list[ProjectListResponse])
- async def list_projects(
- status: str | None = None,
- db: AsyncSession = Depends(get_db),
- ):
- """List all projects with basic stats."""
- query = select(Project)
- if status:
- query = query.where(Project.status == status)
- query = query.order_by(Project.updated_at.desc())
- result = await db.execute(query)
- projects = result.scalars().all()
- # Compute quick stats for each project
- response = []
- for project in projects:
- # Get archive count
- archive_count_result = await db.execute(
- select(func.count(PrintArchive.id)).where(PrintArchive.project_id == project.id)
- )
- archive_count = archive_count_result.scalar() or 0
- # Get queue count
- queue_count_result = await db.execute(
- select(func.count(PrintQueueItem.id)).where(
- PrintQueueItem.project_id == project.id,
- PrintQueueItem.status.in_(["pending", "printing"]),
- )
- )
- queue_count = queue_count_result.scalar() or 0
- # Get completed count for progress
- completed_result = await db.execute(
- select(func.count(PrintArchive.id)).where(
- PrintArchive.project_id == project.id,
- PrintArchive.status == "completed",
- )
- )
- completed_count = completed_result.scalar() or 0
- progress_percent = None
- if project.target_count and project.target_count > 0:
- progress_percent = round((completed_count / project.target_count) * 100, 1)
- # Get archive previews (up to 6 most recent)
- archives_result = await db.execute(
- select(PrintArchive)
- .where(PrintArchive.project_id == project.id)
- .order_by(PrintArchive.created_at.desc())
- .limit(6)
- )
- archives = archives_result.scalars().all()
- archive_previews = [
- ArchivePreview(
- id=a.id,
- print_name=a.print_name,
- thumbnail_path=a.thumbnail_path,
- status=a.status,
- filament_type=a.filament_type,
- filament_color=a.filament_color,
- )
- for a in archives
- ]
- response.append(
- ProjectListResponse(
- id=project.id,
- name=project.name,
- description=project.description,
- color=project.color,
- status=project.status,
- target_count=project.target_count,
- created_at=project.created_at,
- archive_count=archive_count,
- queue_count=queue_count,
- progress_percent=progress_percent,
- archives=archive_previews,
- )
- )
- return response
- @router.post("/", response_model=ProjectResponse)
- async def create_project(
- data: ProjectCreate,
- db: AsyncSession = Depends(get_db),
- ):
- """Create a new project."""
- # Verify parent exists if specified
- parent_name = None
- if data.parent_id:
- parent_result = await db.execute(select(Project).where(Project.id == data.parent_id))
- parent = parent_result.scalar_one_or_none()
- if not parent:
- raise HTTPException(status_code=400, detail="Parent project not found")
- parent_name = parent.name
- project = Project(
- name=data.name,
- description=data.description,
- color=data.color,
- target_count=data.target_count,
- notes=data.notes,
- tags=data.tags,
- due_date=data.due_date,
- priority=data.priority,
- budget=data.budget,
- parent_id=data.parent_id,
- )
- db.add(project)
- await db.flush()
- await db.refresh(project)
- stats = await compute_project_stats(db, project.id, project.target_count)
- return ProjectResponse(
- id=project.id,
- name=project.name,
- description=project.description,
- color=project.color,
- status=project.status,
- target_count=project.target_count,
- notes=project.notes,
- attachments=project.attachments,
- tags=project.tags,
- due_date=project.due_date,
- priority=project.priority,
- budget=project.budget,
- is_template=project.is_template,
- template_source_id=project.template_source_id,
- parent_id=project.parent_id,
- parent_name=parent_name,
- children=[],
- created_at=project.created_at,
- updated_at=project.updated_at,
- stats=stats,
- )
- # ============ Phase 8: Template Endpoints (Static routes BEFORE dynamic {project_id}) ============
- @router.get("/templates", response_model=list[ProjectListResponse])
- async def list_templates(
- db: AsyncSession = Depends(get_db),
- ):
- """List all project templates."""
- result = await db.execute(select(Project).where(Project.is_template.is_(True)).order_by(Project.name))
- templates = result.scalars().all()
- response = []
- for project in templates:
- # Get archive count
- archive_count_result = await db.execute(
- select(func.count(PrintArchive.id)).where(PrintArchive.project_id == project.id)
- )
- archive_count = archive_count_result.scalar() or 0
- response.append(
- ProjectListResponse(
- id=project.id,
- name=project.name,
- description=project.description,
- color=project.color,
- status=project.status,
- target_count=project.target_count,
- created_at=project.created_at,
- archive_count=archive_count,
- queue_count=0,
- progress_percent=None,
- archives=[],
- )
- )
- return response
- @router.post("/from-template/{template_id}", response_model=ProjectResponse)
- async def create_project_from_template(
- template_id: int,
- name: str = None,
- db: AsyncSession = Depends(get_db),
- ):
- """Create a new project from a template."""
- result = await db.execute(select(Project).where(Project.id == template_id))
- template = result.scalar_one_or_none()
- if not template:
- raise HTTPException(status_code=404, detail="Template not found")
- if not template.is_template:
- raise HTTPException(status_code=400, detail="Project is not a template")
- # Create new project
- project = Project(
- name=name or template.name.replace(" (Template)", ""),
- description=template.description,
- color=template.color,
- target_count=template.target_count,
- notes=template.notes,
- tags=template.tags,
- priority=template.priority,
- budget=template.budget,
- is_template=False,
- template_source_id=template.id,
- )
- db.add(project)
- await db.flush()
- # Copy BOM items
- bom_result = await db.execute(select(ProjectBOMItem).where(ProjectBOMItem.project_id == template_id))
- bom_items = bom_result.scalars().all()
- for item in bom_items:
- new_item = ProjectBOMItem(
- project_id=project.id,
- name=item.name,
- quantity_needed=item.quantity_needed,
- quantity_acquired=0,
- unit_price=item.unit_price,
- sourcing_url=item.sourcing_url,
- stl_filename=item.stl_filename,
- remarks=item.remarks,
- sort_order=item.sort_order,
- )
- db.add(new_item)
- await db.flush()
- await db.refresh(project)
- stats = await compute_project_stats(db, project.id, project.target_count)
- return ProjectResponse(
- id=project.id,
- name=project.name,
- description=project.description,
- color=project.color,
- status=project.status,
- target_count=project.target_count,
- notes=project.notes,
- attachments=project.attachments,
- tags=project.tags,
- due_date=project.due_date,
- priority=project.priority,
- budget=project.budget,
- is_template=project.is_template,
- template_source_id=project.template_source_id,
- parent_id=project.parent_id,
- parent_name=None,
- children=[],
- created_at=project.created_at,
- updated_at=project.updated_at,
- stats=stats,
- )
- # ============ Dynamic {project_id} Routes ============
- async def get_child_previews(db: AsyncSession, parent_id: int) -> list[ProjectChildPreview]:
- """Get preview info for child projects."""
- result = await db.execute(select(Project).where(Project.parent_id == parent_id).order_by(Project.name))
- children = result.scalars().all()
- previews = []
- for child in children:
- # Get completed count for progress
- completed_result = await db.execute(
- select(func.count(PrintArchive.id)).where(
- PrintArchive.project_id == child.id,
- PrintArchive.status == "completed",
- )
- )
- completed_count = completed_result.scalar() or 0
- progress = None
- if child.target_count and child.target_count > 0:
- progress = round((completed_count / child.target_count) * 100, 1)
- previews.append(
- ProjectChildPreview(
- id=child.id,
- name=child.name,
- color=child.color,
- status=child.status,
- progress_percent=progress,
- )
- )
- return previews
- @router.get("/{project_id}", response_model=ProjectResponse)
- async def get_project(
- project_id: int,
- db: AsyncSession = Depends(get_db),
- ):
- """Get a project by ID with detailed stats."""
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- # Get parent name
- parent_name = None
- if project.parent_id:
- parent_result = await db.execute(select(Project.name).where(Project.id == project.parent_id))
- parent_name = parent_result.scalar()
- # Get children
- children = await get_child_previews(db, project.id)
- stats = await compute_project_stats(db, project.id, project.target_count)
- return ProjectResponse(
- id=project.id,
- name=project.name,
- description=project.description,
- color=project.color,
- status=project.status,
- target_count=project.target_count,
- notes=project.notes,
- attachments=project.attachments,
- tags=project.tags,
- due_date=project.due_date,
- priority=project.priority,
- budget=project.budget,
- is_template=project.is_template,
- template_source_id=project.template_source_id,
- parent_id=project.parent_id,
- parent_name=parent_name,
- children=children,
- created_at=project.created_at,
- updated_at=project.updated_at,
- stats=stats,
- )
- @router.patch("/{project_id}", response_model=ProjectResponse)
- async def update_project(
- project_id: int,
- data: ProjectUpdate,
- db: AsyncSession = Depends(get_db),
- ):
- """Update a project."""
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- # Update fields if provided
- if data.name is not None:
- project.name = data.name
- if data.description is not None:
- project.description = data.description
- if data.color is not None:
- project.color = data.color
- if data.status is not None:
- if data.status not in ["active", "completed", "archived"]:
- raise HTTPException(status_code=400, detail="Invalid status")
- project.status = data.status
- if data.target_count is not None:
- project.target_count = data.target_count
- if data.notes is not None:
- project.notes = data.notes
- if data.tags is not None:
- project.tags = data.tags
- if data.due_date is not None:
- project.due_date = data.due_date
- if data.priority is not None:
- if data.priority not in ["low", "normal", "high", "urgent"]:
- raise HTTPException(status_code=400, detail="Invalid priority")
- project.priority = data.priority
- if data.budget is not None:
- project.budget = data.budget
- if data.parent_id is not None:
- # Verify parent exists and prevent circular reference
- if data.parent_id == project_id:
- raise HTTPException(status_code=400, detail="Project cannot be its own parent")
- if data.parent_id != 0: # 0 means remove parent
- parent_result = await db.execute(select(Project).where(Project.id == data.parent_id))
- if not parent_result.scalar_one_or_none():
- raise HTTPException(status_code=400, detail="Parent project not found")
- project.parent_id = data.parent_id
- else:
- project.parent_id = None
- await db.flush()
- await db.refresh(project)
- # Get parent name
- parent_name = None
- if project.parent_id:
- parent_result = await db.execute(select(Project.name).where(Project.id == project.parent_id))
- parent_name = parent_result.scalar()
- # Get children
- children = await get_child_previews(db, project.id)
- stats = await compute_project_stats(db, project.id, project.target_count)
- return ProjectResponse(
- id=project.id,
- name=project.name,
- description=project.description,
- color=project.color,
- status=project.status,
- target_count=project.target_count,
- notes=project.notes,
- attachments=project.attachments,
- tags=project.tags,
- due_date=project.due_date,
- priority=project.priority,
- budget=project.budget,
- is_template=project.is_template,
- template_source_id=project.template_source_id,
- parent_id=project.parent_id,
- parent_name=parent_name,
- children=children,
- created_at=project.created_at,
- updated_at=project.updated_at,
- stats=stats,
- )
- @router.delete("/{project_id}")
- async def delete_project(
- project_id: int,
- db: AsyncSession = Depends(get_db),
- ):
- """Delete a project. Archives and queue items will have project_id set to NULL."""
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- await db.delete(project)
- return {"message": "Project deleted"}
- @router.get("/{project_id}/archives")
- async def list_project_archives(
- project_id: int,
- limit: int = 100,
- offset: int = 0,
- db: AsyncSession = Depends(get_db),
- ):
- """List archives in a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- if not result.scalar_one_or_none():
- raise HTTPException(status_code=404, detail="Project not found")
- # Get archives with project relationship eagerly loaded
- query = (
- select(PrintArchive)
- .options(selectinload(PrintArchive.project))
- .where(PrintArchive.project_id == project_id)
- .order_by(PrintArchive.created_at.desc())
- .limit(limit)
- .offset(offset)
- )
- result = await db.execute(query)
- archives = result.scalars().all()
- # Import the response converter from archives module
- from backend.app.api.routes.archives import archive_to_response
- return [archive_to_response(a) for a in archives]
- @router.get("/{project_id}/queue")
- async def list_project_queue(
- project_id: int,
- db: AsyncSession = Depends(get_db),
- ):
- """List queue items in a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- if not result.scalar_one_or_none():
- raise HTTPException(status_code=404, detail="Project not found")
- # Get queue items
- query = select(PrintQueueItem).where(PrintQueueItem.project_id == project_id).order_by(PrintQueueItem.position)
- result = await db.execute(query)
- items = result.scalars().all()
- return items
- @router.post("/{project_id}/add-archives")
- async def add_archives_to_project(
- project_id: int,
- data: BatchAddArchives,
- db: AsyncSession = Depends(get_db),
- ):
- """Batch add archives to a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- if not result.scalar_one_or_none():
- raise HTTPException(status_code=404, detail="Project not found")
- # Update archives
- updated = 0
- for archive_id in data.archive_ids:
- result = await db.execute(select(PrintArchive).where(PrintArchive.id == archive_id))
- archive = result.scalar_one_or_none()
- if archive:
- archive.project_id = project_id
- updated += 1
- return {"message": f"Added {updated} archives to project"}
- @router.post("/{project_id}/add-queue")
- async def add_queue_items_to_project(
- project_id: int,
- data: BatchAddQueueItems,
- db: AsyncSession = Depends(get_db),
- ):
- """Batch add queue items to a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- if not result.scalar_one_or_none():
- raise HTTPException(status_code=404, detail="Project not found")
- # Update queue items
- updated = 0
- for item_id in data.queue_item_ids:
- result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id == item_id))
- item = result.scalar_one_or_none()
- if item:
- item.project_id = project_id
- updated += 1
- return {"message": f"Added {updated} queue items to project"}
- @router.post("/{project_id}/remove-archives")
- async def remove_archives_from_project(
- project_id: int,
- data: BatchAddArchives,
- db: AsyncSession = Depends(get_db),
- ):
- """Remove archives from a project (sets project_id to NULL)."""
- updated = 0
- for archive_id in data.archive_ids:
- result = await db.execute(
- select(PrintArchive).where(
- PrintArchive.id == archive_id,
- PrintArchive.project_id == project_id,
- )
- )
- archive = result.scalar_one_or_none()
- if archive:
- archive.project_id = None
- updated += 1
- return {"message": f"Removed {updated} archives from project"}
- def get_project_attachments_dir(project_id: int) -> Path:
- """Get the attachments directory for a project."""
- base_dir = Path(settings.archive_dir)
- return base_dir / "projects" / str(project_id) / "attachments"
- # Allowed file extensions for attachments
- ALLOWED_ATTACHMENT_EXTENSIONS = {
- # Images
- ".jpg",
- ".jpeg",
- ".png",
- ".gif",
- ".webp",
- ".svg",
- ".bmp",
- ".ico",
- # Documents
- ".pdf",
- ".doc",
- ".docx",
- ".xls",
- ".xlsx",
- ".ppt",
- ".pptx",
- ".odt",
- ".ods",
- ".odp",
- ".txt",
- ".rtf",
- ".csv",
- ".md",
- # 3D/CAD files
- ".stl",
- ".obj",
- ".3mf",
- ".step",
- ".stp",
- ".iges",
- ".igs",
- ".f3d",
- ".scad",
- # Archives
- ".zip",
- ".rar",
- ".7z",
- ".tar",
- ".gz",
- # Code/scripts (for Klipper macros, scripts, etc.)
- ".py",
- ".sh",
- ".cfg",
- ".conf",
- ".gcode",
- ".ini",
- # Other common formats
- ".json",
- ".xml",
- ".yaml",
- ".yml",
- }
- @router.post("/{project_id}/attachments")
- async def upload_attachment(
- project_id: int,
- file: UploadFile = File(...),
- db: AsyncSession = Depends(get_db),
- ):
- """Upload an attachment to a project."""
- logger.info(f"=== UPLOAD START: {file.filename} for project {project_id} ===")
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- # Validate file extension
- original_name = file.filename or "unknown"
- ext = os.path.splitext(original_name)[1].lower()
- if ext not in ALLOWED_ATTACHMENT_EXTENSIONS:
- raise HTTPException(
- status_code=400,
- detail=f"File type '{ext}' not supported. Allowed: images, PDFs, documents, STL, 3MF, archives.",
- )
- # Create attachments directory
- attachments_dir = get_project_attachments_dir(project_id)
- attachments_dir.mkdir(parents=True, exist_ok=True)
- # Generate unique filename
- unique_filename = f"{uuid.uuid4().hex}{ext}"
- file_path = attachments_dir / unique_filename
- # Save file
- try:
- with open(file_path, "wb") as f:
- content = await file.read()
- f.write(content)
- logger.info(f"=== FILE SAVED: {file_path}, size: {len(content)} ===")
- except Exception as e:
- logger.error(f"Failed to save attachment: {e}")
- raise HTTPException(status_code=500, detail="Failed to save attachment")
- # Update project attachments JSON
- attachments = list(project.attachments or [])
- new_attachment = {
- "filename": unique_filename,
- "original_name": original_name,
- "size": len(content),
- "uploaded_at": datetime.now().isoformat(),
- }
- attachments.append(new_attachment)
- # Simple ORM update
- project.attachments = attachments
- db.add(project) # Explicitly add to session
- logger.info(f"=== BEFORE COMMIT: {len(attachments)} attachments ===")
- await db.flush()
- await db.commit()
- logger.info("=== AFTER COMMIT ===")
- # Verify by re-querying
- result = await db.execute(select(Project).where(Project.id == project_id))
- fresh_project = result.scalar_one()
- logger.info(f"=== VERIFIED: {len(fresh_project.attachments or [])} attachments ===")
- return {
- "status": "success",
- "filename": unique_filename,
- "original_name": original_name,
- "attachments": fresh_project.attachments,
- }
- @router.get("/{project_id}/attachments/{filename}")
- async def download_attachment(
- project_id: int,
- filename: str,
- db: AsyncSession = Depends(get_db),
- ):
- """Download an attachment from a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- # Verify attachment exists in project
- attachments = project.attachments or []
- attachment = next((a for a in attachments if a.get("filename") == filename), None)
- if not attachment:
- raise HTTPException(status_code=404, detail="Attachment not found")
- # Check file exists
- file_path = get_project_attachments_dir(project_id) / filename
- if not file_path.exists():
- raise HTTPException(status_code=404, detail="Attachment file not found")
- return FileResponse(
- file_path,
- filename=attachment.get("original_name", filename),
- media_type="application/octet-stream",
- )
- @router.delete("/{project_id}/attachments/{filename}")
- async def delete_attachment(
- project_id: int,
- filename: str,
- db: AsyncSession = Depends(get_db),
- ):
- """Delete an attachment from a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- # Find and remove attachment from list
- attachments = project.attachments or []
- attachment = next((a for a in attachments if a.get("filename") == filename), None)
- if not attachment:
- raise HTTPException(status_code=404, detail="Attachment not found")
- # Remove from list
- attachments = [a for a in attachments if a.get("filename") != filename]
- project.attachments = attachments if attachments else None
- # Delete file
- file_path = get_project_attachments_dir(project_id) / filename
- if file_path.exists():
- try:
- os.remove(file_path)
- except Exception as e:
- logger.warning(f"Failed to delete attachment file: {e}")
- await db.flush()
- await db.refresh(project)
- return {
- "status": "success",
- "message": "Attachment deleted",
- "attachments": project.attachments,
- }
- # ============ Phase 7: BOM Endpoints ============
- @router.get("/{project_id}/bom", response_model=list[BOMItemResponse])
- async def list_bom_items(
- project_id: int,
- db: AsyncSession = Depends(get_db),
- ):
- """List all BOM items for a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- if not result.scalar_one_or_none():
- raise HTTPException(status_code=404, detail="Project not found")
- # Get BOM items
- result = await db.execute(
- select(ProjectBOMItem)
- .where(ProjectBOMItem.project_id == project_id)
- .order_by(ProjectBOMItem.sort_order, ProjectBOMItem.id)
- )
- items = result.scalars().all()
- response = []
- for item in items:
- # Get archive name if linked
- archive_name = None
- if item.archive_id:
- archive_result = await db.execute(select(PrintArchive.print_name).where(PrintArchive.id == item.archive_id))
- archive_name = archive_result.scalar()
- response.append(
- BOMItemResponse(
- id=item.id,
- project_id=item.project_id,
- name=item.name,
- quantity_needed=item.quantity_needed,
- quantity_acquired=item.quantity_acquired,
- unit_price=item.unit_price,
- sourcing_url=item.sourcing_url,
- archive_id=item.archive_id,
- archive_name=archive_name,
- stl_filename=item.stl_filename,
- remarks=item.remarks,
- sort_order=item.sort_order,
- is_complete=item.quantity_acquired >= item.quantity_needed,
- created_at=item.created_at,
- updated_at=item.updated_at,
- )
- )
- return response
- @router.post("/{project_id}/bom", response_model=BOMItemResponse)
- async def create_bom_item(
- project_id: int,
- data: BOMItemCreate,
- db: AsyncSession = Depends(get_db),
- ):
- """Add a BOM item to a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- if not result.scalar_one_or_none():
- raise HTTPException(status_code=404, detail="Project not found")
- # Get max sort order
- max_order_result = await db.execute(
- select(func.max(ProjectBOMItem.sort_order)).where(ProjectBOMItem.project_id == project_id)
- )
- max_order = max_order_result.scalar() or 0
- item = ProjectBOMItem(
- project_id=project_id,
- name=data.name,
- quantity_needed=data.quantity_needed,
- unit_price=data.unit_price,
- sourcing_url=data.sourcing_url,
- archive_id=data.archive_id,
- stl_filename=data.stl_filename,
- remarks=data.remarks,
- sort_order=max_order + 1,
- )
- db.add(item)
- await db.flush()
- await db.refresh(item)
- # Get archive name if linked
- archive_name = None
- if item.archive_id:
- archive_result = await db.execute(select(PrintArchive.print_name).where(PrintArchive.id == item.archive_id))
- archive_name = archive_result.scalar()
- return BOMItemResponse(
- id=item.id,
- project_id=item.project_id,
- name=item.name,
- quantity_needed=item.quantity_needed,
- quantity_acquired=item.quantity_acquired,
- unit_price=item.unit_price,
- sourcing_url=item.sourcing_url,
- archive_id=item.archive_id,
- archive_name=archive_name,
- stl_filename=item.stl_filename,
- remarks=item.remarks,
- sort_order=item.sort_order,
- is_complete=item.quantity_acquired >= item.quantity_needed,
- created_at=item.created_at,
- updated_at=item.updated_at,
- )
- @router.patch("/{project_id}/bom/{item_id}", response_model=BOMItemResponse)
- async def update_bom_item(
- project_id: int,
- item_id: int,
- data: BOMItemUpdate,
- db: AsyncSession = Depends(get_db),
- ):
- """Update a BOM item."""
- result = await db.execute(
- select(ProjectBOMItem).where(
- ProjectBOMItem.id == item_id,
- ProjectBOMItem.project_id == project_id,
- )
- )
- item = result.scalar_one_or_none()
- if not item:
- raise HTTPException(status_code=404, detail="BOM item not found")
- if data.name is not None:
- item.name = data.name
- if data.quantity_needed is not None:
- item.quantity_needed = data.quantity_needed
- if data.quantity_acquired is not None:
- item.quantity_acquired = data.quantity_acquired
- if data.unit_price is not None:
- item.unit_price = data.unit_price if data.unit_price != 0 else None
- if data.sourcing_url is not None:
- item.sourcing_url = data.sourcing_url if data.sourcing_url else None
- if data.archive_id is not None:
- item.archive_id = data.archive_id if data.archive_id != 0 else None
- if data.stl_filename is not None:
- item.stl_filename = data.stl_filename if data.stl_filename else None
- if data.remarks is not None:
- item.remarks = data.remarks if data.remarks else None
- await db.flush()
- await db.refresh(item)
- # Get archive name if linked
- archive_name = None
- if item.archive_id:
- archive_result = await db.execute(select(PrintArchive.print_name).where(PrintArchive.id == item.archive_id))
- archive_name = archive_result.scalar()
- return BOMItemResponse(
- id=item.id,
- project_id=item.project_id,
- name=item.name,
- quantity_needed=item.quantity_needed,
- quantity_acquired=item.quantity_acquired,
- unit_price=item.unit_price,
- sourcing_url=item.sourcing_url,
- archive_id=item.archive_id,
- archive_name=archive_name,
- stl_filename=item.stl_filename,
- remarks=item.remarks,
- sort_order=item.sort_order,
- is_complete=item.quantity_acquired >= item.quantity_needed,
- created_at=item.created_at,
- updated_at=item.updated_at,
- )
- @router.delete("/{project_id}/bom/{item_id}")
- async def delete_bom_item(
- project_id: int,
- item_id: int,
- db: AsyncSession = Depends(get_db),
- ):
- """Delete a BOM item."""
- result = await db.execute(
- select(ProjectBOMItem).where(
- ProjectBOMItem.id == item_id,
- ProjectBOMItem.project_id == project_id,
- )
- )
- item = result.scalar_one_or_none()
- if not item:
- raise HTTPException(status_code=404, detail="BOM item not found")
- await db.delete(item)
- return {"status": "success", "message": "BOM item deleted"}
- @router.post("/{project_id}/create-template", response_model=ProjectResponse)
- async def create_template_from_project(
- project_id: int,
- db: AsyncSession = Depends(get_db),
- ):
- """Create a template from an existing project."""
- result = await db.execute(select(Project).where(Project.id == project_id))
- source = result.scalar_one_or_none()
- if not source:
- raise HTTPException(status_code=404, detail="Project not found")
- # Create template
- template = Project(
- name=f"{source.name} (Template)",
- description=source.description,
- color=source.color,
- target_count=source.target_count,
- notes=source.notes,
- tags=source.tags,
- priority=source.priority,
- budget=source.budget,
- is_template=True,
- template_source_id=source.id,
- )
- db.add(template)
- await db.flush()
- # Copy BOM items
- bom_result = await db.execute(select(ProjectBOMItem).where(ProjectBOMItem.project_id == project_id))
- bom_items = bom_result.scalars().all()
- for item in bom_items:
- new_item = ProjectBOMItem(
- project_id=template.id,
- name=item.name,
- quantity_needed=item.quantity_needed,
- quantity_acquired=0,
- unit_price=item.unit_price,
- sourcing_url=item.sourcing_url,
- stl_filename=item.stl_filename,
- remarks=item.remarks,
- sort_order=item.sort_order,
- )
- db.add(new_item)
- await db.flush()
- await db.refresh(template)
- stats = await compute_project_stats(db, template.id, template.target_count)
- return ProjectResponse(
- id=template.id,
- name=template.name,
- description=template.description,
- color=template.color,
- status=template.status,
- target_count=template.target_count,
- notes=template.notes,
- attachments=template.attachments,
- tags=template.tags,
- due_date=template.due_date,
- priority=template.priority,
- budget=template.budget,
- is_template=template.is_template,
- template_source_id=template.template_source_id,
- parent_id=template.parent_id,
- parent_name=None,
- children=[],
- created_at=template.created_at,
- updated_at=template.updated_at,
- stats=stats,
- )
- # ============ Phase 9: Timeline Endpoint ============
- @router.get("/{project_id}/timeline", response_model=list[TimelineEvent])
- async def get_project_timeline(
- project_id: int,
- limit: int = 50,
- db: AsyncSession = Depends(get_db),
- ):
- """Get timeline of events for a project."""
- # Verify project exists
- result = await db.execute(select(Project).where(Project.id == project_id))
- project = result.scalar_one_or_none()
- if not project:
- raise HTTPException(status_code=404, detail="Project not found")
- events = []
- # Project creation event
- events.append(
- TimelineEvent(
- event_type="project_created",
- timestamp=project.created_at,
- title="Project created",
- description=f"Project '{project.name}' was created",
- )
- )
- # Get archives and add events
- archives_result = await db.execute(
- select(PrintArchive)
- .where(PrintArchive.project_id == project_id)
- .order_by(PrintArchive.created_at.desc())
- .limit(limit)
- )
- archives = archives_result.scalars().all()
- for archive in archives:
- if archive.status == "completed":
- events.append(
- TimelineEvent(
- event_type="print_completed",
- timestamp=archive.completed_at or archive.created_at,
- title="Print completed",
- description=archive.print_name,
- metadata={
- "archive_id": archive.id,
- "print_time_hours": round((archive.print_time_seconds or 0) / 3600, 2),
- "filament_grams": round(archive.filament_used_grams or 0, 1),
- },
- )
- )
- elif archive.status == "failed":
- events.append(
- TimelineEvent(
- event_type="print_failed",
- timestamp=archive.completed_at or archive.created_at,
- title="Print failed",
- description=archive.print_name,
- metadata={"archive_id": archive.id},
- )
- )
- # Get queue items
- queue_result = await db.execute(
- select(PrintQueueItem)
- .where(PrintQueueItem.project_id == project_id)
- .order_by(PrintQueueItem.created_at.desc())
- .limit(limit)
- )
- queue_items = queue_result.scalars().all()
- for item in queue_items:
- if item.status == "printing":
- events.append(
- TimelineEvent(
- event_type="print_started",
- timestamp=item.started_at or item.created_at,
- title="Print started",
- description=item.print_name,
- metadata={"queue_item_id": item.id},
- )
- )
- elif item.status == "pending":
- events.append(
- TimelineEvent(
- event_type="queued",
- timestamp=item.created_at,
- title="Added to queue",
- description=item.print_name,
- metadata={"queue_item_id": item.id},
- )
- )
- # Sort by timestamp descending
- events.sort(key=lambda e: e.timestamp, reverse=True)
- return events[:limit]
|