|
|
@@ -0,0 +1,76 @@ |
|
|
from functools import wraps |
|
|
from flask import Flask, g, session, request, abort, jsonify |
|
|
from flask_migrate import MigrateCommand, Migrate |
|
|
from flask_script import Manager |
|
|
from flask_sqlalchemy import SQLAlchemy |
|
|
|
|
|
|
|
|
flask_app = Flask(__name__, static_folder='./static') |
|
|
db = SQLAlchemy() |
|
|
migrate = Migrate() |
|
|
|
|
|
flask_app.add_url_rule('/static/<path:filename>', |
|
|
endpoint='static', |
|
|
subdomain='<tenant>', |
|
|
view_func=flask_app.send_static_file) |
|
|
|
|
|
|
|
|
flask_app.config['SERVER_NAME'] = 'localhost:5000' |
|
|
|
|
|
|
|
|
class Hello(db.Model): |
|
|
""" |
|
|
demo Table. |
|
|
|
|
|
contains name for eg. |
|
|
""" |
|
|
__tablename__ = 'hello' |
|
|
id = db.Column(db.Integer, primary_key=True) |
|
|
name = db.Column(db.String(2000), index=True, nullable=False) |
|
|
|
|
|
|
|
|
@flask_app.before_request |
|
|
def start(): |
|
|
""" |
|
|
Switching DataBase based on subdomain. |
|
|
|
|
|
This is example multi Tenant with multi DataBase rather than multi schema. |
|
|
Logic for both is same, before request find the subdomain and |
|
|
switch its schema or db. |
|
|
""" |
|
|
from urlparse import urlparse |
|
|
subdomain = urlparse(request.url).hostname.split('.')[0] |
|
|
if subdomain == 'tenant': |
|
|
flask_app.config.update({ |
|
|
"SQLALCHEMY_DATABASE_URI": 'postgresql://postgres:postgres@localhost/tenant' |
|
|
}) |
|
|
else: |
|
|
flask_app.config.update({ |
|
|
"SQLALCHEMY_DATABASE_URI": 'postgresql://postgres:postgres@localhost/tenant2' |
|
|
}) |
|
|
|
|
|
|
|
|
@flask_app.url_value_preprocessor |
|
|
def before_route(endpoint, values): |
|
|
"""Remove tenant from values so it not added all url path.""" |
|
|
if values is not None: |
|
|
values.pop('tenant', None) |
|
|
|
|
|
|
|
|
@flask_app.route('/hello', subdomain='<tenant>') |
|
|
def display_name(): |
|
|
"""Display all object in table hello for each tenant subdomain.""" |
|
|
oHello = Hello.query.all() |
|
|
return jsonify(tenant=[e.name for e in oHello]), 200 |
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
db.init_app(flask_app) |
|
|
# default db |
|
|
flask_app.config.update({ |
|
|
"SQLALCHEMY_DATABASE_URI": 'postgresql://postgres:postgres@localhost/tenant2' |
|
|
}) |
|
|
migrate.init_app(flask_app, db) |
|
|
manager = Manager(flask_app) |
|
|
manager.add_command('db', MigrateCommand) |
|
|
manager.run() |
TonyFrancis renamed this gist