项目基础模块代码

app/model/crud/project_info_crud.py

@@ -0,0 +1,42 @@
+from sqlalchemy.orm import Session
+from sqlalchemy import desc
+
+from app.model.bussiness_model import ProjectInfo
+from app.model.schemas.project_info_schemas import ProjectInfoOut
+from app.model.schemas.project_info_schemas import ProjectInfoPager
+from app.db.page_util import get_pager
+
+
+def get_project_pager(info: ProjectInfoPager, session: Session):
+    """分页查询项目信息"""
+    query = session.query(ProjectInfo).order_by(desc(ProjectInfo.id))
+    filters = []
+    if info.project_name is not None:
+        filters.append(ProjectInfo.project_name.ilike(f"%{info.project_name}%"))
+    if len(filters) > 0:
+        query.filter(*filters)
+    pager = get_pager(query, info.pagerNum, info.pagerSize)
+    pager.data = [ProjectInfoOut.from_orm(info).dict() for info in pager.data]
+    return pager
+
+
+def get_project_by_id(id: str, session: Session):
+    info = session.query(ProjectInfo).filter_by(id=id).first()
+    info_out = ProjectInfoOut.from_orm(info)
+    return info_out
+
+def add_project(info: ProjectInfo, session: Session):
+    """新建项目,并在对应文件夹下面创建文件夹"""
+    session.add(info)
+    session.commit()
+    return info
+
+
+def check_project_name(project_name: str, session: Session):
+    """检验是否存在重名的项目名称"""
+    count = session.query(ProjectInfo).filter(ProjectInfo.project_name == project_name).count()
+    if count > 0:
+        return True
+    else:
+        return False
+