160 lines
5.9 KiB
Python
160 lines
5.9 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# --------------------------------------------------------------------------- #
|
|
# #
|
|
# Plugin for iSida Jabber Bot #
|
|
# Copyright (C) diSabler <dsy@dsy.name> #
|
|
# #
|
|
# 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/>. #
|
|
# #
|
|
# --------------------------------------------------------------------------- #
|
|
|
|
# [u'devel@conference.jabber.ru', u'sulci', u'moderator', u'admin', u'sulci@jabber.ru/Ocaml']
|
|
|
|
AI_PREFIX = '!ai_'
|
|
|
|
AI_PREV = {}
|
|
AI_PHRASES_FOLDER = data_folder % 'ai/%s'
|
|
AI_PHRASES_FILES = [t for t in os.listdir(AI_PHRASES_FOLDER % '') if t.endswith('.txt') and os.path.isfile(AI_PHRASES_FOLDER % t)]
|
|
AI_PHRASES = {}
|
|
|
|
for PH_LANG in AI_PHRASES_FILES:
|
|
PH_L = PH_LANG.split('.',1)[0]
|
|
PHRASES = readfile(AI_PHRASES_FOLDER % PH_LANG).decode('utf-8').replace('\r','').split('\n')
|
|
for LINE in PHRASES:
|
|
if LINE.strip() and not LINE.strip().startswith('#') and '\t' in LINE:
|
|
if not AI_PHRASES.has_key(PH_L): AI_PHRASES[PH_L] = [LINE.split('\t',1)]
|
|
else: AI_PHRASES[PH_L].append(LINE.split('\t',1))
|
|
|
|
def AI_UNIQUE(M):
|
|
m = []
|
|
for t in M:
|
|
if t not in m: m.append(t)
|
|
return m
|
|
|
|
def AI_RAW(CMD, RAW, RN):
|
|
CMD = CMD.replace(AI_PREFIX,'',1)
|
|
if CMD.startswith('drink') and os.path.isfile(date_file):
|
|
RAW_SPLIT = RAW.lower().replace('?','').strip().split()
|
|
DRINK = drink_dmas + drink_mmas1 + drink_mmas2 + drink_wday + drink_lday
|
|
INIT_DRINK = re.findall('[0-9]+', RAW, re.S)
|
|
if INIT_DRINK: INIT_DRINK = ['.'.join(INIT_DRINK)]
|
|
for t in DRINK:
|
|
DRINK_LOCALE = L(t, RN).lower()
|
|
if DRINK_LOCALE in RAW_SPLIT: INIT_DRINK.append(DRINK_LOCALE)
|
|
if len(INIT_DRINK) >= 2: return CMD.replace('{RAW}', ' '.join(AI_UNIQUE(INIT_DRINK))), True
|
|
else:
|
|
ddate = readfile(date_file).decode('UTF').lower().split('\n')
|
|
for DATE in ddate:
|
|
DATE_SPLIT = DATE.split()
|
|
for R in RAW_SPLIT:
|
|
if len(R) >= 3 and R in DATE_SPLIT: INIT_DRINK.append(R)
|
|
if INIT_DRINK: return CMD.replace('{RAW}', ' '.join(AI_UNIQUE(INIT_DRINK))), True
|
|
return '', False
|
|
|
|
def AI_PARSE(room, jid, nick, type, text):
|
|
if not get_config(room,'ai'): return
|
|
LOC = get_L_('%s/%s' % (room,nick))
|
|
if not AI_PHRASES.has_key(LOC): return
|
|
access_mode = get_level(room,nick)[0]
|
|
if access_mode < 4: return
|
|
TEXT = ' %s ' % text.lower()
|
|
SCORE, CMD = 1.0, []
|
|
for PHRASES in AI_PHRASES[LOC]:
|
|
CUR_SCORE = AI_RATING(PHRASES[1].split('||')[0], TEXT, room)
|
|
if CUR_SCORE > SCORE: SCORE, CMD = CUR_SCORE, [PHRASES]
|
|
elif CUR_SCORE == SCORE: CMD.append(PHRASES)
|
|
if not CMD: return
|
|
nowname = getResourse(cur_execute_fetchone('select room from conference where room ilike %s',('%s/%%'%room,))[0])
|
|
if nowname == nick: return
|
|
pprint('AI score %s|%s|%s' % (SCORE, '|'.join([t[0] for t in CMD]), text),'cyan')
|
|
CMD = random.choice(CMD)
|
|
CMD_RAW = CMD[0].split()[0]
|
|
pprint('AI selected %s' % CMD_RAW,'cyan')
|
|
if cur_execute_fetchone('select * from commonoff where room=%s and cmd=%s',(room,CMD_RAW)): return
|
|
ptype = ''
|
|
for t in comms:
|
|
if t[1] == CMD_RAW:
|
|
ptype = t[-1]
|
|
break
|
|
if 'nick' in ptype and '||' in CMD[1]:
|
|
for t in CMD[1].split('||')[1].split('|'):
|
|
text = text.replace(t,nick)
|
|
|
|
PRM = AI_PARAMETER(text, ptype, room)
|
|
|
|
if 'nick' in ptype:
|
|
PP = None
|
|
for t in PRM:
|
|
if t:
|
|
PP = True
|
|
break
|
|
if not PP: PRM.append(nick)
|
|
|
|
COMMAND = CMD[0]
|
|
if COMMAND.startswith(AI_PREFIX):
|
|
for PHRASES in AI_PHRASES[LOC]:
|
|
if PHRASES[0] == COMMAND:
|
|
for t in PHRASES[1].split('||')[0].lower().split('|'):
|
|
TEXT = TEXT.lower().replace(t,'')
|
|
COMMAND, IS_PARSED = AI_RAW(COMMAND, TEXT, '%s/%s'%(jid,nick))
|
|
if IS_PARSED:
|
|
com_parser(access_mode, nowname, type, room, nick, COMMAND, jid)
|
|
return True
|
|
else:
|
|
COMMAND = COMMAND.replace('{TOMORROW}',time.strftime('%d.%m', time.localtime(time.time()+86400)))
|
|
WAS_PARAM = False
|
|
for PARAM in PRM:
|
|
if PARAM:
|
|
if time_nolimit: time.sleep(time_nolimit)
|
|
com_parser(access_mode, nowname, type, room, nick, COMMAND.replace('{PAR}',PARAM), jid)
|
|
WAS_PARAM = True
|
|
if not WAS_PARAM:
|
|
if time_nolimit: time.sleep(time_nolimit)
|
|
com_parser(access_mode, nowname, type, room, nick, COMMAND, jid)
|
|
return True
|
|
|
|
def AI_RATING(s, text, room):
|
|
r,s = 0.0,s.split('|')
|
|
for k in s:
|
|
if k in text: r += 0.34
|
|
if k in AI_PREV.get(room, ''): r += 0.34
|
|
return r
|
|
|
|
def AI_PARAMETER(body, type, room):
|
|
P_NICK, P_JID, P_SERVER = [], [], []
|
|
|
|
if 'nick' in type:
|
|
NICKS = [t[1] for t in megabase if t[0]==room]
|
|
for NICK in NICKS:
|
|
if NICK in body:
|
|
P_NICK.append(NICK)
|
|
|
|
if 'jid' in type:
|
|
JID = re.findall(u'[-a-z0-9а-яё_]+@[-0-9a-z\.]+',body,re.S|re.I|re.U)
|
|
if JID: P_JID += JID
|
|
|
|
if 'server' in type:
|
|
SERVER = re.findall(u'[-0-9a-zа-яё\.]+\.[-a-zа-я]{2,}',body,re.S|re.I|re.U)
|
|
if SERVER: P_SERVER += SERVER
|
|
|
|
RESULT = []
|
|
for t in P_NICK + P_JID + P_SERVER:
|
|
if t not in RESULT: RESULT.append(t)
|
|
|
|
return RESULT
|
|
|
|
gmessage.append(AI_PARSE)
|