upd get_callable_str async
This commit is contained in:
@@ -6,7 +6,7 @@ from ....utils.main import get_callable_str
|
||||
from ..context import ContextData, CallbackCommand
|
||||
|
||||
|
||||
def add_filter_controls(
|
||||
async def add_filter_controls(
|
||||
keyboard_builder: InlineKeyboardBuilder,
|
||||
entity_descriptor: EntityDescriptor,
|
||||
filter: str = None,
|
||||
@@ -15,7 +15,7 @@ def add_filter_controls(
|
||||
):
|
||||
caption = ", ".join(
|
||||
[
|
||||
get_callable_str(
|
||||
await get_callable_str(
|
||||
entity_descriptor.fields_descriptors[field_name].caption,
|
||||
entity_descriptor,
|
||||
)
|
||||
|
||||
@@ -17,7 +17,6 @@ class CallbackCommand(StrEnum):
|
||||
DATE_PICKER_MONTH = "dm"
|
||||
DATE_PICKER_YEAR = "dy"
|
||||
TIME_PICKER = "tp"
|
||||
# STRING_EDITOR_LOCALE = "sl"
|
||||
ENTITY_PICKER_PAGE = "ep"
|
||||
ENTITY_PICKER_TOGGLE_ITEM = "et"
|
||||
VIEW_FILTER_EDIT = "vf"
|
||||
|
||||
@@ -25,13 +25,23 @@ async def show_editor(message: Message | CallbackQuery, **kwargs):
|
||||
value_type = field_descriptor.type_base
|
||||
|
||||
if field_descriptor.edit_prompt:
|
||||
edit_prompt = get_callable_str(
|
||||
field_descriptor.edit_prompt, field_descriptor, None, current_value
|
||||
edit_prompt = await get_callable_str(
|
||||
field_descriptor.edit_prompt,
|
||||
field_descriptor,
|
||||
callback_data
|
||||
if callback_data.context == CommandContext.COMMAND_FORM
|
||||
else None,
|
||||
current_value,
|
||||
)
|
||||
else:
|
||||
if field_descriptor.caption:
|
||||
caption_str = get_callable_str(
|
||||
field_descriptor.caption, field_descriptor, None, current_value
|
||||
caption_str = await get_callable_str(
|
||||
field_descriptor.caption,
|
||||
field_descriptor,
|
||||
callback_data
|
||||
if callback_data.context == CommandContext.COMMAND_FORM
|
||||
else None,
|
||||
current_value,
|
||||
)
|
||||
else:
|
||||
caption_str = field_descriptor.name
|
||||
@@ -42,7 +52,7 @@ async def show_editor(message: Message | CallbackQuery, **kwargs):
|
||||
)
|
||||
).format(
|
||||
name=caption_str,
|
||||
value=get_value_repr(current_value, field_descriptor, user.lang),
|
||||
value=await get_value_repr(current_value, field_descriptor, user.lang),
|
||||
)
|
||||
else:
|
||||
edit_prompt = (
|
||||
|
||||
@@ -56,12 +56,20 @@ async def time_picker(
|
||||
|
||||
if not current_value:
|
||||
current_value = time(0, 0)
|
||||
is_datetime = False
|
||||
else:
|
||||
is_datetime = isinstance(current_value, datetime)
|
||||
if not is_datetime:
|
||||
current_time = datetime.combine(datetime.now(), current_value)
|
||||
remainder = current_value.minute % 5
|
||||
if remainder >= 3:
|
||||
current_value += timedelta(minutes=(5 - remainder))
|
||||
current_time += timedelta(minutes=(5 - remainder))
|
||||
else:
|
||||
current_value -= timedelta(minutes=remainder)
|
||||
current_time -= timedelta(minutes=remainder)
|
||||
if is_datetime:
|
||||
current_value = datetime.combine(current_value.date(), current_time.time())
|
||||
else:
|
||||
current_value = current_time.time()
|
||||
|
||||
for i in range(12):
|
||||
keyboard_builder.row(
|
||||
|
||||
@@ -209,7 +209,7 @@ async def render_entity_picker(
|
||||
type_.bot_entity_descriptor, item
|
||||
)
|
||||
if type_.bot_entity_descriptor.item_repr
|
||||
else get_callable_str(
|
||||
else await get_callable_str(
|
||||
type_.bot_entity_descriptor.full_name,
|
||||
type_.bot_entity_descriptor,
|
||||
item,
|
||||
@@ -259,7 +259,7 @@ async def render_entity_picker(
|
||||
and form_list.filtering
|
||||
and form_list.filtering_fields
|
||||
):
|
||||
add_filter_controls(
|
||||
await add_filter_controls(
|
||||
keyboard_builder=keyboard_builder,
|
||||
entity_descriptor=type_.bot_entity_descriptor,
|
||||
filter=entity_filter,
|
||||
|
||||
@@ -100,13 +100,13 @@ async def entity_item(
|
||||
]
|
||||
field_value = getattr(entity_item, field_descriptor.field_name)
|
||||
if btn_caption:
|
||||
btn_text = get_callable_str(
|
||||
btn_text = await get_callable_str(
|
||||
btn_caption, field_descriptor, entity_item, field_value
|
||||
)
|
||||
else:
|
||||
if field_descriptor.type_base is bool:
|
||||
btn_text = f"{'【✔︎】 ' if field_value else '【 】 '}{
|
||||
get_callable_str(
|
||||
await get_callable_str(
|
||||
field_descriptor.caption,
|
||||
field_descriptor,
|
||||
entity_item,
|
||||
@@ -116,18 +116,20 @@ async def entity_item(
|
||||
else field_name
|
||||
}"
|
||||
else:
|
||||
btn_text = (
|
||||
f"✏️ {
|
||||
get_callable_str(
|
||||
field_descriptor.caption,
|
||||
field_descriptor,
|
||||
entity_item,
|
||||
field_value,
|
||||
)
|
||||
}"
|
||||
btn_text = f"{
|
||||
field_descriptor.icon
|
||||
if field_descriptor.icon
|
||||
else '✏️'
|
||||
} {
|
||||
await get_callable_str(
|
||||
field_descriptor.caption,
|
||||
field_descriptor,
|
||||
entity_item,
|
||||
field_value,
|
||||
)
|
||||
if field_descriptor.caption
|
||||
else f"✏️ {field_name}"
|
||||
)
|
||||
else field_name
|
||||
}"
|
||||
btn_row.append(
|
||||
InlineKeyboardButton(
|
||||
text=btn_text,
|
||||
@@ -144,7 +146,7 @@ async def entity_item(
|
||||
elif isinstance(button, CommandButton):
|
||||
btn_caption = button.caption
|
||||
|
||||
btn_text = get_callable_str(
|
||||
btn_text = await get_callable_str(
|
||||
btn_caption, entity_descriptor, entity_item
|
||||
)
|
||||
|
||||
@@ -215,7 +217,7 @@ async def entity_item(
|
||||
item_text = form.item_repr(entity_descriptor, entity_item)
|
||||
else:
|
||||
entity_caption = (
|
||||
get_callable_str(
|
||||
await get_callable_str(
|
||||
entity_descriptor.full_name, entity_descriptor, entity_item
|
||||
)
|
||||
if entity_descriptor.full_name
|
||||
@@ -223,7 +225,7 @@ async def entity_item(
|
||||
)
|
||||
|
||||
entity_item_repr = (
|
||||
get_callable_str(
|
||||
await get_callable_str(
|
||||
entity_descriptor.item_repr, entity_descriptor, entity_item
|
||||
)
|
||||
if entity_descriptor.item_repr
|
||||
@@ -234,18 +236,18 @@ async def entity_item(
|
||||
|
||||
for field_descriptor in entity_descriptor.fields_descriptors.values():
|
||||
if field_descriptor.is_visible:
|
||||
field_caption = get_callable_str(
|
||||
field_caption = await get_callable_str(
|
||||
field_descriptor.caption, field_descriptor, entity_item
|
||||
)
|
||||
if field_descriptor.caption_value:
|
||||
value = get_callable_str(
|
||||
value = await get_callable_str(
|
||||
field_descriptor.caption_value,
|
||||
field_descriptor,
|
||||
entity_item,
|
||||
getattr(entity_item, field_descriptor.field_name),
|
||||
)
|
||||
else:
|
||||
value = get_value_repr(
|
||||
value = await get_value_repr(
|
||||
value=getattr(entity_item, field_descriptor.field_name),
|
||||
field_descriptor=field_descriptor,
|
||||
locale=user.lang,
|
||||
|
||||
@@ -61,7 +61,7 @@ async def entity_delete_callback(query: CallbackQuery, **kwargs):
|
||||
return await query.message.edit_text(
|
||||
text=(
|
||||
await Settings.get(Settings.APP_STRINGS_CONFIRM_DELETE_P_NAME)
|
||||
).format(name=get_entity_item_repr(entity=entity)),
|
||||
).format(name=await get_entity_item_repr(entity=entity)),
|
||||
reply_markup=InlineKeyboardBuilder()
|
||||
.row(
|
||||
InlineKeyboardButton(
|
||||
|
||||
@@ -197,7 +197,7 @@ async def entity_list(
|
||||
caption = entity_descriptor.item_repr(entity_descriptor, item)
|
||||
elif entity_descriptor.full_name:
|
||||
caption = f"{
|
||||
get_callable_str(
|
||||
await get_callable_str(
|
||||
callable_str=entity_descriptor.full_name,
|
||||
descriptor=entity_descriptor,
|
||||
entity=item,
|
||||
@@ -228,7 +228,7 @@ async def entity_list(
|
||||
)
|
||||
|
||||
if form_list.filtering and form_list.filtering_fields:
|
||||
add_filter_controls(
|
||||
await add_filter_controls(
|
||||
keyboard_builder=keyboard_builder,
|
||||
entity_descriptor=entity_descriptor,
|
||||
filter=entity_filter,
|
||||
@@ -245,17 +245,17 @@ async def entity_list(
|
||||
)
|
||||
|
||||
if form_list.caption:
|
||||
entity_text = get_callable_str(form_list.caption, entity_descriptor)
|
||||
entity_text = await get_callable_str(form_list.caption, entity_descriptor)
|
||||
else:
|
||||
if entity_descriptor.full_name_plural:
|
||||
entity_text = get_callable_str(
|
||||
entity_text = await get_callable_str(
|
||||
entity_descriptor.full_name_plural, entity_descriptor
|
||||
)
|
||||
else:
|
||||
entity_text = entity_descriptor.name
|
||||
|
||||
if entity_descriptor.description:
|
||||
entity_text = f"{entity_text} {get_callable_str(entity_descriptor.description, entity_descriptor)}"
|
||||
entity_text = f"{entity_text} {await get_callable_str(entity_descriptor.description, entity_descriptor)}"
|
||||
|
||||
state: FSMContext = kwargs["state"]
|
||||
state_data = kwargs["state_data"]
|
||||
|
||||
@@ -54,12 +54,12 @@ async def parameters_menu(
|
||||
continue
|
||||
|
||||
if key.caption_value:
|
||||
caption = get_callable_str(
|
||||
caption = await get_callable_str(
|
||||
callable_str=key.caption_value, descriptor=key, entity=None, value=value
|
||||
)
|
||||
else:
|
||||
if key.caption:
|
||||
caption = get_callable_str(
|
||||
caption = await get_callable_str(
|
||||
callable_str=key.caption, descriptor=key, entity=None, value=value
|
||||
)
|
||||
else:
|
||||
@@ -68,7 +68,7 @@ async def parameters_menu(
|
||||
if key.type_ is bool:
|
||||
caption = f"{'【✔︎】' if value else '【 】'} {caption}"
|
||||
else:
|
||||
caption = f"{caption}: {get_value_repr(value=value, field_descriptor=key, locale=user.lang)}"
|
||||
caption = f"{caption}: {await get_value_repr(value=value, field_descriptor=key, locale=user.lang)}"
|
||||
|
||||
keyboard_builder.row(
|
||||
InlineKeyboardButton(
|
||||
|
||||
@@ -8,6 +8,7 @@ from sqlmodel.ext.asyncio.session import AsyncSession
|
||||
from ...main import QBotApp
|
||||
from ...model.settings import Settings
|
||||
from ...model.language import LanguageBase
|
||||
from ...model.user import UserBase
|
||||
from ...utils.main import clear_state
|
||||
|
||||
|
||||
@@ -16,9 +17,24 @@ router = Router()
|
||||
|
||||
|
||||
@router.message(CommandStart())
|
||||
async def start(
|
||||
message: Message, db_session: AsyncSession, app: QBotApp, state: FSMContext
|
||||
):
|
||||
async def start(message: Message, **kwargs):
|
||||
app: QBotApp = kwargs["app"]
|
||||
|
||||
if app.start_handler:
|
||||
await app.start_handler(
|
||||
default_start_handler, message, **kwargs
|
||||
)
|
||||
else:
|
||||
await default_start_handler(message, **kwargs)
|
||||
|
||||
|
||||
async def default_start_handler[UserType: UserBase](
|
||||
message: Message,
|
||||
db_session: AsyncSession,
|
||||
app: QBotApp,
|
||||
state: FSMContext,
|
||||
**kwargs,
|
||||
) -> tuple[UserType, bool]:
|
||||
state_data = await state.get_data()
|
||||
clear_state(state_data=state_data, clear_nav=True)
|
||||
|
||||
@@ -27,6 +43,7 @@ async def start(
|
||||
user = await User.get(session=db_session, id=message.from_user.id)
|
||||
|
||||
if not user:
|
||||
is_new = True
|
||||
msg_text = (await Settings.get(Settings.APP_STRINGS_WELCOME_P_NAME)).format(
|
||||
name=message.from_user.full_name
|
||||
)
|
||||
@@ -61,6 +78,7 @@ async def start(
|
||||
return
|
||||
|
||||
else:
|
||||
is_new = False
|
||||
if user.is_active:
|
||||
msg_text = (
|
||||
await Settings.get(Settings.APP_STRINGS_GREETING_P_NAME)
|
||||
@@ -71,3 +89,5 @@ async def start(
|
||||
).format(name=user.name)
|
||||
|
||||
await message.answer(msg_text)
|
||||
|
||||
return user, is_new
|
||||
|
||||
Reference in New Issue
Block a user