ksenia_mikhailova
/
to_inventory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add image upload

This commit is contained in:
Kseninia Mikhaylova 2024-07-17 17:15:52 +03:00
parent 979c19e67f
commit 463e7451fb
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
back/tgbot/tgbot.py
View File

@ -156,15 +156,14 @@ class TgBot:
await query.answer() await query.answer()
inv = await TgItem.objects.aget(id=context.chat_data["inv"]) inv = await TgItem.objects.aget(id=context.chat_data["inv"])
tmc = await CustomTable.objects.aget(id=context.chat_data["tmc"]) tmc = await inv.tmc.aget(id=context.chat_data["tmc"])
field_id = query.data.split("@")[-1] field_id = query.data.split("@")[-1]
name = await BaseCustomField.objects.aget(id=field_id) field = await BaseCustomField.objects.aget(id=field_id)
element = await TmcElement.objects.acreate(name=name, tmc=tmc)
await inv.element.aadd(element) await tmc.element.aadd(field)
context.chat_data["element"] = element.id context.chat_data["element"] = field.id
context.chat_data["step"] = "add_element" context.chat_data["step"] = "add_element"
await self.inv(update, context) await self.inv(update, context)
@ -231,9 +230,7 @@ class TgBot:
elif current_step == "add_tmc": elif current_step == "add_tmc":
inv = await TgItem.objects.aget(id=context.chat_data["inv"]) inv = await TgItem.objects.aget(id=context.chat_data["inv"])
tmc_element = await inv.tmc.aget(id=context.chat_data["tmc"]) tmc_element = await inv.tmc.aget(id=context.chat_data["tmc"])
logger.info(tmc_element)
tmc = await CustomTable.objects.aget(id=tmc_element.tmc_id) tmc = await CustomTable.objects.aget(id=tmc_element.tmc_id)
logger.info(tmc)
fields = [] fields = []
async for e in tmc.fields.all(): async for e in tmc.fields.all():
@ -269,8 +266,8 @@ class TgBot:
elif current_step == "add_element": elif current_step == "add_element":
inv = await TgItem.objects.aget(id=context.chat_data["inv"]) inv = await TgItem.objects.aget(id=context.chat_data["inv"])
tmc = inv.tmc.aget(id=context.chat_data["tmc"]) tmc = await inv.tmc.aget(id=context.chat_data["tmc"])
element = inv.element.aget(id=context.chat_data["element"]) element = await tmc.element.aget(id=context.chat_data["element"])
context.chat_data["step"] = "add_element_data" context.chat_data["step"] = "add_element_data"
@ -289,8 +286,8 @@ class TgBot:
) )
elif current_step == "add_element_data": elif current_step == "add_element_data":
inv = await TgItem.objects.aget(id=context.chat_data["inv"]) inv = await TgItem.objects.aget(id=context.chat_data["inv"])
tmc = inv.tmc.aget(id=context.chat_data["tmc"]) tmc = await inv.tmc.aget(id=context.chat_data["tmc"])
element = inv.element.aget(id=context.chat_data["element"]) element = await tmc.element.aget(id=context.chat_data["element"])
if update.message.photo: if update.message.photo:
element.photoid = update.message.photo[-1].file_id element.photoid = update.message.photo[-1].file_id