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()