SQLAlchemy - копировать схему и данные подзапроса в другую базу данных

Я пытаюсь скопировать данные из подзапроса из postgres (from_engine) в базу данных sqlite. Я могу добиться этого для копирования таблицы, используя следующую команду:

Тем не менее, я не уверен, как добиться того же для оператора подзапроса.

Изменить: Последуйте за ответом. Как только я создал таблицу, я хочу создать подзапрос stmt следующим образом:

Однако последний stmt заканчивается ошибкой cursor.execute(оператор, параметры) sqlalchemy.exc.ProgrammingError: (ProgrammingError) отношение "newtable" не существует LINE 3: FROM newtable) AS anon_1

Ответы

Ответ 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()

Должен быть более эффективный способ сделать последний цикл. Но массовая вставка - это еще одна тема.