add add_admindomain for admin
This commit is contained in:
@@ -346,6 +346,9 @@ async def create_userdomain(
|
|||||||
db=Depends(get_db),
|
db=Depends(get_db),
|
||||||
):
|
):
|
||||||
try:
|
try:
|
||||||
|
if user.is_superuser:
|
||||||
|
domain = add_admindomain(db,userid,domainid)
|
||||||
|
else:
|
||||||
domain = add_userdomain(db,user.id,userid,domainid)
|
domain = add_userdomain(db,user.id,userid,domainid)
|
||||||
return ApiReturnModel(data = domain)
|
return ApiReturnModel(data = domain)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
@@ -336,6 +336,14 @@ def edit_domain(
|
|||||||
return db_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):
|
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()
|
db_domain = db.query(models.Domain).filter(and_(models.Domain.id == domainid,models.Domain.ownerid == ownerid,models.Domain.is_active)).first()
|
||||||
if db_domain:
|
if db_domain:
|
||||||
|
|||||||
Reference in New Issue
Block a user