feat: allow assigning sites directly from role page

2026-04-03 01:56:22 +08:00
parent d5418d47e7
commit da08bc01ec
2 changed files with 27 additions and 0 deletions
--- a/app/repositories/site_roles_repo.py
+++ b/app/repositories/site_roles_repo.py
@@ -35,3 +35,9 @@ class SiteRolesRepository:
        for role_id in role_ids:
            self.db.add(SiteRole(site_id=site_id, role_id=role_id))
        self.db.commit()
+
+    def set_role_sites(self, *, role_id: str, site_ids: list[str]) -> None:
+        self.db.execute(delete(SiteRole).where(SiteRole.role_id == role_id))
+        for site_id in site_ids:
+            self.db.add(SiteRole(site_id=site_id, role_id=role_id))
+        self.db.commit()