730108937

Build Type

github

Committed by GitHub

Commit Message

<a href="https://github.com/casbin/pycasbin/commit/<a class=hub.com/casbin/pycasbin/commit/<a class="double-link" href="https://git"><a class=hub.com/casbin/pycasbin/commit/<a class="double-link" href="https://git"><a class=hub.com/casbin/pycasbin/commit/<a class="double-link" href="https://git"><a class=hub.com/casbin/pycasbin/commit/616eaef9db04d6ea8e746dbf00aba15ee717d785">616eaef9d">&lt;a href=&quot;https://github.com/casbin/pycasbin/commit/</a><a class="double-link" href="https://github.com/casbin/pycasbin/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/casbin/pycasbin/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/casbin/pycasbin/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/casbin/pycasbin/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/casbin/pycasbin/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/casbin/pycasbin/commit/616eaef9db04d6ea8e746dbf00aba15ee717d785">616eaef9d</a><a href="https://github.com/casbin/pycasbin/commit/616eaef9db04d6ea8e746dbf00aba15ee717d785">&quot;&gt;&amp;lt;a href=&amp;quot;https://github.com/casbin/pycasbin/commit/&lt;/a&gt;&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/casbin/pycasbin/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/casbin/pycasbin/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/casbin/pycasbin/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/casbin/pycasbin/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/casbin/pycasbin/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/casbin/pycasbin/commit/616eaef9db04d6ea8e746dbf00aba15ee717d785&quot;&gt;616eaef9d&lt;/a&gt;&lt;a href=&quot;https://github.com/casbin/pycasbin/commit/616eaef9db04d6ea8e746dbf00aba15ee717d785&quot;&gt;&amp;quot;&amp;g... (continued)

Pull Request Pull Request #138: fix: wrong domain length judge

Run Details

3 of 5 new or added lines in 1 file covered. (60.0%)

1507 of 1800 relevant lines covered (83.72%)

13.4 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.04

/casbin/rbac/default_role_manager/role_manager.py

import logging

from casbin.rbac import RoleManager


class RoleManager(RoleManager):
    """provides a default implementation for the RoleManager interface"""

    all_roles = dict()
    max_hierarchy_level = 0


    def __init__(self, max_hierarchy_level):
        self.logger = logging.getLogger()
        self.all_roles = dict()
        self.max_hierarchy_level = max_hierarchy_level
        self.matching_func = None
        self.domain_matching_func = None
        self.has_pattern = None
        self.has_domain_pattern = None

    def add_matching_func(self, fn=None):
        self.has_pattern = True
        self.matching_func = fn

    def add_domain_matching_func(self, fn=None):
        self.has_domain_pattern = True
        self.domain_matching_func = fn

    def has_role(self, name):
        if self.matching_func is None:
            return name in self.all_roles.keys()
        else:
            for key in self.all_roles.keys():
                if self.matching_func(name, key):
                    return True
        return False

    def create_role(self, name):
        if name not in self.all_roles.keys():
            self.all_roles[name] = Role(name)

        return self.all_roles[name]

    def clear(self):
        self.all_roles.clear()

    def add_link(self, name1, name2, *domain):
        if len(domain) == 1:
            if len(domain[0]) > 0:
                name1 = domain[0] + "::" + name1
                name2 = domain[0] + "::" + name2
        elif len(domain) > 1:
            raise RuntimeError("error: domain should be 1 parameter")

        role1 = self.create_role(name1)
        role2 = self.create_role(name2)
        role1.add_role(role2)

        if self.matching_func is not None:
            for key, role in self.all_roles.items():
                if self.matching_func(key, name1) and name1 != key:
                    self.all_roles[key].add_role(role1)
                if self.matching_func(key, name2) and name2 != key:
                    self.all_roles[name2].add_role(role)
                if self.matching_func(name1, key) and name1 != key:
                    self.all_roles[key].add_role(role1)
                if self.matching_func(name2, key) and name2 != key:
                    self.all_roles[name2].add_role(role)

    def delete_link(self, name1, name2, *domain):
        if len(domain) == 1:
            if len(domain[0]) > 0:
                name1 = domain[0] + "::" + name1
                name2 = domain[0] + "::" + name2
        elif len(domain) > 1:
            raise RuntimeError("error: domain should be 1 parameter")

        if not self.has_role(name1) or not self.has_role(name2):
            raise RuntimeError("error: name1 or name2 does not exist")

        role1 = self.create_role(name1)
        role2 = self.create_role(name2)
        role1.delete_role(role2)

    def has_link(self, name1, name2, *domain):
        if len(domain) == 1:
            name1 = domain[0] + "::" + name1
            name2 = domain[0] + "::" + name2
        elif len(domain) > 1:
            raise RuntimeError("error: domain should be 1 parameter")

        if name1 == name2:
            return True

        if not self.has_role(name1) or not self.has_role(name2):
            return False

        if self.matching_func is None:
            role1 = self.create_role(name1)
            return role1.has_role(name2, self.max_hierarchy_level)
        else:
            for key, role in self.all_roles.items():
                if self.matching_func(name1, key) and role.has_role(name2, self.max_hierarchy_level, self.matching_func):
                    return True
            return False        

    def get_roles(self, name, domain=None):
        """
        gets the roles that a subject inherits.
        domain is a prefix to the roles.
        """
        if domain:
            name = domain + "::" + name

        if not self.has_role(name):
            return []

        roles = self.create_role(name).get_roles()
        if domain:
            for key, value in enumerate(roles):
                roles[key] = value[len(domain) + 2:]

        return roles

    def get_users(self, name, *domain):
        """
        gets the users that inherits a subject.
        domain is an unreferenced parameter here, may be used in other implementations.
        """
        if len(domain) == 1:
            name = domain[0] + "::" + name
        elif len(domain) > 1:
            return RuntimeError("error: domain should be 1 parameter")

        if not self.has_role(name):
            return []

        names = []
        for role in self.all_roles.values():
            if role.has_direct_role(name):
                if len(domain) == 1:
                    names.append(role.name[len(domain[0]) + 2:])
                else:
                    names.append(role.name)

        return names

    def print_roles(self):
        line = []
        for role in self.all_roles.values():
            text = role.to_string()
            if text:
                line.append(text)
        self.logger.info(", ".join(line))


