SQLAlchemy печать необработанного SQL из create () - PullRequest
36 голосов
/ 24 января 2010

Я даю Pylons попытку с SQLAlchemy, и мне это нравится, есть только одна вещь: можно ли распечатать необработанные данные SQL CREATE TABLE, сгенерированные из Table (). Create (), перед тем, как они будут выполнены? *

Ответы [ 6 ]

82 голосов
/ 30 июня 2010
from sqlalchemy.schema import CreateTable

print(CreateTable(table))

Если вы используете декларативный синтаксис:

print(CreateTable(Model.__table__))

Обновление:

Поскольку у меня есть принятый ответ, и в klenwell answer есть важная информация, я также добавлю его сюда.

Вы можете получить SQL для вашей конкретной базы данных (MySQL, Postgresql и т. Д.), Скомпилировав ее с вашим движком.

print(CreateTable(Model.__table__).compile(engine))

Обновление 2:

@ jackotonye Добавил в комментарии способ сделать это без движка.

print(CreateTable(Model.__table__).compile(dialect=postgresq‌​l.dialect()))
13 голосов
/ 18 апреля 2012

Мне нужно было получить необработанную таблицу sql, чтобы настроить тесты для некоторых существующих моделей. Вот успешный модульный тест, который я создал для SQLAlchemy 0.7.4 на основе ответа Антуана в качестве подтверждения концепции:

from sqlalchemy import create_engine
from sqlalchemy.schema import CreateTable
from model import Foo

sql_url = "sqlite:///:memory:"    
db_engine = create_engine(sql_url)

table_sql = CreateTable(Foo.table).compile(db_engine)
self.assertTrue("CREATE TABLE foos" in str(table_sql))
9 голосов
/ 10 октября 2017

Вы можете настроить свой движок для вывода последовательности создания метаданных, используя следующее:

def metadata_dump(sql, *multiparams, **params):
    # print or write to log or file etc
    print(sql.compile(dialect=engine.dialect))

engine = create_engine(myDatabaseURL, strategy='mock', executor=metadata_dump)
metadata.create_all(engine)

Одним из преимуществ этого подхода является то, что в распечатку включаются перечисления и индексы. Использование CreateTable исключает это.

Другое преимущество состоит в том, что порядок определений схемы является правильным и (почти) применимым в качестве сценария.

8 голосов
/ 25 января 2010

Как то так? (из FAQ по SQLA)

http://docs.sqlalchemy.org/en/latest/faq/sqlexpressions.html

2 голосов
/ 13 апреля 2017

Оказывается, это просто:

from sqlalchemy.dialects import postgresql
from sqlalchemy.schema import CreateTable
from sqlalchemy import Table, Column, String, MetaData

metadata = MetaData()

users = Table('users', metadata,
              Column('username', String)
)

statement = CreateTable(users)

print(statement.compile(dialect=postgresql.dialect()))

Выводит это:

CREATE TABLE users (
    username VARCHAR
)

Если пойти дальше, он может даже поддерживать связанные параметры в подготовленных выражениях.

Ссылка

Как отобразить выражения SQL в виде строк, возможно, со встроенными связанными параметрами?

...

или без двигателя:

from sqlalchemy.dialects import postgresql
print(statement.compile(dialect=postgresql.dialect()))

ИСТОЧНИК: http://docs.sqlalchemy.org/en/latest/faq/sqlexpressions.html#faq-sql-expression-string

Пример: использование SQL Alchemy для генерации сценария переименования пользователя

#!/usr/bin/env python
import csv
from sqlalchemy.dialects import postgresql
from sqlalchemy import bindparam, Table, Column, String, MetaData

metadata = MetaData()

users = Table('users', metadata,
              Column('username', String)
)

renames = []

with open('users.csv') as csvfile:
    for row in csv.DictReader(csvfile):
        renames.append({
            'from': row['sAMAccountName'],
            'to': row['mail']
        })

for rename in renames:
    stmt = (users.update()
            .where(users.c.username == rename['from'])
            .values(username=rename['to']))
    print(str(stmt.compile(dialect=postgresql.dialect(),
                           compile_kwargs={"literal_binds": True})) + ';')

При обработке этого users.csv:

sAMAccountName,mail
bmcboatface,boaty.mcboatface@example.com
ndhyani,naina.dhyani@contoso.com

Дает вывод, как это:

UPDATE users SET username='boaty.mcboatface@example.com' WHERE users.username = 'bmcboatface';
UPDATE users SET username='naina.dhyani@contoso.com' WHERE users.username = 'ndhyani';users.username = 'ndhyani';

Почему у исследовательского судна есть адрес электронной почты, еще предстоит определить. Я связывался с ИТ-командой Example Inc и не получил ответа.

1 голос
/ 24 марта 2011

Может быть, вы имеете в виду echo параметр sqlalchemy.create_engine?

/ tmp $ cat test_s.py

import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Department(Base):
    __tablename__ = "departments"

    department_id = sa.Column(sa.types.Integer, primary_key=True)
    name = sa.Column(sa.types.Unicode(100), unique=True)
    chief_id = sa.Column(sa.types.Integer)
    parent_department_id = sa.Column(sa.types.Integer,
                                     sa.ForeignKey("departments.department_id"))

    parent_department = sa.orm.relation("Department")


engine = sa.create_engine("sqlite:///:memory:", echo=True)
Base.metadata.create_all(bind=engine)

/ tmp $ python test_s.py

2011-03-24 15:09:58,311 INFO sqlalchemy.engine.base.Engine.0x...42cc PRAGMA table_info("departments")
2011-03-24 15:09:58,312 INFO sqlalchemy.engine.base.Engine.0x...42cc ()
2011-03-24 15:09:58,312 INFO sqlalchemy.engine.base.Engine.0x...42cc 
CREATE TABLE departments (
    department_id INTEGER NOT NULL, 
    name VARCHAR(100), 
    chief_id INTEGER, 
    parent_department_id INTEGER, 
    PRIMARY KEY (department_id), 
    UNIQUE (name), 
    FOREIGN KEY(parent_department_id) REFERENCES departments (department_id)
)

2011-03-24 15:09:58,312 INFO sqlalchemy.engine.base.Engine.0x...42cc ()
2011-03-24 15:09:58,312 INFO sqlalchemy.engine.base.Engine.0x...42cc COMMIT
...