sqlalchemy - SQLAlchemy，如何向两个表中插入数据并引用外键？

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，而是通过它的 id表1

所以我会得到:

表 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)

提前致谢。

编辑:这有效，但由于循环中的每个项目都有一个新 session ，它看起来不太好:/

def write_tracker_data(self):

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 没有设置任何值。相反，pl 被附加到 ts.tracker_data 并且当您调用 commit 时，一切都应该为您处理。