TelegramIndex-Fork/app/routes.py

75 lines
2.7 KiB
Python
Raw Normal View History

2020-08-17 09:44:02 +02:00
import logging
2020-08-13 05:58:55 +02:00
2020-08-10 09:27:52 +02:00
from aiohttp import web
2020-12-12 15:16:44 +01:00
from telethon.tl.types import Channel, Chat, User
2020-08-10 09:27:52 +02:00
2020-12-12 15:16:44 +01:00
from .config import index_settings
2020-08-12 12:39:29 +02:00
2020-08-10 09:27:52 +02:00
2020-08-17 09:44:02 +02:00
log = logging.getLogger(__name__)
async def setup_routes(app, handler):
2020-08-10 09:27:52 +02:00
h = handler
2020-08-17 09:44:02 +02:00
client = h.client
2020-12-17 04:16:10 +01:00
index_all = index_settings["index_all"]
index_private = index_settings["index_private"]
index_group = index_settings["index_group"]
index_channel = index_settings["index_channel"]
exclude_chats = index_settings["exclude_chats"]
include_chats = index_settings["include_chats"]
routes = [web.get("/", h.home)]
2020-08-17 09:44:02 +02:00
if index_all:
2020-12-17 04:16:10 +01:00
# print(await client.get_dialogs())
# dialogs = await client.get_dialogs()
# for chat in dialogs:
2020-08-17 09:44:02 +02:00
async for chat in client.iter_dialogs():
alias_id = None
if chat.id in exclude_chats:
continue
2020-09-20 18:38:52 +02:00
2020-12-12 15:16:44 +01:00
entity = chat.entity
2020-09-20 18:38:52 +02:00
2020-12-12 15:16:44 +01:00
if isinstance(entity, User) and not index_private:
2020-12-17 04:16:10 +01:00
log.debug(f"{chat.title}, private: {index_private}")
2020-12-12 15:16:44 +01:00
continue
elif isinstance(entity, Channel) and not index_channel:
2020-12-17 04:16:10 +01:00
log.debug(f"{chat.title}, channel: {index_channel}")
2020-12-12 15:16:44 +01:00
continue
elif isinstance(entity, Chat) and not index_group:
2020-12-17 04:16:10 +01:00
log.debug(f"{chat.title}, group: {index_group}")
2020-08-13 05:58:55 +02:00
continue
2020-12-12 15:16:44 +01:00
alias_id = h.generate_alias_id(chat)
p = "/{chat:" + alias_id + "}"
2020-12-17 04:16:10 +01:00
routes.extend(
[
web.get(p, h.index),
web.get(p + r"/logo", h.logo),
web.get(p + r"/{id:\d+}/view", h.info),
web.get(p + r"/{id:\d+}/download", h.download_get),
web.head(p + r"/{id:\d+}/download", h.download_head),
web.get(p + r"/{id:\d+}/thumbnail", h.thumbnail_get),
]
)
2020-12-12 15:16:44 +01:00
log.debug(f"Index added for {chat.id} at /{alias_id}")
2020-09-20 18:38:52 +02:00
2020-08-17 09:44:02 +02:00
else:
for chat_id in include_chats:
chat = await client.get_entity(chat_id)
2020-12-12 15:16:44 +01:00
alias_id = h.generate_alias_id(chat)
p = "/{chat:" + alias_id + "}"
2020-12-17 04:16:10 +01:00
routes.extend(
[
web.get(p, h.index),
web.get(p + r"/logo", h.logo),
web.get(p + r"/{id:\d+}/view", h.info),
web.get(p + r"/{id:\d+}/download", h.download_get),
web.head(p + r"/{id:\d+}/download", h.download_head),
web.get(p + r"/{id:\d+}/thumbnail", h.thumbnail_get),
]
)
2020-12-12 15:16:44 +01:00
log.debug(f"Index added for {chat.id} at /{alias_id}")
2020-12-17 04:16:10 +01:00
routes.append(web.view(r"/{wildcard:.*}", h.wildcard))
2020-08-12 12:39:29 +02:00
app.add_routes(routes)