Как можно исправить эту middleware?

Question

[Dejcv2]

Приветствую! Не так давно написал мидлварь для регистрации обычных пользователей и рефералов, вроде бы всё проверил, но смущает размер кода, от этого ощущение, что я много где накосячил

class RegisterCheck(BaseMiddleware):
    async def __call__(
            self,
            handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]],
            event: Union[Message, CallbackQuery],
            data: Dict[str, Any]):
        session_maker: sessionmaker = async_session
        async with session_maker() as session:
            async with session.begin():
                start_message = r"https://t.me/[\w]+?\?start=\d+"
                result = await session.execute(select(Users).where(Users.user_id == event.from_user.id))
                user = result.scalar()

                if isinstance(event, Message):
                    if event.chat.type == 'private':
                        if event.text != start_message:
                            if user is not None:
                                pass
                            else:
                                user = Users(
                                    user_id=event.from_user.id,
                                    user_name=event.from_user.username
                                )
                                await session.merge(user)
                        else:
                            if user is not None:
                                pass
                            else:
                                user = Users(
                                    user_id=event.from_user.id,
                                    user_name=event.from_user.username,
                                    referrer=event.text
                                )
                                await session.merge(user)
                else:
                    if user is not None:
                        pass
                    else:
                        user = Users(
                            user_id=event.from_user.id,
                            user_name=event.from_user.username
                        )
                        await session.merge(user)

        return await handler(event, data)

Как можно улучшить этот код? Из первого, что приходит на ум - заменить условные операторы на match/case, но может ещё что-то можно сделать?

Answer 1

[Everything_is_bad]

user = result.scalar()

if user is None:
    referrer = None
    if event.chat.type == 'private' and event.text == start_message:
        referrer = event.text 

    user = Users(
        user_id=event.from_user.id,
        user_name=event.from_user.username,
        referrer=referrer
    )
    await session.merge(user)

хотя зачем такое делать в middleware?

@dp.message(CommandStart())
async def cmd_start(message, command):
    # а command.args будет твой referrer

Comments

[Dejcv2]

хотя зачем такое делать в middleware?

Ты конкретно о регистрации рефералов или вообще о регистрации?

[Dejcv2]

как по мне, то очевидно почему стоит использовать мтдлварь для простой регистрации, но теперь я и сам не понимаю зачем я исполую её для регистрации рефералов, если проще это сделать через стартовый обработчик

[Everything_is_bad]

Dejcv2, ну про рефы

[Everything_is_bad]

Dejcv2, регистрацию и создания юзера имеет смысл делать в middleware, что бы потом сделать доступным этого юзера в всех хендлерах. А реф удобной отдельной логикой в start, хотя тут тоже может быть вкусовщина и тебе удобно в месте создания юзера.

Answer 2

[Jack444]

Если код работает то не парся. По улучшакам вот что заметил:
if user is not None: повторяется несколько раз, можно один раз проверить выше
if user := result.scalar():
и ниже писать остальную логику