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

fix logic by using a task to schedule title reverting

This commit is contained in:
udf 2019-04-02 22:46:36 +02:00
parent fd20c8804b
commit a3981e0994
Signed by: kate
GPG Key ID: F17AECB58C3A3096
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
stdplugins/sp_prog_rename.py
View File

@ -5,12 +5,13 @@ from telethon import events
from telethon.tl.functions.channels import EditTitleRequest from telethon.tl.functions.channels import EditTitleRequest
from telethon.errors.rpcerrorlist import ChatNotModifiedError from telethon.errors.rpcerrorlist import ChatNotModifiedError
MULTI_EDIT_TIMEOUT = 10 #80 MULTI_EDIT_TIMEOUT = 80
REVERT_TIMEOUT = 25 #2 * 60 * 60 REVERT_TIMEOUT = 2 * 60 * 60
CHANNEL_ID = 1286178907 #1040270887 CHANNEL_ID = 1040270887
DEFAULT_TITLE = "Programming & Tech" DEFAULT_TITLE = "Programming & Tech"
prog_tech_channel = None prog_tech_channel = None
rename_lock = asyncio.Lock() rename_lock = asyncio.Lock()
revert_task = None
def fix_title(s): def fix_title(s):
@ -38,16 +39,21 @@ async def edit_title(title):
async def wait_for_delete(deleted_fut, timeout): async def wait_for_delete(deleted_fut, timeout):
try: try:
await asyncio.wait_for(deleted_fut, timeout) await asyncio.wait_for(deleted_fut, timeout)
await edit_title(DEFAULT_TITLE)
return True return True
except asyncio.TimeoutError: except asyncio.TimeoutError:
pass pass
return False return False
async def wait_and_revert(deleted_fut, timeout):
await wait_for_delete(deleted_fut, timeout)
await edit_title(DEFAULT_TITLE)
@borg.on(events.NewMessage( @borg.on(events.NewMessage(
pattern=re.compile(r"(?i)programming (?:&|and) (.+)"), chats=CHANNEL_ID)) pattern=re.compile(r"(?i)programming (?:&|and) (.+)"), chats=CHANNEL_ID))
async def on_name(event): async def on_name(event):
global revert_task
new_topic = fix_title(event.pattern_match.group(1)) new_topic = fix_title(event.pattern_match.group(1))
new_title = f"Programming & {new_topic}" new_title = f"Programming & {new_topic}"
if "Tech" not in new_title: if "Tech" not in new_title:
@ -63,9 +69,11 @@ async def on_name(event):
func=lambda e: e.deleted_id == event.message.id func=lambda e: e.deleted_id == event.message.id
)) ))
if await wait_for_delete(asyncio.shield(deleted_fut), MULTI_EDIT_TIMEOUT): if await wait_for_delete(asyncio.shield(deleted_fut), MULTI_EDIT_TIMEOUT):
await edit_title(DEFAULT_TITLE)
await asyncio.sleep(MULTI_EDIT_TIMEOUT) await asyncio.sleep(MULTI_EDIT_TIMEOUT)
return return
if await wait_for_delete(deleted_fut, REVERT_TIMEOUT) or rename_lock.locked():
return if revert_task and not revert_task.done():
with (await rename_lock): revert_task.cancel()
await edit_title(DEFAULT_TITLE)
revert_task = asyncio.create_task(wait_and_revert(deleted_fut, REVERT_TIMEOUT))