add add_admindomain for admin

This commit is contained in:
2024-11-27 16:51:24 +09:00
parent 41aa11720d
commit 49d9475304
2 changed files with 12 additions and 1 deletions

View File

@@ -346,7 +346,10 @@ async def create_userdomain(
db=Depends(get_db),
):
try:
domain = add_userdomain(db,user.id,userid,domainid)
if user.is_superuser:
domain = add_admindomain(db,userid,domainid)
else:
domain = add_userdomain(db,user.id,userid,domainid)
return ApiReturnModel(data = domain)
except Exception as e:
raise APIException('platform:domain',request.url._url,f"Error occurred while add user({userid}) domain:",e)

View File

@@ -336,6 +336,14 @@ def edit_domain(
return db_domain
def add_admindomain(db: Session,userid:int,domainid:int):
db_domain = db.query(models.Domain).filter(and_(models.Domain.id == domainid,models.Domain.is_active)).first()
if db_domain:
user_domain = models.UserDomain(userid = userid, domainid = domainid )
db.add(user_domain)
db.commit()
return db_domain
def add_userdomain(db: Session,ownerid:int, userid:int,domainid:int):
db_domain = db.query(models.Domain).filter(and_(models.Domain.id == domainid,models.Domain.ownerid == ownerid,models.Domain.is_active)).first()
if db_domain: