from database import Database


class BaseStorage:

    def __init__(self, db: Database):
        self.db = db

    def __fetch_all(self, sql, *args):
        return self.db.fetch_all(sql, *args)

    def __fetch_one(self, sql, *args):
        return self.db.fetch_one(sql, *args)

    def __form_query(self, table_name: str, kwargs) -> str:
        sql = f'select * from {table_name}'

        if len(kwargs) != 0:
            filters = [f"{key}='{val}'" for key, val in kwargs.items()]
            sql = f"select * from {table_name} where {' and '.join(filters)}"

        return sql

    def get_by(self, table_name: str, **kwargs):
        return self.__fetch_one(self.__form_query(table_name, kwargs))

    def list_by(self, table_name: str, **kwargs):
        return self.__fetch_all(self.__form_query(table_name, kwargs))