set_tenant problem

[rorik302]

I have middleware that set schema for queryset

class TenantMiddleware(MiddlewareProtocol):
    def __init__(self, app: "ASGIApp"):
        super().__init__(app)
        self.app = app

    async def __call__(self, scope: "Scope", receive: "Receive", send: "Send"):
        schema_name = Request(scope).state.subdomain
        set_tenant(schema_name)
        await self.app(scope, receive, send)

But when I use querysets only first run normally, other seems refer to default schema and get asyncpg.exceptions.UndefinedTableError: relation "company_types" does not exist

async def create(self, data: CompanyTypeCreate) -> ServiceResult:
        try:
            return ServiceResult(data=await CompanyType.query.create(**data.model_dump()), status_code=201) <- this works normally
        except UniqueViolationError:
            company_type = await CompanyType.query.filter(name_short=data.name_short).first() <- this not working
            errors = {}
            if company_type.name_short == data.name_short:
                errors["name_short"] = "Правовая форма уже существует"
            if company_type.name_full == data.name_full:
                errors["name_full"] = "Правовая форма уже существует"
            return ServiceResult(
                error=True, error_message="conflict", data=errors, status_code=status_codes.HTTP_409_CONFLICT
            )

[tarsil]

@rorik302 i might need some more details about what are trying to achieve and what are you effectively doing because you example we have in the tests and it works so without further details, it's hard.

[rorik302]

I've trying to create CompanyType. If similar already exists, then return fields that conflict to api response

[rorik302]

Maybe exception resets schema to default?

[rorik302]

And I'm not using TenantMixin or TenantModel

[tarsil]

Ok, if you check the TenantMixin and TenantModel or the contrib section itself you will see how the examples work. Did you have a look to see some examples?

Also check out how the tests do the multi tenancy change (as example)

https://github.com/tarsil/edgy/blob/main/tests/integration/test_esmerald_tenant.py