generated from Luis/nextjs-python-web-template
139 lines
4.0 KiB
Python
139 lines
4.0 KiB
Python
from typing import List, Optional
|
|
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy.sql.expression import true
|
|
|
|
from ..errors import ClanError
|
|
from . import schemas, sql_models
|
|
|
|
#########################
|
|
# #
|
|
# service #
|
|
# #
|
|
#########################
|
|
|
|
|
|
def create_service(db: Session, service: schemas.ServiceCreate) -> sql_models.Service:
|
|
db_service = sql_models.Service(**service.dict())
|
|
db.add(db_service)
|
|
db.commit()
|
|
db.refresh(db_service)
|
|
return db_service
|
|
|
|
|
|
def get_services(
|
|
db: Session, skip: int = 0, limit: int = 100
|
|
) -> List[sql_models.Service]:
|
|
return db.query(sql_models.Service).offset(skip).limit(limit).all()
|
|
|
|
|
|
def get_services_by_entity_did(
|
|
db: Session, entity_did: str, skip: int = 0, limit: int = 100
|
|
) -> List[sql_models.Service]:
|
|
return (
|
|
db.query(sql_models.Service)
|
|
.filter(sql_models.Service.entity_did == entity_did)
|
|
.offset(skip)
|
|
.limit(limit)
|
|
.all()
|
|
)
|
|
|
|
|
|
def delete_service_by_entity_did(db: Session, entity_did: str) -> None:
|
|
db.query(sql_models.Service).filter(
|
|
sql_models.Service.entity_did == entity_did
|
|
).delete()
|
|
db.commit()
|
|
|
|
|
|
def get_services_without_entity_id(
|
|
db: Session, entity_did: str, skip: int = 0, limit: int = 100
|
|
) -> List[sql_models.Service]:
|
|
return (
|
|
db.query(sql_models.Service)
|
|
.filter(sql_models.Service.entity_did != entity_did)
|
|
.offset(skip)
|
|
.limit(limit)
|
|
.all()
|
|
)
|
|
|
|
|
|
#########################
|
|
# #
|
|
# Entity #
|
|
# #
|
|
#########################
|
|
def create_entity(db: Session, entity: schemas.EntityCreate) -> sql_models.Entity:
|
|
db_entity = sql_models.Entity(
|
|
**entity.dict(), attached=False, stop_health_task=False
|
|
)
|
|
db.add(db_entity)
|
|
db.commit()
|
|
db.refresh(db_entity)
|
|
return db_entity
|
|
|
|
|
|
def get_entities(
|
|
db: Session, skip: int = 0, limit: int = 100
|
|
) -> List[sql_models.Entity]:
|
|
return db.query(sql_models.Entity).offset(skip).limit(limit).all()
|
|
|
|
|
|
def get_entity_by_did(db: Session, did: str) -> Optional[sql_models.Entity]:
|
|
return db.query(sql_models.Entity).filter(sql_models.Entity.did == did).first()
|
|
|
|
|
|
def get_entity_by_name(db: Session, name: str) -> Optional[sql_models.Entity]:
|
|
return db.query(sql_models.Entity).filter(sql_models.Entity.name == name).first()
|
|
|
|
|
|
# get attached
|
|
def get_attached_entities(
|
|
db: Session, skip: int = 0, limit: int = 100
|
|
) -> List[sql_models.Entity]:
|
|
return (
|
|
db.query(sql_models.Entity)
|
|
.filter(sql_models.Entity.attached == true())
|
|
# https://stackoverflow.com/questions/18998010/flake8-complains-on-boolean-comparison-in-filter-clause
|
|
.offset(skip)
|
|
.limit(limit)
|
|
.all()
|
|
)
|
|
|
|
|
|
# Returns same entity if setting didnt changed something
|
|
def set_stop_health_task(db: Session, entity_did: str, value: bool) -> None:
|
|
db_entity = get_entity_by_did(db, entity_did)
|
|
if db_entity is None:
|
|
raise ClanError(f"Entity with did '{entity_did}' not found")
|
|
|
|
setattr(db_entity, "stop_health_task", value)
|
|
|
|
# save changes in db
|
|
db.add(db_entity)
|
|
db.commit()
|
|
db.refresh(db_entity)
|
|
|
|
|
|
def set_attached_by_entity_did(db: Session, entity_did: str, attached: bool) -> None:
|
|
db_entity = get_entity_by_did(db, entity_did)
|
|
if db_entity is None:
|
|
raise ClanError(f"Entity with did '{entity_did}' not found")
|
|
|
|
setattr(db_entity, "attached", attached)
|
|
|
|
# save changes in db
|
|
db.add(db_entity)
|
|
db.commit()
|
|
db.refresh(db_entity)
|
|
|
|
|
|
def delete_entity_by_did(db: Session, did: str) -> None:
|
|
db.query(sql_models.Entity).filter(sql_models.Entity.did == did).delete()
|
|
db.commit()
|
|
|
|
|
|
def delete_entity_by_did_recursive(db: Session, did: str) -> None:
|
|
delete_service_by_entity_did(db, did)
|
|
delete_entity_by_did(db, did)
|