algebra.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. def _roles_acls(key_name, roles):
  2. acls = []
  3. for role in roles:
  4. role_permissions = role.permissions.get(key_name)
  5. if role_permissions:
  6. acls.append(role_permissions)
  7. return acls
  8. def sum_acls(result_acl, acls=None, roles=None, key=None, **permissions):
  9. if acls and roles:
  10. raise ValueError('You can not provide both "acls" and "roles" arguments')
  11. if (acls is None) and (roles is None):
  12. raise ValueError('You have to provide either "acls" and "roles" argument')
  13. if roles is not None:
  14. if not key:
  15. raise ValueError(
  16. 'You have to provide "key" argument if you '
  17. "are passing roles instead of acls"
  18. )
  19. acls = _roles_acls(key, roles)
  20. for permission, compare in permissions.items():
  21. try:
  22. permission_value = result_acl[permission]
  23. except KeyError:
  24. message = 'Default value for permission "%s" is not provided'
  25. raise ValueError(message % permission)
  26. for acl in acls:
  27. try:
  28. permission_value = compare(permission_value, acl[permission])
  29. except KeyError:
  30. pass
  31. result_acl[permission] = permission_value
  32. return result_acl
  33. # Common comparisions
  34. def greater(a, b):
  35. return a if a > b else b
  36. def greater_or_zero(a, b):
  37. if a == 0:
  38. return a
  39. if b == 0:
  40. return b
  41. return greater(a, b)
  42. def lower(a, b):
  43. return a if a < b else b
  44. def lower_non_zero(a, b):
  45. if a == 0:
  46. return b
  47. if b == 0:
  48. return a
  49. return lower(a, b)