123456789101112131415161718192021222324252627 |
- import time
- from typing import List
- from app import models, schemas
- from sqlalchemy.orm import Session
- def create_data_management(db: Session, item: schemas.DataManagementCreate):
- create_time: int = int(time.time())
- db_item = models.DataManagement(**item.dict(), **{
- 'create_time': create_time,
- })
- db.add(db_item)
- db.commit()
- db.refresh(db_item)
- return db_item
- def get_data_managements(db: Session, user_id: str, project_id: str):
- res: List[models.DataManagement] = db.query(models.DataManagement).filter(models.DataManagement.project_id == project_id,models.DataManagement.user_id == user_id).all()
- return res
- def delete_data_management(db: Session, d_id: int):
- dm_item = db.query(models.DataManagement).filter(models.DataManagement.id == d_id).first()
- if not dm_item:
- raise Exception("Data management not found")
- db.delete(dm_item)
- db.commit()
- db.flush()
- return dm_item
|