aboutsummaryrefslogtreecommitdiff
path: root/migrations/auth/20230306_01_pRfxl-add-system-user-list-privilege.py
blob: 17db17c074867056cc85bdf7c44a77169f2dc1c7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
"""
Add system:user:list privilege
"""
import contextlib

from yoyo import step

__depends__ = {'20230216_02_0ZHSl-make-dataset-id-and-trait-id-foreign-keys-in-tables'}

def insert_users_list_priv(conn):
    """Create a new 'system:user:list' privilege."""
    with contextlib.closing(conn.cursor()) as cursor:
        cursor.execute(
            "INSERT INTO privileges(privilege_id, privilege_description) "
            "VALUES('system:user:list', 'List users in the system') "
            "ON CONFLICT (privilege_id) DO NOTHING")

def delete_users_list_priv(conn):
    """Delete the new 'system:user:list' privilege."""
    with contextlib.closing(conn.cursor()) as cursor:
        cursor.execute(
            "DELETE FROM privileges WHERE privilege_id='system:user:list'")

steps = [
    step(insert_users_list_priv, delete_users_list_priv)
]