import random
import psycopg


class DBConn:
    def __init__(self, *args, **kwargs) -> None:
        self.connection = psycopg.connect(*args, **kwargs)
        self.cursor = self.connection.cursor()

    def update(self, query: str, values: tuple) -> list:
        self.cursor.execute(query, values)
        return self.connection.commit()

    def query_raw(self, query: str, values: tuple) -> list:
        self.cursor.execute(query, values)
        return self.cursor.fetchall()

    def query_random(self, query: str, values: tuple) -> str:
        result = self.query_raw(query, values)
        if not result:
            return None
        return random.choice(result)[0]