sqlalchemy 会话创建或更新记录与 * 到许多关系

Question

我有一个相当简单的烧瓶应用程序连接到 postgresql 数据库。我主要使用带有flask-admin 的flask 应用程序，这样我就可以将记录添加到数据库中，并可能稍后将其构建到仪表板中。它基本上是一个内部使用目录。

我正在尝试做的也是编写一个连接到第三方 API 以在数据库中添加/更新记录的脚本，因此它不会通过烧瓶应用程序。我正在使用 SQLAlchemy 来执行此操作，因为它与应用程序一致，并且我只需要一些工作即可，而无需担心 SQL 语句。

flask 应用的数据模型是这样定义的：

app.py

from flask import Flask, render_template, request
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.dialects import postgresql
from flask_admin import Admin

# ... APPLICATION CONFIGURATION ...

# db Models

## Table for many to many
keywords = db.Table('keywords',
    db.Column('keyword_id', db.Integer, db.ForeignKey('keyword.id')),
    db.Column('dataset_id', db.String(24), db.ForeignKey('dataset.dataset_id')),
)

## Model classes
class Dataset(db.Model):
    title = db.Column(db.String(120))
    description = db.Column(db.Text())
    dataset_id = db.Column(db.String(24), primary_key=True, unique=True)

    #relationships
    dataset_documentation = db.relationship('DataDocument', backref='dataset', lazy='dynamic')
    keywords = db.relationship('Keyword', secondary=keywords, backref='dataset', lazy='dynamic')

    def __str__(self):
        return self.title

class Keyword(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    keyword = db.Column(db.String(80))

    def __str__(self):
        return self.keyword

class DataDocument(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    document = db.Column(db.String(120))
    dataset_id = db.Column(db.String(24), db.ForeignKey('dataset.dataset_id'))

    def __str__(self):
        return self.document

# ... APPLICATION VIEWS ...

所以我们有带有一些基本元数据的datasets，它们与文档的文件路径具有一对多关系，与任意数量的关键字具有多对多关系。

单独的脚本直接连接到数据库，并将现有表映射到我可以用来创建会话和修改数据库的对象。

script.py

import config #local config only
from sqlalchemy import create_engine, MetaData, Table
from sqlalchemy.orm import mapper, sessionmaker


# Connecting to postgres database and creating a session with database objects, intantiate empty classes to populate
class Dataset(object):
    pass

class DataDocument(object):
    pass

class Keyword(object):
    pass

## How to instantiate the MTM association table?

db_uri = config.SQLALCHEMY_DATABASE_URI
engine = create_engine(db_uri)
meta = MetaData(engine)
dataset_table = Table('dataset', meta, autoload=True) #correct
datadocument_table = Table('dataset', meta, autoload=True) #incorrect?
keyword_table = Table('keyword', meta, autoload=True) #incorrect?
mapper(Dataset, dataset_table) #correct
mapper(DataDocument, datadocument_table, meta, autoload=True) #??
mapper(Keyword, keyword_table, meta, autoload=True) #??

Session = sessionmaker(bind=engine)
session = Session()
# sample update

data_upsert = Dataset()
data_upsert.title = "Some title"
data_upsert.dataset_id = "Uniq_ID-123"
data_upsert.description = "lorem ipsum foo bar foo"

session.merge(data_upsert)

#attempt to add related properties

key1 = Keyword('test1')
key2 = Keyword('test2')
datadoc = DataDocument('path/to/document.txt')

# FAIL.

data_upsert.append(key1)
data_upsert.append(key2)
data_upsert.append(datadoc)

session.flush()

我是 sqlalchemy 的新手，我几乎无法在数据库引擎的脚本中创建 Dataset 对象。但是我在加载Keyword 和Datadocument 表时也在考虑，它已经根据从数据库加载的内容理解了关系，但这是我的理解薄弱的地方。

有没有简单的方法来完成这里的图片？我假设在script.py 中再次明确定义我的模型没有意义，但是在查看文档和一些教程时，我没有看到将这些关系加载到会话中的缺失部分，以便我可以摄取所有数据进入数据库。

Answer 1

更新您的模型定义以添加构造函数。在这种情况下，它允许您在实例化时将参数传递给对象。

models.py

## Model classes
class Dataset(db.Model):
    title = db.Column(db.String(120))
    description = db.Column(db.Text())
    dataset_id = db.Column(db.String(24), primary_key=True, unique=True)

    #relationships
    dataset_documentation = db.relationship('DataDocument', backref='dataset', lazy='dynamic')
    keywords = db.relationship('Keyword', secondary=keywords, backref='dataset', lazy='dynamic')

    def __init__(self, title=None, desc=None, dataset_id=None):
        self.title = title
        self.description = desc
        self.dataset_id = dataset_id

    def __str__(self):
        return self.title


class Keyword(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    keyword = db.Column(db.String(80))

    def __init__(self, keyword=None):
        self.keyword = keyword

    def __str__(self):
        return self.keyword


class DataDocument(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    document = db.Column(db.String(120))
    dataset_id = db.Column(db.String(24), db.ForeignKey('dataset.dataset_id'))

    def __init__(self, document, dataset_id):
        self.document = document
        self.dataset_id = dataset_id

    def __str__(self):
        return self.document

无需在 script.py 中再次定义模型类。您可以简单地从 models.py 中导入您想要使用的类。然后你可以通过这种方式将数据对象及其相关对象完全插入数据库中：

脚本.py

import config  #local config only
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from models import Dataset, DataDocument, Keyword


def loadSession(engine):
    """"""
    Session = sessionmaker(bind=engine)
    session = Session()
    return session


engine = create_engine(config.SQLALCHEMY_DATABASE_URI, echo=False)
Base = declarative_base(engine)
# load session
session = loadSession(engine)
data_upsert = Dataset(title="Some title", dataset_id="Uniq_ID-125", desc="lorem ipsum foo bar foo")

# add related properties here
key1 = Keyword('test1')
key2 = Keyword('test2')
datadoc = DataDocument('path/to/document.txt', dataset_id="Uniq_ID-125")

# append the properties to the object 
data_upsert.dataset_documentation.append(datadoc)
data_upsert.keywords.append(key1)
data_upsert.keywords.append(key2)

session.add(data_upsert)

session.commit()

我已经在本地测试了代码，希望它对你有用。