octospacc/TelegramIndex-Fork
1
0
Fork 0
mirror of https://gitlab.com/octospacc/TelegramIndex-Fork.git synced 2025-06-05 22:09:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

new changes

Browse Source
This commit is contained in:
odysseusmax
2020-12-12 19:46:44 +05:30
parent 2fffd9353d
commit 2c0160b21e
15 changed files with 539 additions and 554 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
app/routes.py
View File

@ -3,75 +3,70 @@ import string
import logging
from aiohttp import web
from telethon.tl.types import Channel, Chat, User
from .config import index_settings, alias_ids, chat_ids
from .config import index_settings
log = logging.getLogger(__name__)
def generate_alias_id(chat):
chat_id = chat.id
title = chat.title
while True:
alias_id = ''.join([random.choice(string.ascii_letters + string.digits) for _ in range(len(str(chat_id)))])
if alias_id in alias_ids:
continue
alias_ids.append(alias_id)
chat_ids.append({
'chat_id': chat_id,
'alias_id': alias_id,
'title': title
})
return alias_id
async def setup_routes(app, handler):
h = handler
client = h.client
p = r"/{chat:[^/]+}"
routes = [
web.get('/', h.home),
web.post('/otg', h.dynamic_view),
web.get('/otg', h.otg_view),
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),
web.view(r'/{wildcard:.*}', h.wildcard)
]
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)
]
if index_all:
#print(await client.get_dialogs())
async for chat in client.iter_dialogs():
alias_id = None
if chat.id in exclude_chats:
continue
if chat.is_user:
if index_private:
alias_id = generate_alias_id(chat)
elif chat.is_channel:
if index_channel:
alias_id = generate_alias_id(chat)
else:
if index_group:
alias_id = generate_alias_id(chat)
entity = chat.entity
if not alias_id:
if isinstance(entity, User) and not index_private:
print(f'{chat.title}, private: {index_private}')
continue
log.debug(f"Index added for {chat.id} :: {chat.title} at /{alias_id}")
elif isinstance(entity, Channel) and not index_channel:
print(f'{chat.title}, channel: {index_channel}')
continue
elif isinstance(entity, Chat) and not index_group:
print(f'{chat.title}, group: {index_group}')
continue
alias_id = h.generate_alias_id(chat)
p = "/{chat:" + alias_id + "}"
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),
])
log.debug(f"Index added for {chat.id} at /{alias_id}")
else:
for chat_id in include_chats:
chat = await client.get_entity(chat_id)
alias_id = generate_alias_id(chat)
log.debug(f"Index added for {chat.id} :: {chat.title} at /{alias_id}")
alias_id = h.generate_alias_id(chat)
p = "/{chat:" + alias_id + "}"
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),
])
log.debug(f"Index added for {chat.id} at /{alias_id}")
routes.append(web.view(r'/{wildcard:.*}', h.wildcard))
app.add_routes(routes)