SQL Alchemy，如何将数据插入两个表并引用外键？

Question

我正在使用 SQL Alchemy （通过 Flask_sqlalchemy）将 python 字典列表插入 Postgres 数据库。

其中一个表是所有唯一项目的列表（表 1），而第二个表是与项目相关的数据的时间序列（表 2）。

本质上，我想将任何新行（具有唯一散列）插入到表1中，然后将其数据插入到表2中。如果它已经存在于表1中，只需将“子”插入表2中引用表中的条目1.

这是列表中的一项，列表中有数百项。

{'placement': '5662448s608653114', 't1': datetime.datetime(2018, 4, 15, 17, 47, 7, 434982), 't2': datetime.datetime(2018, 4, 25, 17, 47, 7, 434994), 'camp_id': 1, 'clicks': '0', 'visits': '3', 'conversions': '0', 'revenue': '0'}

我想将 5662448s608653114 插入table1，然后将所有其他数据插入table2，其中我不是通过 5662448s608653114 引用该项目，而是通过它在表 1 中的 id 引用该项目

所以我会得到：

表格1：

____________________
1| 5662448s608653114
2| 5520103

表2：

ID | Pl id | T1 | T2 | cost | revenue | clicks
_______________________________________________
499| 1     |
500| 2     |

我测试了这个，它不起作用：

    def write_tracker_data(self):

    for item in self.data:
        ts = Placements(placement_ts_hash=item["placement"])
        pl = TrackerPlacementData(placement_id=ts.id, t1=item["t1"], t2=item["t2"], camp_id=1,  revenue=item["revenue"], clicks=item["clicks"], conversions=item["conversions"])
        db.session.add(pl)

    db.session.commit()

上面的代码插入了数据，但没有插入数据，而不是表 1 中的 id。它看起来也不是很有效，你知道当某件事绝对可以用更好的方式完成时的感觉......

以下是可供参考的模型类：

class Placements(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    traffic_source = db.Column(db.Integer, db.ForeignKey('ts_types.id'))
    placement_ts_hash = db.Column(db.String, index=True)
    placement_url = db.Column(db.String)
    placement_type = db.Column(db.String)

    # Relationship betwwen unique placement table and tracker_placeemnt_data
    tracker_data = db.relationship("TrackerPlacementData", backref="placement_hash")

class TrackerPlacementData(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    t1 = db.Column(db.DateTime(timezone=True))
    t2 = db.Column(db.DateTime(timezone=True), index=True)
    camp_id = db.Column(db.Integer, db.ForeignKey('campaigns.id'), nullable=False)
    placement_id = db.Column(db.Integer, db.ForeignKey('placements.id'), nullable=True, index=True)
    revenue = db.Column(db.Float)
    clicks = db.Column(db.Integer)
    conversions = db.Column(db.Integer)

提前致谢。

编辑：这可行，但由于循环中的每个项目都有一个新会话，它看起来不太好：/

def write_tracker_data(自身):

for item in self.data:
    ts = Placements(placement_ts_hash=item["placement"])
    db.session.add(ts)
    db.session.commit()

    pl = TrackerPlacementData(placement_hash=ts, t1=item["t1"], t2=item["t2"], camp_id=1,
                              revenue=item["revenue"], clicks=item["clicks"], conversions=item["conversions"])
    db.session.add(pl)

    db.session.commit()

Answer 1

您的Placement实例在提交之前不会有 ID。这就是tracker_data关系可以帮助你的地方......

for item in self.data:
    ts = Placements(placement_ts_hash=item["placement"])
    pl = TrackerPlacementData(
        t1=item["t1"], 
        t2=item["t2"], 
        camp_id=1, 
        revenue=item["revenue"], 
        clicks=item["clicks"], 
        conversions=item["conversions"]
    )
    ts.tracker_data.append(pl)
    db.session.add(ts)
db.session.commit()

请注意，它pl.placement_id没有设置任何内容。相反，当您调用 commit 时，一切都会被pl附加到ts.tracker_data并为您照顾。