Question

我正在尝试将子查询中的数据从postgres（from_engine）复制到sqlite数据库。我可以使用以下命令来复制表：

smeta = MetaData(bind=from_engine)
table = Table(table_name, smeta, autoload=True)
table.metadata.create_all(to_engine)

但是，我不确定如何为子查询语句实现相同的功能。

-Sandeep

编辑：跟进答案。创建表后，我想创建一个子查询stmt，如下所示：

table = Table("newtable", dest_metadata, *columns)
stmt = dest_session.query(table).subquery();

但是，最后一个stmt最终会出错 cursor.execute（语句，参数） sqlalchemy.exc.ProgrammingError :( ProgrammingError）关系“newtable”不存在第3行：FROM newtable）AS anon_1

Answer 1

至少在某些情况下有效的方法之一：

使用查询对象的column_descriptions获取有关结果集中列的一些信息。
使用该信息，您可以构建架构以在其他数据库中创建新表。
在源数据库中运行查询并将结果插入新表中。

示例中的第一个设置：

from sqlalchemy import create_engine, MetaData, 
from sqlalchemy import Column, Integer, String, Table
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# Engine to the database to query the data from
# (postgresql)
source_engine = create_engine('sqlite:///:memory:', echo=True)
SourceSession = sessionmaker(source_engine)

# Engine to the database to store the results in
# (sqlite)
dest_engine = create_engine('sqlite:///:memory:', echo=True)
DestSession = sessionmaker(dest_engine)

# Create some toy table and fills it with some data
Base = declarative_base()
class Pet(Base):
    __tablename__ = 'pets'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    race = Column(String)

Base.metadata.create_all(source_engine)
sourceSession = SourceSession()
sourceSession.add(Pet(name="Fido", race="cat"))
sourceSession.add(Pet(name="Ceasar", race="cat"))
sourceSession.add(Pet(name="Rex", race="dog"))
sourceSession.commit()

现在到了有趣的地方：

# This is the query we want to persist in a new table:
query= sourceSession.query(Pet.name, Pet.race).filter_by(race='cat')

# Build the schema for the new table
# based on the columns that will be returned 
# by the query:
metadata = MetaData(bind=dest_engine)
columns = [Column(desc['name'], desc['type']) for desc in query.column_descriptions]
column_names = [desc['name'] for desc in query.column_descriptions]
table = Table("newtable", metadata, *columns)

# Create the new table in the destination database
table.create(dest_engine)

# Finally execute the query
destSession = DestSession()
for row in query:
    destSession.execute(table.insert(row))
destSession.commit()

应该有更有效的方法来完成最后一个循环。但批量插入是另一个话题。

Answer 2

您还可以浏览熊猫数据框。例如，一种方法将使用pandas.read_sql(query, source.connection)和df.to_sql(table_name, con=destination.connection)。

SQLAlchemy - 将子查询的模式和数据复制到另一个数据库

2 个答案: