mirror of
https://github.com/Pythagora-io/gpt-pilot.git
synced 2026-01-08 12:53:50 -05:00
5b474ccc1f
This is a complete rewrite of the GPT Pilot core, from the ground up, making the agentic architecture front and center, and also fixing some long-standing problems with the database architecture that weren't feasible to solve without breaking compatibility. As the database structure and config file syntax have changed, we have automatic imports for projects and current configs, see the README.md file for details. This also relicenses the project to FSL-1.1-MIT license.
125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
import re
|
|
from datetime import datetime
|
|
from typing import TYPE_CHECKING, Optional, Union
|
|
from unicodedata import normalize
|
|
from uuid import UUID, uuid4
|
|
|
|
from sqlalchemy import delete, inspect, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship, selectinload
|
|
from sqlalchemy.sql import func
|
|
|
|
from core.db.models import Base
|
|
|
|
if TYPE_CHECKING:
|
|
from core.db.models import Branch
|
|
|
|
|
|
class Project(Base):
|
|
__tablename__ = "projects"
|
|
|
|
# ID and parent FKs
|
|
id: Mapped[UUID] = mapped_column(primary_key=True, default=uuid4)
|
|
|
|
# Attributes
|
|
name: Mapped[str] = mapped_column()
|
|
created_at: Mapped[datetime] = mapped_column(server_default=func.now())
|
|
folder_name: Mapped[str] = mapped_column(
|
|
default=lambda context: Project.get_folder_from_project_name(context.get_current_parameters()["name"])
|
|
)
|
|
|
|
# Relationships
|
|
branches: Mapped[list["Branch"]] = relationship(back_populates="project", cascade="all")
|
|
|
|
@staticmethod
|
|
async def get_by_id(session: "AsyncSession", project_id: Union[str, UUID]) -> Optional["Project"]:
|
|
"""
|
|
Get a project by ID.
|
|
|
|
:param session: The SQLAlchemy session.
|
|
:param project_id: The project ID (as str or UUID value).
|
|
:return: The Project object if found, None otherwise.
|
|
"""
|
|
if not isinstance(project_id, UUID):
|
|
project_id = UUID(project_id)
|
|
|
|
result = await session.execute(select(Project).where(Project.id == project_id))
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_branch(self, name: Optional[str] = None) -> Optional["Branch"]:
|
|
"""
|
|
Get a project branch by name.
|
|
|
|
:param session: The SQLAlchemy session.
|
|
:param branch_name: The name of the branch (default "main").
|
|
:return: The Branch object if found, None otherwise.
|
|
"""
|
|
from core.db.models import Branch
|
|
|
|
session = inspect(self).async_session
|
|
if session is None:
|
|
raise ValueError("Project instance not associated with a DB session.")
|
|
|
|
if name is None:
|
|
name = Branch.DEFAULT
|
|
|
|
result = await session.execute(select(Branch).where(Branch.project_id == self.id, Branch.name == name))
|
|
return result.scalar_one_or_none()
|
|
|
|
@staticmethod
|
|
async def get_all_projects(session: "AsyncSession") -> list["Project"]:
|
|
"""
|
|
Get all projects.
|
|
|
|
This assumes the projects have at least one branch and one state.
|
|
|
|
:param session: The SQLAlchemy session.
|
|
:return: List of Project objects.
|
|
"""
|
|
from core.db.models import Branch, ProjectState
|
|
|
|
latest_state_query = (
|
|
select(ProjectState.branch_id, func.max(ProjectState.id).label("max_id"))
|
|
.group_by(ProjectState.branch_id)
|
|
.subquery()
|
|
)
|
|
|
|
query = (
|
|
select(Project, Branch, ProjectState)
|
|
.join(Branch, Project.branches)
|
|
.join(ProjectState, Branch.states)
|
|
.join(latest_state_query, ProjectState.id == latest_state_query.columns.max_id)
|
|
.options(selectinload(Project.branches), selectinload(Branch.states))
|
|
.order_by(Project.name, Branch.name)
|
|
)
|
|
|
|
results = await session.execute(query)
|
|
return results.scalars().all()
|
|
|
|
@staticmethod
|
|
def get_folder_from_project_name(name: str):
|
|
"""
|
|
Get the folder name from the project name.
|
|
|
|
:param name: Project name.
|
|
:return: Folder name.
|
|
"""
|
|
# replace unicode with accents with base characters (eg "šašavi" → "sasavi")
|
|
name = normalize("NFKD", name).encode("ascii", "ignore").decode("utf-8")
|
|
|
|
# replace spaces/interpunction with a single dash
|
|
return re.sub(r"[^a-zA-Z0-9]+", "-", name).lower().strip("-")
|
|
|
|
@staticmethod
|
|
async def delete_by_id(session: "AsyncSession", project_id: UUID) -> int:
|
|
"""
|
|
Delete a project by ID.
|
|
|
|
:param session: The SQLAlchemy session.
|
|
:param project_id: The project ID
|
|
:return: Number of rows deleted.
|
|
"""
|
|
|
|
result = await session.execute(delete(Project).where(Project.id == project_id))
|
|
return result.rowcount
|