60 lines
3.0 KiB
Python
60 lines
3.0 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# --------------------------------------------------------------------------- #
|
|
# #
|
|
# Plugin for iSida Jabber Bot #
|
|
# Copyright (C) Vit@liy <vitaliy@root.ua> #
|
|
# #
|
|
# This program is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>. #
|
|
# #
|
|
# --------------------------------------------------------------------------- #
|
|
|
|
def sokr(type, jid, nick, text):
|
|
target = ''
|
|
text = text.strip()
|
|
if not text: msg = L('What?','%s/%s'%(jid,nick))
|
|
else:
|
|
tmp_target = re.search('(\A| )\d+?(-\d+?)?(\Z| )', text)
|
|
if tmp_target:
|
|
target = tmp_target.group().strip()
|
|
text = text.replace(target, ' ')
|
|
text = reduce_spaces_all(text)
|
|
if re.match('[a-zA-Z]+\Z', text):
|
|
data = load_page('http://www.abbreviations.com/%s' % text)
|
|
results = re.findall('<p class="desc">(.+?)</p>', data)
|
|
results = [i for k,i in enumerate(results) if results.index(i) == k]
|
|
else:
|
|
data = load_page("http://www.sokr.ru/search/?", {'abbr': text.encode('utf-8'), 'abbr_exact': '1'})
|
|
results = re.findall('em class="got_clear">.+?</em></a>.+?<p class="value">(.+?)</p>' , data, re.S)
|
|
cr = len(results)
|
|
if not results: msg = L('I don\'t know!','%s/%s'%(jid,nick))
|
|
else:
|
|
if not target:
|
|
if cr == 1: target = '1'
|
|
elif cr < 6: target = '1-%s' % cr
|
|
else: target = '1-5'
|
|
try: n1 = n2 = int(target)
|
|
except: n1, n2 = map(int, target.split('-'))
|
|
if 0 < n1 <= n2 <= cr:
|
|
msg = L('Total found %s matches. Result(s) %s:\n','%s/%s'%(jid,nick)) % (cr, target)
|
|
msg += '\n'.join(['%s. %s' % (i[0]+n1, i[1].replace('\n',' ')) for i in enumerate(results[n1-1: n2])]).decode('utf8')
|
|
msg = msg.replace('<br>', '').replace('\r','')
|
|
else: msg = L('I don\'t know!','%s/%s'%(jid,nick))
|
|
send_msg(type, jid, nick, msg)
|
|
|
|
global execute
|
|
|
|
execute = [(3, 'sokr', sokr, 2, 'Abbreviations.\nExamples: sokr abbr, sokr 6 abbr, sokr 3-7 abbr')]
|