self-constructing_graph/agent/libs/sys.py

import logging
logger = logging.getLogger(__name__)

from agent.models.db.sys import SysOrgan,SysUserRoleOrgan
from datetime import datetime

class SysOrganBusiness:
    def __init__(self, db):
        self.db = db

    def insert_data(self, name, manager="", phone="", parent_id=0):
        sysOrgan = SysOrgan(
            name=name, 
            manager=manager,
            phone=phone,
            parent_id=parent_id
        )
        self.db.add(sysOrgan)
        self.db.commit()
        self.db.refresh(sysOrgan)
        return sysOrgan

    def update_data(self,id, name, manager="", phone="", parent_id=0):
        self.db.query(SysOrgan).filter(SysOrgan.id == id).update({
            SysOrgan.name : name, 
            SysOrgan.manager : manager,
            SysOrgan.phone : phone,
            SysOrgan.parent_id : parent_id
        })
        self.db.commit()

    def get_data_by_name(self, name):
        return self.db.query(SysOrgan).filter(SysOrgan.name == name).first()
    
    def get_data_by_nameAndParent(self, name,parentId):
        return self.db.query(SysOrgan).filter(SysOrgan.parent_id == parentId,SysOrgan.name == name).first()
    
    def get_data(self, data_id):
        return self.db.query(SysOrgan).filter(SysOrgan.id == data_id).first()

    def get_all_data(self):
        return self.db.query(SysOrgan).all()
    
    def get_all_data_by_parentId(self,parentId):
        return self.db.query(SysOrgan).filter(SysOrgan.parent_id == parentId).all()

    def delete_data(self, data_id):
        sysOrgan = self.get_data(data_id)
        if sysOrgan:
            self.db.delete(sysOrgan)
            self.db.commit()

class SysUserRoleOrganBusiness:
    def __init__(self, db):
        self.db = db

    def get_last_use_SURO(self,userId):
        return self.db.query(SysUserRoleOrgan).filter(SysUserRoleOrgan.user_id == userId).order_by(SysUserRoleOrgan.last_use_time.desc()).first()
    
    def getSUROById(self,id):
        return self.db.query(SysUserRoleOrgan).filter(SysUserRoleOrgan.id == id).first()
    

    def get_SURO(self,user_id, organ_id):
        return self.db.query(SysUserRoleOrgan).filter(SysUserRoleOrgan.user_id == user_id,SysUserRoleOrgan.organ_id == organ_id).first()
    
    def insert_data(self, user_id, role_id, organ_id, data_type,create_by):
        sysUserRoleOrgan = SysUserRoleOrgan(
            user_id=user_id, 
            role_id=role_id,
            organ_id=organ_id,
            data_type=data_type,
            create_time=datetime.now(),
            create_by=create_by
        )
        self.db.add(sysUserRoleOrgan)
        self.db.commit()
        self.db.refresh(sysUserRoleOrgan)
        return sysUserRoleOrgan
    
    def update_data(self,id, user_id, role_id, organ_id, data_type,update_by):
        self.db.query(SysUserRoleOrgan).filter(SysUserRoleOrgan.id == id).update({
            SysUserRoleOrgan.user_id : user_id, 
            SysUserRoleOrgan.role_id : role_id,
            SysUserRoleOrgan.organ_id : organ_id,
            SysUserRoleOrgan.data_type : data_type,
            SysUserRoleOrgan.update_time : datetime.now(),
            SysUserRoleOrgan.update_by : update_by
        })
        self.db.commit()

    def updateLastUseTime(self,id):
        self.db.query(SysUserRoleOrgan).filter(SysUserRoleOrgan.id == id).update({
            SysUserRoleOrgan.last_use_time : datetime.now()
        })
        self.db.commit()

    def delete_data(self, data_id):
        sysUserRoleOrgan = self.getSUROById(data_id)
        if sysUserRoleOrgan:
            self.db.delete(sysUserRoleOrgan)
            self.db.commit()