Merge branch 'master' of https://github.com/udf/uniborg

stdplugins/sp_prog_rename.py

@@ -1,10 +1,9 @@
 import asyncio
+import re
+
 from telethon import events
 from telethon.tl.functions.channels import EditTitleRequest
 
-import telethon.utils
-
-import re
 
 prog_tech_id = 1040270887
 prog_tech_channel = None
@@ -12,6 +11,16 @@ default_title = "Programming & Tech"
 lock = asyncio.Lock()
 
 
+def fix_title(s):
+    # Ideally this would be a state machine, but ¯\_(ツ)_/¯
+    def replace(m):
+        token = m.group(1)
+        if token.lower() == 'and':
+            token = '&'
+        return token.capitalize() + (' ' if m.group(2) else '')
+    return re.sub(r'(\S+)(\s+)?', replace, s)
+
+
 async def edit_title(title):
     global prog_tech_channel
     if prog_tech_channel is None:
@@ -24,7 +33,7 @@ async def edit_title(title):
 @borg.on(events.NewMessage(
     pattern=re.compile(r"(?i)programming (?:&|and) (.+)"), chats=prog_tech_id))
 async def on_name(event):
-    new_topic = event.pattern_match.group(1).title()
+    new_topic = fix_title(event.pattern_match.group(1))
     new_title = f"Programming & {new_topic}"
     if "Tech" not in new_title:
         new_title += " & Tech"