fix create_userdomain

2024-11-27 16:12:41 +09:00
parent aa7daf4447
commit 3b15dabedc
3 changed files with 18 additions and 7 deletions
--- a/backend/app/db/crud.py
+++ b/backend/app/db/crud.py
@@ -337,11 +337,12 @@ def edit_domain(


 def add_userdomain(db: Session, userid:int,domainid:int):
-    user_domain = models.UserDomain(userid = userid, domainid = domainid )
-    db.add(user_domain)
-    db.commit()
-    db.refresh(user_domain)
-    return user_domain
+    db_domain = db.query(models.Domain).get(domainid)
+    if db_domain:
+        user_domain = models.UserDomain(userid = userid, domainid = domainid )
+        db.add(user_domain)
+        db.commit()
+    return db_domain

 def add_userdomains(db: Session, userid:int,domainids:list[str]):
    dbCommits = list(map(lambda domainid: models.UserDomain(userid = userid, domainid = domainid ), domainids))