def add(
db: Session,
req_body: schemas.AddButton,
) -> models.Message:
"""Make and add button to message."""
msg = message.get_check_user(db, req_body)
next_number = len(msg.own_buttons)
new_button = models.Button(
text=req_body.text,
parrent_message=msg,
number=next_number,
)
if req_body.link_to_msg_id:
req_msg = schemas.GetUserMsg(
tg_id=req_body.tg_id,
msg_id=req_body.link_to_msg_id,
)
req_msg.msg_id = req_body.link_to_msg_id
next_msg = message.get_check_user(db, req_msg)
new_button.next_message = next_msg
db.add(new_button)
db.commit()
db.refresh(msg)
return msg
def edit(
db: Session,
req_body: schemas.EditButton,
) -> models.Message:
"""Edit button."""
msg = message.get_check_user(db, req_body)
btn = get_check_user(db, req_body)
if req_body.text:
btn.text = req_body.text
if req_body.link_to_msg_id:
req_msg = schemas.GetUserMsg(
tg_id=req_body.tg_id,
msg_id=req_body.link_to_msg_id,
)
req_msg.msg_id = req_body.link_to_msg_id
next_msg = message.get_check_user(db, req_msg)
btn.next_message = next_msg
if req_body.move:
sorted_btns = sorted(msg.own_buttons, key=lambda x: x.number)
if req_body.move > 0 and len(sorted_btns) > btn.number + 1:
next_btn = sorted_btns[btn.number + 1]
next_btn.number = next_btn.number - 1
btn.number = btn.number + 1
elif req_body.move < 0 and btn.number != 0:
prev_btn = sorted_btns[btn.number - 1]
prev_btn.number = prev_btn.number + 1
btn.number = btn.number - 1
db.commit()
db.refresh(msg)
return msg
def rm(
db: Session,
req_body: schemas.GetMsgButton,
) -> models.Message:
"""Remove button."""
msg = message.get_check_user(db, req_body)
btn = get_check_user(db, req_body)
db.delete(btn)
db.commit()
db.refresh(msg)
return msg
def get_check_user(
db: Session,
req_body: schemas.GetMsgButton,
) -> models.Button:
"""Get button."""
msg = message.get_check_user(db, req_body)
btn = db.query(models.Button).filter_by(
id=req_body.button_id,
parrent_message=msg,
).first()
if btn:
return btn
err_msg = 'There is not button id - {button_id} for message {msg_id}'.format(
button_id=req_body.button_id,
msg_id=req_body.msg_id,
)
logger.error(err_msg)
raise ValueError(err_msg)
def make(
db: Session,
message_id: int,
tg_id: int,
file_data: UploadFile,
):
msg_req_body = schemas.GetUserMsg(
msg_id=message_id,
tg_id=tg_id,
)
usr_message = message.get_check_user(db, msg_req_body)
if usr_message.media:
media = usr_message.media
else:
media = models.Media(parrent_message=usr_message, )
db.add(media)
media.content_type = file_data.content_type
media.file_data = file_data.file.read()
db.commit()
db.refresh(media)
media.make_uid()
db.commit()