Python SQLAlchemy 更改 __init__ 中的 __tablename__

Question

我有 2 个具有相同列结构的表。

该脚本从 2 个不同的 json 源中提取，其键略有不同。

我的 Item 类识别源，然后解析数据。

在我的 Item 类中，我希望能够根据数据源更改 __tablename__。

这可能吗？还是我需要为每个数据源编写一个单独的类？

谢谢

代码:

Base = declarative_base()


class Item(Base):
    __tablename__ = 'products'

    timestamp = Column(TIMESTAMP)
    itemid = Column(String, primary_key=True, index=True, unique=True)
    name = Column(String)    

    def __init__(self, item):

        if type(item) == Product_A:
            self.__tablename__ = "A_products"
            # Parse Data
        elif type(item) == Product_B:
            self.__tablename__ = "B_products"
            # Parse Data

Answer 1

这不是一个好主意，在 sqlalchemy 中每个类都应该映射到单个表。解决方案是创建两个类和一个在它们之间分派(dispatch)的自由函数:

Base = declarative_base()
class Item_A(Base):
    __tablename__ = 'A_products'
    timestamp = Column(TIMESTAMP)
    itemid = Column(String, primary_key=True, index=True, unique=True)
    name = Column(String)    

class Item_B(Base):
    __tablename__ = 'B_products'
    timestamp = Column(TIMESTAMP)
    itemid = Column(String, primary_key=True, index=True, unique=True)
    name = Column(String)    

def create_item_object(item):
    if isinstance(item, Product_A):
        result = Item_A()
        #... more stuff
    elif isinstance(item, Product_B):
        result = Item_B()
        #... more stuff
    return result