kate
/
uniborg
forked from uniborg/uniborg
1
0
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Merge branch 'kate' of github.com:udf/uniborg into kate

This commit is contained in:
udf 2019-04-29 20:09:09 +02:00
parent 9f61d75ec9 0948bd1d5b
commit fc78927fa6
Signed by: kate
GPG Key ID: F17AECB58C3A3096
1 changed files with 48 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
stdplugins/sp_prog_rename.py
View File

@ -1,20 +1,30 @@
import asyncio
import re
import html
from dataclasses import dataclass
from telethon import events
from telethon import events, utils
from telethon.tl.functions.channels import EditTitleRequest
from telethon.errors.rpcerrorlist import ChatNotModifiedError
MULTI_EDIT_TIMEOUT = 80
REVERT_TIMEOUT = 2 * 60 * 60
DEFAULT_TITLES = {
# 1040270887: 'Programming & Tech',
1384391544: 'Programming & Tech for girls',
1286178907: 'test supergroup'
}
@dataclass
class Group:
id: int
title: str
rename_lock = asyncio.Lock()
revert_task = None
revert_task: asyncio.Task = None
GROUPS = {group.id: group for group in (
# Group(1040270887, 'Programming & Tech'),
Group(1384391544, 'Programming & Tech for girls'),
Group(1286178907, 'test supergroup')
)}
def fix_title(s):
@ -36,35 +46,50 @@ async def edit_title(chat, title):
pass # Everything is ok
async def wait_and_revert(chat_id, timeout):
async def wait_and_revert(chat_id, title, timeout):
await asyncio.sleep(timeout)
await edit_title(chat_id, DEFAULT_TITLES[chat_id])
await edit_title(chat_id, title)
@borg.on(events.NewMessage(
pattern=re.compile(r"(?i)programming (?:&|and) (.+)"),
chats=list(DEFAULT_TITLES.keys())))
chats=list(GROUPS.keys())))
async def on_name(event):
global revert_task
new_topic = fix_title(event.pattern_match.group(1))
new_title = f"Programming & {new_topic}"
if "Tech" not in new_title:
new_title += " & Tech"
if len(new_title) > 255 or rename_lock.locked():
group = GROUPS[utils.get_peer_id(event.chat_id, False)] # Thanks Lonami
logger.info(f'{event.from_id} in {group.id} '
f'requested a title change to {new_title}')
if len(new_title) > 255:
logger.info('Not changing group title because new title is too long')
return
if group.rename_lock.locked():
logger.info('Not changing group title because the rename lock is already held')
return
with (await group.rename_lock):
logger.info(f'Changing group title to {new_title}')
await event.respond(
f'<a href="tg://user?id={event.from_id}">{html.escape(event.sender.first_name)}</a>'
' changed the group title!',
parse_mode='html'
)
with (await rename_lock):
await edit_title(event.chat_id, new_title)
logger.info(f'Holding rename lock for {MULTI_EDIT_TIMEOUT} seconds')
await asyncio.sleep(MULTI_EDIT_TIMEOUT)
if revert_task and not revert_task.done():
revert_task.cancel()
revert_task = asyncio.create_task(
wait_and_revert(event.chat_id, REVERT_TIMEOUT))
if group.revert_task and not group.revert_task.done():
logger.info('Cancelling previous revert task')
group.revert_task.cancel()
logger.info('Creating revert task')
group.revert_task = asyncio.create_task(wait_and_revert(
event.chat_id,
group.title,
REVERT_TIMEOUT
))