fix setuserdomain& activedomian
This commit is contained in:
@@ -345,7 +345,7 @@ async def create_userdomain(
|
|||||||
db=Depends(get_db),
|
db=Depends(get_db),
|
||||||
):
|
):
|
||||||
try:
|
try:
|
||||||
domain = add_userdomain(db, 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:
|
||||||
raise APIException('platform:domain',request.url._url,f"Error occurred while add user({userid}) domain:",e)
|
raise APIException('platform:domain',request.url._url,f"Error occurred while add user({userid}) domain:",e)
|
||||||
@@ -389,18 +389,17 @@ async def get_activeuserdomain(
|
|||||||
|
|
||||||
@r.put(
|
@r.put(
|
||||||
"/activedomain/{domainid}",
|
"/activedomain/{domainid}",
|
||||||
response_model=ApiReturnModel[Domain],
|
response_model=ApiReturnModel[DomainOut|None],
|
||||||
response_model_exclude_none=True,
|
response_model_exclude_none=True,
|
||||||
)
|
)
|
||||||
async def update_activeuserdomain(
|
async def update_activeuserdomain(
|
||||||
request: Request,
|
request: Request,
|
||||||
domainid:int,
|
domainid:int,
|
||||||
userId: Optional[int] = Query(None, alias="userId"),
|
|
||||||
user=Depends(get_current_active_user),
|
user=Depends(get_current_active_user),
|
||||||
db=Depends(get_db),
|
db=Depends(get_db),
|
||||||
):
|
):
|
||||||
try:
|
try:
|
||||||
domain = active_userdomain(db, userId if userId is not None else user.id,domainid)
|
domain = active_userdomain(db,user.id,domainid)
|
||||||
return ApiReturnModel(data= domain)
|
return ApiReturnModel(data= domain)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise APIException('platform:activedomain',request.url._url,f"Error occurred while update user({user.id}) activedomain:",e)
|
raise APIException('platform:activedomain',request.url._url,f"Error occurred while update user({user.id}) activedomain:",e)
|
||||||
|
|||||||
@@ -336,8 +336,8 @@ def edit_domain(
|
|||||||
return db_domain
|
return db_domain
|
||||||
|
|
||||||
|
|
||||||
def add_userdomain(db: Session, userid:int,domainid:int):
|
def add_userdomain(db: Session,ownerid:int, userid:int,domainid:int):
|
||||||
db_domain = db.query(models.Domain).get(domainid)
|
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:
|
||||||
user_domain = models.UserDomain(userid = userid, domainid = domainid )
|
user_domain = models.UserDomain(userid = userid, domainid = domainid )
|
||||||
db.add(user_domain)
|
db.add(user_domain)
|
||||||
@@ -361,16 +361,18 @@ def delete_userdomain(db: Session, userid: int,domainid: int):
|
|||||||
|
|
||||||
def active_userdomain(db: Session, userid: int,domainid: int):
|
def active_userdomain(db: Session, userid: int,domainid: int):
|
||||||
db_userdomains = db.query(models.UserDomain).filter(models.UserDomain.userid == userid).all()
|
db_userdomains = db.query(models.UserDomain).filter(models.UserDomain.userid == userid).all()
|
||||||
if not db_userdomains:
|
# if not db_userdomains:
|
||||||
raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
|
# raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
|
||||||
|
|
||||||
for domain in db_userdomains:
|
for domain in db_userdomains:
|
||||||
if domain.domainid == domainid:
|
if domain.domainid == domainid:
|
||||||
domain.active = True
|
domain.active = True
|
||||||
|
active_domain = domain
|
||||||
else:
|
else:
|
||||||
domain.active = False
|
domain.active = False
|
||||||
db.add(domain)
|
db.add(domain)
|
||||||
db.commit()
|
db.commit()
|
||||||
return db_userdomains
|
return active_domain
|
||||||
|
|
||||||
def get_activedomain(db: Session, userid: int):
|
def get_activedomain(db: Session, userid: int):
|
||||||
# user_domains = (db.query(models.Domain,models.UserDomain.active)
|
# user_domains = (db.query(models.Domain,models.UserDomain.active)
|
||||||
|
|||||||
Reference in New Issue
Block a user