0

我正在尝试将 SQLAlchemy 与 Marshmallow 结合起来。我有一个包含一些资产和交易对的 Flask API。我想要这些模型之间的双向一对多关系。我有以下代码：

class Asset(db.Model):
    __tablename__ = 'asset'

    id  = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), unique=True, nullable=False)
    abbreviation = db.Column(db.String(20), unique=True, nullable=True)
    trading_bases = relationship("TradingPair", back_populates="base_asset", foreign_keys="TradingPair.base_id")
    trading_quotes = relationship("TradingPair", back_populates="quote_asset", foreign_keys="TradingPair.quote_id")


class TradingPair(db.Model):
    __tablename__ = 'trading_pair'

    id  = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), unique=True, nullable=False)

    # One to many (a pair can have only one base, but 1 asset can be the base of many pairs)
    base_id = db.Column(db.Integer, db.ForeignKey("asset.id"), nullable=False)
    base_asset = relationship("Asset", foreign_keys=[base_id], uselist=False, back_populates="trading_bases")

    # One to many (same reasoning)
    quote_id = db.Column(db.Integer, db.ForeignKey("asset.id"), nullable=False)
    quote_asset = relationship("Asset", foreign_keys=[quote_id], uselist=False, back_populates="trading_quotes")

使用以下资源进行交易对 POST：

def post(self, name):
    pair = TradingPair.query.filter_by(name=name).first()
    if pair:
        return {"Error": "This resource already exists"}, 409

    data = request.get_json()
    schema = TradingPairSchema()
    try:
        pair = schema.load(data, session=db.session)
        if not pair.name == name:
            return {"Error": f"{name} does not correspond to name in data"}
        db.session.add(pair)
        db.session.commit()
        return {"Success":f"Added: {pair.name}"} 
    except ValidationError as e:
        return {"Error": e.messages}, 400
    except:
        return {"Error":"Database error"}, 500

我希望 SQLAlchemy 添加作为新交易对的一部分发布的新资产。但是，如果我想使用以下 JSON 通过 API 发布新对：

{'name': 'DASHUSDT', 
 'base_asset': {
   'name': 'Dash', 
   'abbreviation': 'DASH'}, 
 'quote_asset': {
    'name': 'Tether', 
    'abbreviation': 
    'USDT'}}

这可以正常工作，并且该对按预期添加到数据库中。当我尝试添加另一对包含 Dash 或 Tether 时，就会出现问题。该对再次添加到数据库中，并且违反了我对 Asset 表的唯一性约束。如何确保未创建新实例但使用现有资产？

4

1 回答 1

0

我最终检查了资产是否存在，如果它们还不存在，则将它们添加到数据库中。我在交易对的 POST 函数中使用的代码是：

        loaded = schema.load(data, session=db.session)
        if not loaded.name == name:
            return {"Error": f"{name} does not correspond to name in data"}
        base = Asset.query.filter_by(abbreviation=loaded.base_asset.abbreviation).first()
        if not base:
            base = Asset(name=loaded.base_asset.name , abbreviation=loaded.base_asset.abbreviation)
            db.session.add(base)
        quote = Asset.query.filter_by(abbreviation=loaded.quote_asset.abbreviation).first()
        if not quote:
            quote = Asset(name=loaded.quote_asset.name, abbreviation=loaded.quote_asset.abbreviation)
            db.session.add(quote)
        pair = TradingPair(name=name, base_asset=base, quote_asset=quote)
        db.session.add(pair)
        db.session.commit()

当资产已经存在时，这似乎可以正常工作，但在通过交易对的 POST 插入新资产时也不会崩溃。我在 SQLAlchemy、Flask-SQLAlchemy 或 Marshmallow-SQLAlchemy 中都找不到任何关于如何正确处理它的文档，但现在，它可以工作。

于 2020-01-21T07:27:27.380 回答