报表

api/api_v1/endpoints/interview.py

@@ -848,6 +848,28 @@ async def add_mode(
     return schemas.Msg(code=200, msg='ok', data='')
 
 
+@router.post("/add_table")
+async def add_table(
+        request: Request,
+        data_in: schemas.InsertTables,
+        db: AsyncIOMotorDatabase = Depends(get_database),
+) -> schemas.Msg:
+    """添加分组报表"""
+    await crud.api_interview_tables.insert_modes(db, data_in)
+    return schemas.Msg(code=200, msg='ok', data='')
+
+
+@router.post("/update_table")
+async def update_table(
+        request: Request,
+        data_in: schemas.UpdateTables,
+        db: AsyncIOMotorDatabase = Depends(get_database),
+) -> schemas.Msg:
+    """修改表报"""
+    await crud.api_interview_tables.insert_modes(db, data_in)
+    return schemas.Msg(code=200, msg='ok', data='')
+
+
 @router.post("/find_mode")
 async def find_mode(
         request: Request,

crud/__init__.py

@@ -24,6 +24,7 @@ from .crud_event_list import event_list
 from .crud_jobs import jobs
 from .crud_interview_remark import api_interview_remark
 from .crud_interview_modes import api_interview_modes
+from .crud_interview_tables import api_interview_tables
 from .crud_email_record import email_record
 from .crud_operate_log import operate_log
 from .crud_interview_record import interview_record

crud/crud_interview_tables.py

@@ -0,0 +1,28 @@
+from motor.motor_asyncio import AsyncIOMotorDatabase
+import schemas
+from crud.base import CRUDBase
+
+__all__ = 'api_interview_tables',
+
+
+class ApiInterviewModes(CRUDBase):
+
+    # 获取分组所有报表
+    async def get_interview_modes(self, db: AsyncIOMotorDatabase, data_in: schemas.FindTables):
+        where = data_in.where
+        where.update({'mode_id': data_in.mode_id})
+        return await self.find_many(db, where, {'_id': 0})
+
+    # 插入一条全新的分组
+    async def insert_tables(self, db: AsyncIOMotorDatabase, data_in: schemas.InsertTables):
+        data = {k: v for k, v in data_in.dict().items() if k != 'table_data'}
+        insert_data = data_in.table_data
+        insert_data.update(data)
+        return await self.insert_one(db, insert_data)
+
+    # 更新一条报表信息
+    async def update_tables(self, db: AsyncIOMotorDatabase, data_in: schemas.UpdateTables):
+        return await self.update_one(db, {'table_id': data_in.table_id}, {'$set': data_in.update_data})
+
+
+api_interview_modes = ApiInterviewModes('interview_tables')

schemas/__init__.py

@@ -32,6 +32,7 @@ from .interview_remark import *
 from .email_record import *
 from .operate_log import *
 from .interview_modes import *
+from .interview_tables import *
 from .interview_record import *
 from .worker import *
 from .owner_info import *

schemas/interview_modes.py

@@ -15,8 +15,7 @@ class InsertModes(BaseModel):
     mode_name: str  # 模版名称
     hr_name: str  # 模板所属hr名字
     hr_uid: str  # 模板所属hr的uid
-    mode_list: list = ['招聘职位', '流程中的申请', '完成招聘职位', '归档的申请', '招聘漏斗', '性别分布', '渠道质量',
+    mode_list: list = []  # 分组包含的基础分析报表
-                       '年龄分布', '招聘网站质量', '职位投录比']  # 分组包含的基础分析报表
     mode_time: datetime = datetime.now()  # 修改时间
 
 

schemas/interview_tables.py

@@ -0,0 +1,30 @@
+import time
+import random
+from datetime import datetime
+
+from pydantic import BaseModel
+
+
+def get_id():
+    return hex(int(time.time() * 10 ** 7) + random.randint(0, 10000))[2:]
+
+
+# 报表分组模板
+class InsertTables(BaseModel):
+    table_id: str = get_id()  # 报表唯一id
+    mode_id: str  # 模版id
+    table_name: str  # 报表名称
+    table_data: dict = {}  # 报表其他数据
+    mode_time: datetime = datetime.now()  # 修改时间
+
+
+# 查询报表
+class FindTables(BaseModel):
+    mode_id: str  # 模板所属hr的uid
+    where: dict = {}  # 其他条件
+
+
+# 更新报表
+class UpdateTables(BaseModel):
+    table_id: str  # 模板唯一id
+    update_data: dict  # 更新的数据