happybot/ichi.py
zgrep 8b6c9697e0 The happybot revolves around the Earth.
The Earth revolves around the God.
The God gives you a world full of viral diseases.
You curse God.
You revolve around God.
God revolves around a door.
The door opens.
This is going nowhere, you can stop reading now.

I said you could stop.

Okay, that's it, I'm really ending it now.
2018-08-14 03:24:50 -04:00

77 lines
1.9 KiB
Python

#!/usr/bin/env python3
from sys import argv
import re
import requests as rq
from lxml import html
from subprocess import Popen, PIPE
if len(argv) != 2:
print('Usage:', argv[0], '#channel')
exit(1)
chan = argv[1]
def cmd(args):
proc = Popen(args, stdout=PIPE)
while True:
line = proc.stdout.readline()
if line:
try:
yield str(line[:-1], 'utf-8', 'ignore')
except:
pass
else:
break
fdir = '/home/zgrep/offtopiabday/irc.freenode.net/' + chan
fin = fdir + '/in'
fout = fdir + '/out'
def irc():
for line in cmd(['tail', '-n', '0', '-f', fout]):
date, time, nick, line = line.split(' ', 3)
print(line)
m = re.match(r'(?i)((happy|hate)bot[:,]\s+ichi|ハピロボット[:,])\s(.+)', line)
if m:
q = m.group(3)
print('Matched!', q)
r = get(q)
print('Result!', r)
with open(fin, 'w') as fh:
fh.write(r + '\n')
omitnum = 3
def ichi(q, extra=False):
r = rq.get('http://ichi.moe/cl/qr/', params={'q': q})
if r.status_code != 200:
return 'Non-200 status code. Bug zgrep.'
t = html.fromstring(r.content)
ichi = ''.join(t.xpath('//span[@class="ds-text"]//text()'))
more = ''
if extra:
e = t.xpath('//span[@class="gloss-desc"]//text()')
more = ' - '.join(e[:omitnum])
if len(e) > omitnum:
more += ' - \x1dmore omitted\x0f'
if more:
more = ' (' + more + ')'
return '"' + ichi + '"' + more
def kanji(q):
r = rq.get('http://ichi.moe/cl/kanji/', params={'q': q})
if r.status_code != 200:
return None
if 'No kanji found' in r.text:
return None
t = html.fromstring(r.content)
return 'Not yet implemented.'
def get(q):
if len(q) == 1:
# k = kanji(q)
# if k: return k
return ichi(q, True)
return ichi(q)
irc()