import sqlalchemy as sa
from src.database import db


class BaseQuery:
    def __init__(self, model):
        self._model = model

    def get(self, primary_key):
        return db.session.get(self._model, primary_key)