from sqlalchemy.orm import DeclarativeBase, relationship, sessionmaker, declarative_base class Base(DeclarativeBase): pass # class BaseModel: # # @classmethod # def model_lookup_by_table_name(cls, table_name): # registry_instance = getattr(cls, "registry") # for mapper_ in registry_instance.mappers: # model = mapper_.class_ # model_class_name = model.__tablename__ # if model_class_name == table_name: # return model # Base = declarative_base(cls=BaseModel)