from sqlalchemy.orm import sessionmaker
from aws_testing_utils.logger import log


class BaseConnector:

    def __init__(self, engine):
        self.log = log
        self.engine = engine
        self.Session = sessionmaker(bind=self.engine)

    def get_row_by_id(self, db_object, id_field, id_value):
        session = self.Session()
        row = session.query(db_object).filter_by(
            **{id_field: id_value}).first()
        session.close()
        return row