Merge branch 'umask' of ssh://bitbucket.org/yackob03/quay into umask

data/model.py

@@ -4,6 +4,7 @@ import datetime
 import dateutil.parser
 import operator
 import json
+import uuid
 
 from datetime import timedelta
 
@@ -671,6 +672,52 @@ def get_all_user_permissions(user):
                                 (UserThroughTeam.id == user))
 
 
+def delete_prototype_permission(org, uid):
+  found = get_prototype_permission(org, uid)
+  if not found:
+    return None
+
+  found.delete_instance()
+  return found
+
+
+def get_prototype_permission(org, uid):
+  found = None
+  try:
+    return PermissionPrototype.get(PermissionPrototype.org == org, PermissionPrototype.uuid == uid)
+  except PermissionPrototype.DoesNotExist:
+    return None
+
+
+def get_prototype_permissions(org):
+  ActivatingUser = User.alias()
+  DelegateUser = User.alias()
+  where = PermissionPrototype.select().where(PermissionPrototype.org == org)
+  join1 = where.join(ActivatingUser, JOIN_LEFT_OUTER, on=(ActivatingUser.id == PermissionPrototype.activating_user))
+  join2 = join1.join(DelegateUser, JOIN_LEFT_OUTER, on=(DelegateUser.id == PermissionPrototype.delegate_user))
+  join3 = join2.join(Team, JOIN_LEFT_OUTER, on=(Team.id == PermissionPrototype.delegate_team))
+  join4 = join3.join(Role, JOIN_LEFT_OUTER, on=(Role.id == PermissionPrototype.role))
+  return join4
+
+
+def update_prototype_permission(org, uid, role_name):
+  found = get_prototype_permission(org, uid)
+  if not found:
+    return None
+
+  new_role = Role.get(Role.name == role_name)
+  found.role = new_role
+  found.save()
+  return found
+
+
+def add_prototype_permission(org, role_name, activating_user, delegate_user=None, delegate_team=None):
+  new_role = Role.get(Role.name == role_name)
+  uid = uuid.uuid4()
+  return PermissionPrototype.create(org=org, uuid=uid, role=new_role, activating_user=activating_user,
+      delegate_user=delegate_user, delegate_team=delegate_team)
+
+
 def get_org_wide_permissions(user):
   Org = User.alias()
   team_with_role = Team.select(Team, Org, TeamRole).join(TeamRole)