##SQLAlchemy-SQLSchema

Provides a context manager to modify the active SQL Schema, implemented for PostgreSQL and Oracle.

http://sqlalchemy-sqlschema.readthedocs.org

###Usage

As a context manager:

from sqlalchemy_sqlschema import maintain_schema

with maintain_schema("my_schema", session):
    schema = session.execute("show search_path").scalar()
    assert schema == "my_schema"

    # a rollback still maintains the schema
    session.rollback()
    assert session.execute("show search_path").scalar() == "my_schema"

As a decorator:

from sqlalchemy_sqlschema import maintain_schema

@maintain_schema("my_schema", session):
def query_data():
    assert session.execute("show search_path").scalar() == "my_schema"

Example application:

Dynamically change the SQL schema based on the current user in a Flask application. Assumes that the user has a schema attribute that points to the desired SQL schema.

from flask import Flask, jsonify
from flask_login import current_user

app = Flask(__name__)

@app.route("/api/data")
def data():
    with maintain_schema(current_user.schema, session)
        data = session.query(MyModel).all()
        return jsonify(data=data)

Tests

You need to create your own tests/test.config file that will provide a URL to a database to be used for the tests. Check tests/test.config.sample for options for different databases.

You can run the tests by invoking PYTHONPATH=. py.test tests/ in the repository root.

TODO

1. Consider how to integrate with Alembic