From 040487fac9aa26c96806d8d15e0eb230f140a748 Mon Sep 17 00:00:00 2001
From: wuaho <wuaho@legu.cc>
Date: Mon, 23 Aug 2021 17:42:18 +0800
Subject: [PATCH] 1

---
 api/api_v1/authz/authz.py       | 2 +-
 api/api_v1/endpoints/project.py | 1 +
 utils/casbin/enforcer.py        | 8 ++++++++
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/api/api_v1/authz/authz.py b/api/api_v1/authz/authz.py
index a3a4368..8496248 100644
--- a/api/api_v1/authz/authz.py
+++ b/api/api_v1/authz/authz.py
@@ -55,7 +55,7 @@ async def get_permissions_for_user_in_domain(
         else:
             item['is_authz'] = False
 
-    return schemas.Msg(code='0', msg='ok', data=all_api)
+    return schemas.Msg(code=0, msg='ok', data=all_api)
 
 
 @router.post("/del_role_user_domain")
diff --git a/api/api_v1/endpoints/project.py b/api/api_v1/endpoints/project.py
index d6b1a1b..bffcddd 100644
--- a/api/api_v1/endpoints/project.py
+++ b/api/api_v1/endpoints/project.py
@@ -66,6 +66,7 @@ async def read_project(request: Request,
                        current_user: schemas.UserDB = Depends(deps.get_current_user)
                        ):
     """查看自己拥有的项目"""
+
     res = await crud.project.read_project(db, username=request.user.username)
 
     return schemas.Msg(code=0, msg='ok', data=res)
diff --git a/utils/casbin/enforcer.py b/utils/casbin/enforcer.py
index 9244846..b9d0c92 100644
--- a/utils/casbin/enforcer.py
+++ b/utils/casbin/enforcer.py
@@ -221,3 +221,11 @@ class Enforcer(ManagementEnforcer):
                 'role_id': item[1]
             })
         return res
+
+    def get_domains_for_user(self, user):
+        """获取用户拥有的所有域名"""
+        data = self.get_filtered_grouping_policy(0, user)
+        res = []
+        for item in data:
+            res.append(item[2])
+        return res