Connection Tenant (multiple postgresql schemas with Tortoise Orm)

[ystekno]

Normally postgresql connection in tortoise orm:

register_tortoise(
    app, db_url='postgres://postgres:root@localhost:5432/myweb',
    modules={"models": ["models"]},
    generate_schemas=True
)

connection schema:

DATABASE_SCHEMA = "demo"

register_tortoise(
    app, db_url='postgres://postgres:root@localhost:5432/myweb?schema={DATABASE_SCHEMA}',
    modules={"models": ["models"]},
    generate_schemas=True
)

There are many schemas in postgresql. The customer name entered on the login page is the schema name. When user is log in , it connects to this schema. How can we do this?

from aiohttp import web
from models import Users
from tortoise.contrib.aiohttp import register_tortoise
import jinja2
import aiohttp_jinja2


async def home(request):
    return web.Response(text='Hello Aiohttp!')

@aiohttp_jinja2.template("login.html")
class Login(web.View):
    async def get(self):
        return {}

    async def post(self) -> web.Response:
        form = await self.request.post()
        username = form["username"]
        password = form["password"]
        customer_code = form["customer_code"]
        #customer_code ise schema
        #-----------------------------
        #connect to schema ???????
        DATABASE_SCHEMA = customer_code
        #----------------------------
        if conn == True:
            raise web.HTTPFound(location="/")
        else:
            return {'username': username}


app = web.Application()
app.router.add_view('/', home, name="home")
app.router.add_view('/login', Login, name="login")
register_tortoise(
    app, db_url='postgres://postgres:root@localhost:5433/myweb',
    modules={"models": ["models"]},
    generate_schemas=True
)
aiohttp_jinja2.setup(app, loader=jinja2.FileSystemLoader('templates'))
web.run_app(app, port=8000)

[Dreamsorcerer]

Seems like a question for tortoise...

But, I expect it would require creating a new DB connection manually for each login, without using the register_tortoise() function (which likely will only work while creating the app).

[ystekno]

It gives error:

from aiohttp import web
from tortoise.contrib.aiohttp import register_tortoise
import jinja2
import aiohttp_jinja2
from tortoise import Tortoise, run_async


async def connect_database(DATABASE_SCHEMA):
    register_tortoise(
        app,
        db_url='postgres://postgres:root@localhost:5432/myweb?schema={DATABASE_SCHEMA}',
        modules={"models": ["models"]},
        generate_schemas=True
    )
    return True
    
async def home(request):
    return web.Response(text='Hello Aiohttp!')


class Login(web.View):
    @aiohttp_jinja2.template("login.html")
    async def get(self):
        return {}

    async def post(self) -> web.Response:
        form = await self.request.post()
        username = form["username"]
        password = form["password"]
        customer_code = form["customer_code"]
        #customer_code ise schema
        #-----------------------------
        #connect to schema ???????
        DATABASE_SCHEMA = customer_code
        conn = await connect_database(DATABASE_SCHEMA)
        #----------------------------
        if conn == True:
            raise web.HTTPFound(location="/")
        else:
            return {'username': username}

app = web.Application()
app.router.add_get('/', home, name="home")
app.router.add_view('/login', Login, name="login")
aiohttp_jinja2.setup(app, loader=jinja2.FileSystemLoader('templates'))
web.run_app(app, port=8000)

[Dreamsorcerer]

...without using the register_tortoise() function (which likely will only work while creating the app).