chat_moderator_bot/utils/defs.py

44 lines
1.1 KiB
Python
Raw Permalink Normal View History

2024-07-17 20:35:45 +00:00
import logging
from aiogram.types import Message
import pandas
from utils.db import Postgres
async def delete_msg(msg: Message) -> None:
"""
Безопасно удаляет сообщение
:param msg: Message
:return: True, если текст является ссылкой, иначе False.
"""
try:
await msg.delete()
except:
pass
async def create_xlsx() -> str:
"""
Составляет xlsx файл с данными бан слова
:return: Путь к готовому xlsx файлу
"""
try:
p = Postgres()
ban_words = await p.get_data(
table_name='ban_words'
)
table_dict = {'ID': [], 'Слово': []}
for ban_word in ban_words:
table_dict['ID'].append(ban_word['id'])
table_dict['Слово'].append(ban_word['word'])
df = pandas.DataFrame(table_dict)
file_path = 'data/ban_words.xlsx'
df.to_excel(file_path, index=False)
return file_path
except Exception as e:
logging.error('Ошибка в create_xlsx', e)