class Role:
    """represents the data structure for a role in RBAC."""

    name = ""

    roles = []

    def __init__(self, name):
        self.name = name
        self.roles = []

    def add_role(self, role):
        for rr in self.roles:
            if rr.name == role.name:
                return

        self.roles.append(role)

    def delete_role(self, role):
        for rr in self.roles:
            if rr.name == role.name:
                self.roles.remove(rr)
                return

    def has_role(self, name, hierarchy_level, matching_func=None):
        if self.has_direct_role(name, matching_func):
            return True
        if hierarchy_level <= 0:
            return False

        for role in self.roles:
            if role.has_role(name, hierarchy_level - 1, matching_func):
                return True

        return False

    def has_direct_role(self, name, matching_func=None):
        if matching_func is None:
            for role in self.roles:
                if role.name == name:
                    return True
        else:
            for role in self.roles:
                if matching_func(name, role.name):
                    return True
        return False

    def to_string(self):
        if len(self.roles) == 0:
            return ""

        names = ", ".join(self.get_roles())

        if len(self.roles) == 1:
            return self.name + " < " + names
        else:
            return self.name + " < (" + names + ")"

    def get_roles(self):
        names = []
        for role in self.roles:
            names.append(role.name)

        return names

1	import logging	16✔
2
3	from casbin.rbac import RoleManager	16✔
4
5
6	class RoleManager(RoleManager):	16✔
7	"""provides a default implementation for the RoleManager interface"""
8
9	all_roles = dict()	16✔
10	max_hierarchy_level = 0	16✔
11
12
13	def __init__(self, max_hierarchy_level):	16✔
14	self.logger = logging.getLogger()	16✔
15	self.all_roles = dict()	16✔
16	self.max_hierarchy_level = max_hierarchy_level	16✔
17	self.matching_func = None	16✔
18	self.domain_matching_func = None	16✔
19	self.has_pattern = None	16✔
20	self.has_domain_pattern = None	16✔
21
22	def add_matching_func(self, fn=None):	16✔
23	self.has_pattern = True	16✔
24	self.matching_func = fn	16✔
25
26	def add_domain_matching_func(self, fn=None):	16✔
27	self.has_domain_pattern = True	×
28	self.domain_matching_func = fn	×
29
30	def has_role(self, name):	16✔
31	if self.matching_func is None:	16✔
32	return name in self.all_roles.keys()	16✔
33	else:
34	for key in self.all_roles.keys():	16✔
35	if self.matching_func(name, key):	16✔
36	return True	16✔
37	return False	16✔
38
39	def create_role(self, name):	16✔
40	if name not in self.all_roles.keys():	16✔
41	self.all_roles[name] = Role(name)	16✔
42
43	return self.all_roles[name]	16✔
44
45	def clear(self):	16✔
46	self.all_roles.clear()	16✔
47
48	def add_link(self, name1, name2, *domain):	16✔
49	if len(domain) == 1:	16✔
50	if len(domain[0]) > 0:	16✔
51	name1 = domain[0] + "::" + name1	16✔
52	name2 = domain[0] + "::" + name2	16✔
53	elif len(domain) > 1:	16✔
NEW 54	raise RuntimeError("error: domain should be 1 parameter")	×
55
56	role1 = self.create_role(name1)	16✔
57	role2 = self.create_role(name2)	16✔
58	role1.add_role(role2)	16✔
59
60	if self.matching_func is not None:	16✔
61	for key, role in self.all_roles.items():	16✔
62	if self.matching_func(key, name1) and name1 != key:	16✔
63	self.all_roles[key].add_role(role1)	16✔
64	if self.matching_func(key, name2) and name2 != key:	16✔
65	self.all_roles[name2].add_role(role)	16✔
66	if self.matching_func(name1, key) and name1 != key:	16✔
67	self.all_roles[key].add_role(role1)	16✔
68	if self.matching_func(name2, key) and name2 != key:	16✔
69	self.all_roles[name2].add_role(role)	16✔
70
71	def delete_link(self, name1, name2, *domain):	16✔
72	if len(domain) == 1:	16✔
73	if len(domain[0]) > 0:	16✔
74	name1 = domain[0] + "::" + name1	×
NEW 75	name2 = domain[0] + "::" + name2	×
76	elif len(domain) > 1:	16✔
77	raise RuntimeError("error: domain should be 1 parameter")	×
78
79	if not self.has_role(name1) or not self.has_role(name2):	16✔
80	raise RuntimeError("error: name1 or name2 does not exist")	×
81
82	role1 = self.create_role(name1)	16✔
83	role2 = self.create_role(name2)	16✔
84	role1.delete_role(role2)	16✔
85
86	def has_link(self, name1, name2, *domain):	16✔
87	if len(domain) == 1:	16✔
88	name1 = domain[0] + "::" + name1	16✔
89	name2 = domain[0] + "::" + name2	16✔
90	elif len(domain) > 1:	16✔
91	raise RuntimeError("error: domain should be 1 parameter")	×
92
93	if name1 == name2:	16✔
94	return True	16✔
95
96	if not self.has_role(name1) or not self.has_role(name2):	16✔
97	return False	16✔
98
99	if self.matching_func is None:	16✔
100	role1 = self.create_role(name1)	16✔
101	return role1.has_role(name2, self.max_hierarchy_level)	16✔
102	else:
103	for key, role in self.all_roles.items():	16✔
104	if self.matching_func(name1, key) and role.has_role(name2, self.max_hierarchy_level, self.matching_func):	16✔
105	return True	16✔
106	return False	16✔
107
108	def get_roles(self, name, domain=None):	16✔
109	"""
110	gets the roles that a subject inherits.
111	domain is a prefix to the roles.
112	"""
113	if domain:	16✔
114	name = domain + "::" + name	16✔
115
116	if not self.has_role(name):	16✔
117	return []	16✔
118
119	roles = self.create_role(name).get_roles()	16✔
120	if domain:	16✔
121	for key, value in enumerate(roles):	16✔
122	roles[key] = value[len(domain) + 2:]	16✔
123
124	return roles	16✔
125
126	def get_users(self, name, *domain):	16✔
127	"""
128	gets the users that inherits a subject.
129	domain is an unreferenced parameter here, may be used in other implementations.
130	"""
131	if len(domain) == 1:	16✔
132	name = domain[0] + "::" + name	16✔
133	elif len(domain) > 1:	16✔
134	return RuntimeError("error: domain should be 1 parameter")	×
135
136	if not self.has_role(name):	16✔
137	return []	16✔
138
139	names = []	16✔
140	for role in self.all_roles.values():	16✔
141	if role.has_direct_role(name):	16✔
142	if len(domain) == 1:	16✔
143	names.append(role.name[len(domain[0]) + 2:])	16✔
144	else:
145	names.append(role.name)	16✔
146
147	return names	16✔
148
149	def print_roles(self):	16✔
150	line = []	16✔
151	for role in self.all_roles.values():	16✔
152	text = role.to_string()	16✔
153	if text:	16✔
154	line.append(text)	16✔
155	self.logger.info(", ".join(line))	16✔
156
157
158	class Role:	16✔
159	"""represents the data structure for a role in RBAC."""
160
161	name = ""	16✔
162
163	roles = []	16✔
164
165	def __init__(self, name):	16✔
166	self.name = name	16✔
167	self.roles = []	16✔
168
169	def add_role(self, role):	16✔
170	for rr in self.roles:	16✔
171	if rr.name == role.name:	16✔
172	return	×
173
174	self.roles.append(role)	16✔
175
176	def delete_role(self, role):	16✔
177	for rr in self.roles:	16✔
178	if rr.name == role.name:	16✔
179	self.roles.remove(rr)	16✔
180	return	16✔
181
182	def has_role(self, name, hierarchy_level, matching_func=None):	16✔
183	if self.has_direct_role(name, matching_func):	16✔
184	return True	16✔
185	if hierarchy_level <= 0:	16✔
186	return False	×
187
188	for role in self.roles:	16✔
189	if role.has_role(name, hierarchy_level - 1, matching_func):	16✔
190	return True	16✔
191
192	return False	16✔
193
194	def has_direct_role(self, name, matching_func=None):	16✔
195	if matching_func is None:	16✔
196	for role in self.roles:	16✔
197	if role.name == name:	16✔
198	return True	16✔
199	else:
200	for role in self.roles:	16✔
201	if matching_func(name, role.name):	16✔
202	return True	16✔
203	return False	16✔
204
205	def to_string(self):	16✔
206	if len(self.roles) == 0:	16✔
207	return ""	16✔
208
209	names = ", ".join(self.get_roles())	16✔
210
211	if len(self.roles) == 1:	16✔
212	return self.name + " < " + names	16✔
213	else:
214	return self.name + " < (" + names + ")"	16✔
215
216	def get_roles(self):	16✔
217	names = []	16✔
218	for role in self.roles:	16✔
219	names.append(role.name)	16✔
220
221	return names	16✔

casbin / pycasbin / 730108937

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous