wehub-resource-sync
151 lines
4.3 KiB
Python
151 lines
4.3 KiB
Python
from fastapi_sqlalchemy import db
|
|
from fastapi import HTTPException, Depends
|
|
from fastapi_jwt_auth import AuthJWT
|
|
from pydantic import BaseModel
|
|
|
|
from superagi.models.project import Project
|
|
from superagi.models.organisation import Organisation
|
|
from fastapi import APIRouter
|
|
from superagi.helper.auth import check_auth
|
|
from superagi.lib.logger import logger
|
|
# from superagi.types.db import ProjectIn, ProjectOut
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class ProjectOut(BaseModel):
|
|
id: int
|
|
name: str
|
|
organisation_id: int
|
|
description: str
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
class ProjectIn(BaseModel):
|
|
name: str
|
|
organisation_id: int
|
|
description: str
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
# CRUD Operations
|
|
@router.post("/add", response_model=ProjectOut, status_code=201)
|
|
def create_project(project: ProjectIn,
|
|
Authorize: AuthJWT = Depends(check_auth)):
|
|
"""
|
|
Create a new project.
|
|
|
|
Args:
|
|
project (Project): Project data.
|
|
|
|
Returns:
|
|
dict: Dictionary containing the created project.
|
|
|
|
Raises:
|
|
HTTPException (status_code=404): If the organization with the specified ID is not found.
|
|
|
|
"""
|
|
|
|
logger.info("Organisation_id : ", project.organisation_id)
|
|
organisation = db.session.query(Organisation).get(project.organisation_id)
|
|
|
|
if not organisation:
|
|
raise HTTPException(status_code=404, detail="Organisation not found")
|
|
|
|
project = Project(
|
|
name=project.name,
|
|
organisation_id=organisation.id,
|
|
description=project.description
|
|
)
|
|
|
|
db.session.add(project)
|
|
db.session.commit()
|
|
|
|
return project
|
|
|
|
|
|
@router.get("/get/{project_id}", response_model=ProjectOut)
|
|
def get_project(project_id: int, Authorize: AuthJWT = Depends(check_auth)):
|
|
"""
|
|
Get project details by project_id.
|
|
|
|
Args:
|
|
project_id (int): ID of the project.
|
|
|
|
Returns:
|
|
dict: Dictionary containing the project details.
|
|
|
|
Raises:
|
|
HTTPException (status_code=404): If the project with the specified ID is not found.
|
|
|
|
"""
|
|
|
|
db_project = db.session.query(Project).filter(Project.id == project_id).first()
|
|
if not db_project:
|
|
raise HTTPException(status_code=404, detail="project not found")
|
|
return db_project
|
|
|
|
|
|
@router.put("/update/{project_id}", response_model=ProjectOut)
|
|
def update_project(project_id: int, project: ProjectIn,
|
|
Authorize: AuthJWT = Depends(check_auth)):
|
|
"""
|
|
Update a project detail by project_id.
|
|
|
|
Args:
|
|
project_id (int): ID of the project.
|
|
project (Project): Updated project data.
|
|
|
|
Returns:
|
|
dict: Dictionary containing the updated project details.
|
|
|
|
Raises:
|
|
HTTPException (status_code=404): If the project with the specified ID is not found.
|
|
HTTPException (status_code=404): If the organization with the specified ID is not found.
|
|
|
|
"""
|
|
|
|
db_project = db.session.query(Project).get(project_id)
|
|
if not db_project:
|
|
raise HTTPException(status_code=404, detail="Project not found")
|
|
|
|
if project.organisation_id:
|
|
organisation = db.session.query(Organisation).get(project.organisation_id)
|
|
if not organisation:
|
|
raise HTTPException(status_code=404, detail="Organisation not found")
|
|
db_project.organisation_id = organisation.id
|
|
db_project.name = project.name
|
|
db_project.description = project.description
|
|
db.session.add(db_project)
|
|
db.session.commit()
|
|
return db_project
|
|
|
|
|
|
@router.get("/get/organisation/{organisation_id}")
|
|
def get_projects_organisation(organisation_id: int,
|
|
Authorize: AuthJWT = Depends(check_auth)):
|
|
"""
|
|
Get all projects by organisation_id and create default if no project.
|
|
|
|
Args:
|
|
organisation_id (int): ID of the organisation.
|
|
|
|
Returns:
|
|
List[Project]: List of projects belonging to the organisation.
|
|
|
|
Raises:
|
|
HTTPException (status_code=404): If the organization with the specified ID is not found.
|
|
|
|
"""
|
|
|
|
Project.find_or_create_default_project(db.session, organisation_id)
|
|
projects = db.session.query(Project).filter(Project.organisation_id == organisation_id).all()
|
|
if len(projects) <= 0:
|
|
default_project = Project.find_or_create_default_project(db.session, organisation_id)
|
|
projects.append(default_project)
|
|
|
|
return projects
|