proDream/Notion_Bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Обновление

Browse Source
This commit is contained in:
Ivan Ashikhmin 2024-04-28 20:05:23 +04:00
parent 59c685ba49
commit bd989270bc
2 changed files with 39 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/handlers/message.py
View File

@ -3,7 +3,7 @@ from aiogram.types import Message, MessageEntity, PhotoSize, ReactionTypeEmoji
from app.utils.send_to_notion import send_to_notion
async def parse_message(message: Message):
await send_to_notion(message)
async def parse_message(message: Message, album=None):
await send_to_notion(message, album)
await message.react([ReactionTypeEmoji(emoji="👌")])

61
app/utils/send_to_notion.py
View File

@ -9,27 +9,25 @@ from aiogram.types import Message, MessageEntity, PhotoSize
from app.settings import notion, bot, secrets
async def send_to_notion(message: Message):
async def send_to_notion(message: Message, album):
imgur_client = Imgur(
{"client_id": secrets.imgur_client_id.get_secret_value()}
)
pattern = r"(https?://[^\s]+|t\.me/[^\s]+)"
image_url = None
photos = []
if message.caption:
links2 = re.findall(pattern, message.caption)
text: str = message.caption
photos: PhotoSize = [photo.file_id for photo in message.photo]
links: MessageEntity = [
link.url for link in message.caption_entities if link.type == "text_link"
]
if photos:
file_name = f"images/{photos[0]}.jpg"
await bot.download(message.photo[-1], destination=file_name)
imgur_client = Imgur(
{"client_id": secrets.imgur_client_id.get_secret_value()}
)
image = imgur_client.image_upload(
path.realpath(file_name), "Untitled", "My first image upload"
)
image_url = image["response"]["data"]["link"]
os.remove(file_name)
if album:
photos: PhotoSize = [photo.photo[0].file_id for photo in album]
elif message.photo:
photos: PhotoSize = [message.photo[0].file_id]
else:
text: str = message.text
links2 = re.findall(pattern, message.text)
@ -43,7 +41,7 @@ async def send_to_notion(message: Message):
links.extend(links2)
links = set(links)
properties = {
"Name": {"title": [{"text": {"content": text[: text.index("\n")]}}]},
"Name": {"title": [{"text": {"content": text[: text.find("\n") or -1]}}]},
"Text": {"rich_text": [{"text": {"content": text}}]},
"Added at": {
"date": {
@ -58,16 +56,31 @@ async def send_to_notion(message: Message):
properties[f"Link{i}"] = {"url": link}
cover = None
if photos:
properties["Image"] = {
"files": [
{
"name": "image.jpg",
"type": "external",
"external": {"url": image_url},
}
]
}
cover = {"type": "external", "external": {"url": image_url}}
for i, photo in enumerate(photos, start=1):
if i > 9:
break
file_name = f"images/{photo}.jpg"
await bot.download(photo, destination=file_name)
image = imgur_client.image_upload(
path.realpath(file_name), "Untitled", "My first image upload"
)
image_url = image["response"]["data"]["link"]
os.remove(file_name)
if i == 1:
cover = {"type": "external", "external": {"url": image_url}}
properties[f"Image{i}"] = {
"files": [
{
"name": "image.jpg",
"type": "external",
"external": {"url": image_url},
}
]
}
icon = {"type": "emoji", "emoji": "🎉"}
parent = {
"type": "database_id",