add demo users and update the error code

Author: jczhong84

querybook/server/logic/datadoc_permission.py

@@ -55,7 +55,7 @@ def assert_can_read(doc_id, session=None):
         api_assert(
             user_can_read(doc_id, uid=current_user.id, session=session),
             "CANNOT_READ_DATADOC",
-            UNAUTHORIZED_STATUS_CODE,
+            ACCESS_RESTRICTED_STATUS_CODE,
         )
     except DocDoesNotExist:
         api_assert(False, "DOC_DNE", RESOURCE_NOT_FOUND_STATUS_CODE)

querybook/server/logic/demo.py

@@ -6,6 +6,7 @@
     DataElementAssociationType,
 )
 from const.schedule import ScheduleTaskType
+from const.user import UserGroup
 from lib.lineage.utils import lineage as lineage_logic
 from logic import (
     admin as admin_logic,
@@ -128,6 +129,32 @@ def set_up_demo(uid: int, session=None):
         }
 
 
+@with_session
+def create_demo_users(session=None):
+    user_logic.create_user(
+        username="demo-user-1",
+        password="password",
+        fullname="Demo User 1",
+        email="demo-user-1@pins.com",
+        session=session,
+    )
+    user_logic.create_user(
+        username="demo-user-2",
+        password="password",
+        fullname="Demo User 2",
+        email="demo-user-2@pins.com",
+        session=session,
+    )
+    demo_user_group = UserGroup(
+        name="demo-group",
+        display_name="Demo User Group",
+        description="Demo user group",
+        email="demo-group@pins.com",
+        members=["demo-user-1", "demo-user-2"],
+    )
+    user_logic.create_or_update_user_group(demo_user_group, session=session)
+
+
 @with_session
 def create_demo_table_stats(table_id, uid, session=None):
     m_logic.upsert_table_stat(