Move get_publication_writer to layman.util

LayerManager · Dec 13, 2023 · f0b16f7 · f0b16f7
1 parent b6f7c58
commit f0b16f7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/src/layman/util.py b/src/layman/util.py
@@ -701,3 +701,12 @@ def get_complete_publication_info(workspace, publication_type, publication_name,
             f"publication_status={publication_status}, is_updating_after={is_updating_after}")
 
     return complete_info
+
+
+def get_publication_writer(workspace, publication_type, publication_name):
+    from layman.common.prime_db_schema.publications import is_user
+    info = get_publication_info(workspace, publication_type, publication_name, context={'keys': ['access_rights']})
+    return next(
+        user_or_role for user_or_role in info['access_rights']['write']
+        if user_or_role == settings.RIGHTS_EVERYONE_ROLE or is_user(user_or_role)
+    )
diff --git a/tests/asserts/util.py b/tests/asserts/util.py
@@ -1,6 +1,5 @@
 import inspect
 from layman import util as layman_util, settings, app
-from layman.common.prime_db_schema.publications import is_user
 from test_tools import process_client
 from .. import Action, Publication
 
@@ -10,11 +9,7 @@
 
 def get_publication_writer(publication):
     with app.app_context():
-        info = layman_util.get_publication_info(publication.workspace, publication.type, publication.name, context={'keys': ['access_rights']})
-    writer = next(
-        user_or_role for user_or_role in info['access_rights']['write']
-        if user_or_role == settings.RIGHTS_EVERYONE_ROLE or is_user(user_or_role)
-    )
+        writer = layman_util.get_publication_writer(publication.workspace, publication.type, publication.name)
     return writer