Thomas Peetz
20 lines
584 B
Python
20 lines
584 B
Python
from sqlalchemy import Column, String, DateTime, Integer
|
|
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) |