domain

backend/app/db/crud.py

@@ -1,5 +1,6 @@
 from fastapi import HTTPException, status
 from sqlalchemy.orm import Session
+from sqlalchemy import and_
 import typing as t
 
 from . import models, schemas
@@ -182,8 +183,8 @@ def get_flows_by_app(db: Session, appid: str):
     return flows
 
 def create_domain(db: Session, domain: schemas.DomainBase):
-    db_domain = models.UserDomain(
-        userid=domain.userid,
+    db_domain = models.Domain(
+        tenantid = domain.tenantid,
         name=domain.name,
         url=domain.url,
         kintoneuser=domain.kintoneuser,
@@ -194,9 +195,9 @@ def create_domain(db: Session, domain: schemas.DomainBase):
     db.refresh(db_domain)
     return db_domain
 
-def delete_domain(db: Session, userid: int,id: int):
-    db_domain = db.query(models.UserDomain).get(id)
-    if not db_domain or db_domain.userid != userid:
+def delete_domain(db: Session,id: int):
+    db_domain = db.query(models.Domain).get(id)
+    if not db_domain:
         raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
     db.delete(db_domain)
     db.commit()
@@ -206,8 +207,8 @@ def delete_domain(db: Session, userid: int,id: int):
 def edit_domain(
     db: Session, domain: schemas.DomainBase
 ) -> schemas.Domain:
-    db_domain = db.query(models.UserDomain).get(domain.id)
-    if not db_domain or db_domain.userid != domain.userid:
+    db_domain = db.query(models.Domain).get(domain.id)
+    if not db_domain:
         raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
     update_data = domain.dict(exclude_unset=True)
 
@@ -220,8 +221,52 @@ def edit_domain(
     db.refresh(db_domain)
     return db_domain
 
+def add_userdomain(db: Session, userid:int,domainids:list):
+    for domainid in domainids:
+        db_domain = models.UserDomain(
+            userid = userid,
+            domainid = domainid
+        )
+        db.add(db_domain)
+    db.commit()
+    db.refresh(db_domain)
+    return db_domain
+
+def delete_userdomain(db: Session, userid: int,domainid: int):
+    db_domain = db.query(models.UserDomain).filter(and_(models.UserDomain.userid == userid,models.UserDomain.domainid == domainid)).first()
+    if not db_domain:
+        raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
+    db.delete(db_domain)
+    db.commit()
+    return db_domain
+
+def active_userdomain(db: Session, userid: int,domainid: int):
+    db_userdomains = db.query(models.UserDomain).filter(models.UserDomain.userid == userid).all()
+    if not db_userdomains:
+        raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
+    for domain in db_userdomains:
+        if domain.domainid == domainid:
+            domain.active = True
+        else:
+            domain.active = False
+        db.add(domain)
+    db.commit()
+    return db_userdomains
+
+def get_activedomain(db: Session, userid: int):
+    db_domain = db.query(models.Domain).join(models.UserDomain,models.UserDomain.domainid == models.Domain.id ).filter(and_(models.UserDomain.userid == userid,models.UserDomain.active == True)).first()
+    if not db_domain:
+        raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Domain not found")
+    return db_domain
+
 def get_domain(db: Session, userid: str):
-    domains = db.query(models.UserDomain).filter(models.UserDomain.userid == userid).all()
+    domains = db.query(models.Domain).join(models.UserDomain,models.UserDomain.domainid == models.Domain.id ).filter(models.UserDomain.userid == userid).all()
+    if not domains:
+        raise HTTPException(status_code=404, detail="Data not found")
+    return domains
+
+def get_domains(db: Session,tenantid:str):
+    domains = db.query(models.Domain).filter(models.Domain.tenantid == tenantid ).all()
     if not domains:
         raise HTTPException(status_code=404, detail="Data not found")
     return domains