summaryrefslogtreecommitdiff
path: root/jb/managers/__init__.py
blob: e2aab6d5d265d7ff99e0b128bc6601ad55df4a38 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from enum import IntEnum
from typing import Collection

from generalresearchutils.pg_helper import PostgresConfig


class Permission(IntEnum):
    READ = 1
    UPDATE = 2
    CREATE = 3
    DELETE = 4


class PostgresManager:
    def __init__(
        self,
        pg_config: PostgresConfig,
        permissions: Collection[Permission] = None,
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.pg_config = pg_config
        self.permissions = set(permissions) if permissions else set()