data_management.py 950 B

123456789101112131415161718192021222324252627
  1. import time
  2. from typing import List
  3. from app import models, schemas
  4. from sqlalchemy.orm import Session
  5. def create_data_management(db: Session, item: schemas.DataManagementCreate):
  6. create_time: int = int(time.time())
  7. db_item = models.DataManagement(**item.dict(), **{
  8. 'create_time': create_time,
  9. })
  10. db.add(db_item)
  11. db.commit()
  12. db.refresh(db_item)
  13. return db_item
  14. def get_data_managements(db: Session, user_id: str, project_id: str):
  15. res: List[models.DataManagement] = db.query(models.DataManagement).filter(models.DataManagement.project_id == project_id,models.DataManagement.user_id == user_id).all()
  16. return res
  17. def delete_data_management(db: Session, d_id: int):
  18. dm_item = db.query(models.DataManagement).filter(models.DataManagement.id == d_id).first()
  19. if not dm_item:
  20. raise Exception("Data management not found")
  21. db.delete(dm_item)
  22. db.commit()
  23. db.flush()
  24. return dm_item