headpat: rework style

This commit is contained in:
Dan Elkouby 2018-05-13 13:33:40 +03:00
parent dbf561880e
commit b72ddba35e
1 changed files with 11 additions and 12 deletions

View File

@ -12,31 +12,30 @@ import urllib.request
import urllib.parse
pats = []
user = None
oops = 'OOPSIE WOOPSIE!! Uwu We madea fucky wucky!! A wittle fucko boingo! '\
'The code monkeys at our headquarters are working VEWY HAWD to fix this!'
oops = "OOPSIE WOOPSIE!! Uwu We madea fucky wucky!! A wittle fucko boingo! " \
"The code monkeys at our headquarters are working VEWY HAWD to fix " \
"this!"
@borg.on(events.NewMessage)
async def on_pat(event):
global user
if user is None:
user = (await borg.get_me()).username or ''
user = borg.me.username
if not user or not re.match(fr'(?i)/headpat@{user}', event.raw_text):
if not user or not re.match(fr"(?i)/headpat@{user}", event.raw_text):
return
global pats
if not pats:
try:
pats = json.loads(urllib.request.urlopen(urllib.request.Request(
'http://headp.at/js/pats.json',
headers={'User-Agent': 'Mozilla/5.0 (X11; U; Linux i686) '
'Gecko/20071127 Firefox/2.0.0.11'}
)).read().decode('utf-8'))
"http://headp.at/js/pats.json",
headers={"User-Agent": "Mozilla/5.0 (X11; U; Linux i686) "
"Gecko/20071127 Firefox/2.0.0.11"}
)).read().decode("utf-8"))
except Exception as e:
print(e)
await event.reply(oops)
return
await event.reply(f'[Pat!](https://headp.at/pats/{urllib.parse.quote(random.choice(pats))})')
choice = urllib.parse.quote(random.choice(pats))
await event.reply(f"[Pat!](https://headp.at/pats/{choice})")