python-3.x - 发布嵌套模型 SQLAlchemy-Marshmallow
问题描述
我正在尝试将 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 表的唯一性约束。如何确保未创建新实例但使用现有资产?
解决方案
我最终检查了资产是否存在,如果它们还不存在,则将它们添加到数据库中。我在交易对的 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 中都找不到任何关于如何正确处理它的文档,但现在,它可以工作。
推荐阅读
- scanf - 在c编程中输入时出错
- internet-explorer - 在 ie11 上运行 Nextjs 的配置
- amazon-web-services - 加载经常变化的源列
- angular - 轮播动画无法使用真实数据正常运行
- architecture - 在(基于位置的服务)LBS 应用程序上与 COARSE 位置兼容的可能性和服务
- neo4j - apoc.text.sorensenDiceSimilarity 的意外结果?
- python - 如何根据特定列中的特定值识别索引?
- r - 梯度提升模型的变量重要性(以 50 天为间隔滚动)
- spring-boot - 在 Spring Boot 中,当我从 DB 中获取数据时,弹出错误 List 为 null,但 db 表中的数据可用
- git - 获取 git 历史中最后一个标签的提交 ID