Add Roles to export cache

Author: zidarsk8

src/ggrc/converters/base_block.py

@@ -97,6 +97,7 @@ def __init__(self, converter, **options):
     # The protected access is a false warning for inflector access.
     self._mapping_cache = None
     self._owners_cache = None
+    self._roles_cache = None
     self._ca_definitions_cache = None
     self.converter = converter
     self.offset = options.get("offset", 0)
@@ -239,6 +240,13 @@ def get_mapping_cache(self):
       self._mapping_cache = self._create_mapping_cache()
     return self._mapping_cache
 
+  def get_role(self, name):
+    """Get role from local cache for a given name."""
+    if not self._roles_cache:
+      self._roles_cache = {role.name: role for role in
+                           models.all_models.Role.query}
+    return self._roles_cache[name]
+
   def _create_owners_cache(self):
     """Create a cache of emails for all object owners."""
     owner_ids = set()

src/ggrc_basic_permissions/converters/handlers.py

@@ -70,48 +70,48 @@ def insert_object(self):
 class ProgramOwnerColumnHandler(ObjectRoleColumnHandler):
 
   def __init__(self, row_converter, key, **options):
-    self.role = Role.query.filter_by(name="ProgramOwner").one()
+    self.role = row_converter.block_converter.get_role("ProgramOwner")
     super(ProgramOwnerColumnHandler, self).__init__(
         row_converter, key, **options)
 
 
 class ProgramEditorColumnHandler(ObjectRoleColumnHandler):
 
   def __init__(self, row_converter, key, **options):
-    self.role = Role.query.filter_by(name="ProgramEditor").one()
+    self.role = row_converter.block_converter.get_role("ProgramEditor")
     super(ProgramEditorColumnHandler, self).__init__(
         row_converter, key, **options)
 
 
 class ProgramReaderColumnHandler(ObjectRoleColumnHandler):
 
   def __init__(self, row_converter, key, **options):
-    self.role = Role.query.filter_by(name="ProgramReader").one()
+    self.role = row_converter.block_converter.get_role("ProgramReader")
     super(ProgramReaderColumnHandler, self).__init__(
         row_converter, key, **options)
 
 
 class WorkflowOwnerColumnHandler(ObjectRoleColumnHandler):
 
   def __init__(self, row_converter, key, **options):
-    self.role = Role.query.filter_by(name="WorkflowOwner").one()
+    self.role = row_converter.block_converter.get_role("WorkflowOwner")
     super(WorkflowOwnerColumnHandler, self).__init__(
         row_converter, key, **options)
 
 
 class WorkflowMemberColumnHandler(ObjectRoleColumnHandler):
 
   def __init__(self, row_converter, key, **options):
-    self.role = Role.query.filter_by(name="WorkflowMember").one()
+    self.role = row_converter.block_converter.get_role("WorkflowMember")
     super(WorkflowMemberColumnHandler, self).__init__(
         row_converter, key, **options)
 
 
 class AuditAuditorColumnHandler(ObjectRoleColumnHandler):
 
   def __init__(self, row_converter, key, **options):
-    self.role = Role.query.filter_by(name="Auditor").one()
-    self.reader = Role.query.filter_by(name="ProgramReader").one()
+    self.role = row_converter.block_converter.get_role("Auditor")
+    self.reader = row_converter.block_converter.get_role("ProgramReader")
     super(AuditAuditorColumnHandler, self).__init__(
         row_converter, key, **options)
 
@@ -192,6 +192,7 @@ def insert_object(self):
     db.session.add(user_role)
     self.dry_run = True
 
+
 COLUMN_HANDLERS = {
     "default": {
         "program_editor": ProgramEditorColumnHandler,