eleksir/isida-4
1
Fork 0
Code Activity

add: initial AI support

Browse Source
This commit is contained in:
diSabler
2014-05-01 14:07:44 +04:00
parent 5ffac3efcf
commit e5295eb994
217 changed files with 78827 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

166
plugins/backup.py Normal file
View File

@@ -0,0 +1,166 @@
#!/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/>. #
# #
# --------------------------------------------------------------------------- #
backup_async = {}
def getMucItems(jid,affil,ns,back_id):
iqid = get_id()
if ns == xmpp.NS_MUC_ADMIN: i = xmpp.Node('iq', {'id': iqid, 'type': 'get', 'to':getRoom(jid)}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':affil})])])
else: i = xmpp.Node('iq', {'id': iqid, 'type': 'get', 'to':getRoom(jid)}, payload = [xmpp.Node('query', {'xmlns': ns},[])])
iq_request[iqid]=(time.time(),getMucItems_async,[ns,affil,back_id],ns)
sender(i)
def getMucItems_async(ns,affil,back_id,iq_stanza):
global backup_async
iq_stanza = iq_stanza[1][0]
if ns == xmpp.NS_MUC_ADMIN: backup_async[back_id][affil] = [[tmp.getAttr('jid'),['',tmp.getTagData('reason')][tmp.getTagData('reason') != None]] for tmp in iq_stanza.getTag('query',namespace=xmpp.NS_MUC_ADMIN).getTags('item')]
else: backup_async[back_id]['room_config'] = [[tmp.getAttr('var'),tmp.getAttr('type'),tmp.getTagData('value')] for tmp in iq_stanza.getTag('query',namespace=xmpp.NS_MUC_OWNER).getTag('x',namespace=xmpp.NS_DATA).getTags('field')]
def make_stanzas_array(raw_back,jid,affil):
stanza_array = []
try: make_stanza_jid_count = get_config_int(jid,'make_stanza_jid_count')
except: make_stanza_jid_count = 1
for tmp in range(0,len(raw_back),make_stanza_jid_count):
i = xmpp.Iq(typ='set',to=jid,xmlns=xmpp.NS_CLIENT)
i.setTag('query',namespace=xmpp.NS_MUC_ADMIN)
for t in raw_back[tmp:tmp+make_stanza_jid_count]:
i.getTag('query',namespace=xmpp.NS_MUC_ADMIN).\
setTag('item',attrs={'affiliation':affil,'jid':t[0]}).\
setTagData('reason',t[1])
stanza_array.append(i)
return stanza_array
def conf_backup(type, jid, nick, text):
global backup_async,conn
if len(text):
text = text.split(' ')
mode = text[0]
if mode == 'show':
a = os.listdir(back_folder % '')
b = []
for c in a:
if not c.endswith('.txt') and not c.endswith('.back'): b.append((c,os.path.getmtime(back_folder % c)))
if len(b): msg = '%s %s' % (L('Available copies:','%s/%s'%(jid,nick)), ', '.join(['%s (%s)' % (c[0],disp_time(c[1],'%s/%s'%(jid,nick))) for c in b]))
else: msg = L('Backup copies not found.','%s/%s'%(jid,nick))
elif mode == 'now':
if get_xtype(jid) != 'owner': msg = L('I need an owner affiliation for backup settings!','%s/%s'%(jid,nick))
else:
back_id = get_id()
backup_async[back_id] = {}
getMucItems(jid,'outcast',xmpp.NS_MUC_ADMIN,back_id)
getMucItems(jid,'member',xmpp.NS_MUC_ADMIN,back_id)
getMucItems(jid,'admin',xmpp.NS_MUC_ADMIN,back_id)
getMucItems(jid,'owner',xmpp.NS_MUC_ADMIN,back_id)
getMucItems(jid,'',xmpp.NS_MUC_OWNER,back_id)
bst = GT('backup_sleep_time')
while len(backup_async[back_id]) != 5 and not game_over: time.sleep(bst)
iqid = get_id()
i = xmpp.Node('iq', {'id': iqid, 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'admin', 'jid':getRoom(str(selfjid))},[])])])
sender(i)
backup_async[back_id]['alias'] = cur_execute_fetchall('select match ,cmd from alias where room=%s',(jid,))
backup_async[back_id]['bot_config'] = cur_execute_fetchall('select option ,value from config_conf where room=%s',(jid,))
backup_async[back_id]['acl'] = cur_execute_fetchall('select action,type,text,command,time,level from acl where jid=%s',(jid,))
backup_async[back_id]['rss'] = cur_execute_fetchall('select url, update ,type,time,hash from feed where room=%s',(jid,))
msg = L('Copying completed!','%s/%s'%(jid,nick))
msg += L('\nOwners: %s | Admins: %s | Members: %s | Banned: %s','%s/%s'%(jid,nick)) % (\
len(backup_async[back_id]['owner'])-1,\
len(backup_async[back_id]['admin'])+1,\
len(backup_async[back_id]['member']),\
len(backup_async[back_id]['outcast']))
msg += L('\nRoom config: %s | Bot config: %s | Aliases: %s | Acl\'s: %s | RSS: %s','%s/%s'%(jid,nick)) % (\
len(backup_async[back_id]['room_config']),\
len(backup_async[back_id]['bot_config']),\
len(backup_async[back_id]['alias']),\
len(backup_async[back_id]['acl']),\
len(backup_async[back_id]['rss']))
writefile(back_folder % unicode(jid),json.dumps(backup_async[back_id]))
elif mode == 'restore':
if len(text)>1:
if text[1] in os.listdir(back_folder % ''):
if text[1].endswith('.acl'):
raw_back = json.loads(readfile(back_folder % unicode(text[1])))
cnt = 0
for tmp in raw_back:
isit = cur_execute_fetchall('select action,type,text,command,time from acl where jid=%s and action=%s and type=%s and text=%s and level=%s',(jid,tmp[0],tmp[1],tmp[2],tmp[5]))
if not isit:
cur_execute('insert into acl values (%s,%s,%s,%s,%s,%s,%s)', tmp)
cnt += 1
msg = L('Restored %s action(s).','%s/%s'%(jid,nick)) % cnt
elif get_xtype(jid) != 'owner': msg = L('I need an owner affiliation for restore settings!','%s/%s'%(jid,nick))
else:
bst = GT('backup_sleep_time')
raw_back=json.loads(readfile(back_folder % unicode(text[1])))
for tmp in ['outcast','member','admin','owner']:
for t in make_stanzas_array(raw_back[tmp],jid,tmp):
sender(t)
time.sleep(bst)
i = xmpp.Iq(typ='set',to=jid,xmlns=xmpp.NS_CLIENT)
i.setTag('query',namespace=xmpp.NS_MUC_OWNER)
i.getTag('query',namespace=xmpp.NS_MUC_OWNER).setTag('x',namespace=xmpp.NS_DATA).setAttr('type','submit')
for tmp in raw_back['room_config']:
i.getTag('query',namespace=xmpp.NS_MUC_OWNER).getTag('x',namespace=xmpp.NS_DATA).\
setTag('field',attrs={'var':tmp[0],'type':tmp[1]}).\
setTagData('value',tmp[2])
sender(i)
time.sleep(bst)
sender(xmpp.Node('iq', {'id': get_id(), 'type': 'set', 'to':jid}, payload = [xmpp.Node('query', {'xmlns': xmpp.NS_MUC_ADMIN},[xmpp.Node('item',{'affiliation':'admin', 'jid':getRoom(unicode(selfjid))},[])])]))
time.sleep(bst)
cur_execute_fetchall('delete from config_conf where room=%s',(jid,))
for tmp in raw_back['bot_config']: cur_execute('insert into config_conf (room, option ,value) values (%s,%s,%s)',(jid,tmp[0],tmp[1]))
for tmp in raw_back['alias']:
cur_execute('delete from alias where room=%s and match =%s',(jid,tmp[0]))
cur_execute('insert into alias (room, match ,cmd) values (%s,%s,%s)',(jid,tmp[0],tmp[1]))
for tmp in raw_back['acl']:
isit = cur_execute_fetchall('select action,type,text,command,time from acl where jid=%s and action=%s and type=%s and text=%s and level=%s',(jid,tmp[0],tmp[1],tmp[2],tmp[5]))
if not isit: cur_execute('insert into acl values (%s,%s,%s,%s,%s,%s,%s)', tuple([jid]+list(tmp)))
for tmp in raw_back['rss']:
isit = cur_execute_fetchone('select * from feed where room=%s and url=%s',(jid,tmp[0]))
if not isit: cur_execute('insert into feeed (url, update ,type,time,room,hash) values (%s,%s,%s,%s,%s,%s)',(tmp[0],tmp[1],tmp[2],tmp[3],jid,tmp[4]))
msg = L('Restore completed.','%s/%s'%(jid,nick))
msg += L('\nOwners: %s | Admins: %s | Members: %s | Banned: %s','%s/%s'%(jid,nick)) % (\
len(raw_back['owner'])-1,\
len(raw_back['admin'])+1,\
len(raw_back['member']),\
len(raw_back['outcast']))
msg += L('\nRoom config: %s | Bot config: %s | Aliases: %s | Acl\'s: %s | RSS: %s','%s/%s'%(jid,nick)) % (\
len(raw_back['room_config']),\
len(raw_back['bot_config']),\
len(raw_back['alias']),\
len(raw_back['acl']),\
len(raw_back['rss']))
else: msg = L('Copy not found. Use key "show" for lisen available copies.','%s/%s'%(jid,nick))
else: msg = L('What do you want to restore?','%s/%s'%(jid,nick))
else: msg = L('Unknown item!','%s/%s'%(jid,nick))
else: msg = 'backup now|show|restore'
send_msg(type, jid, nick, msg)
global execute
execute = [(8, 'backup', conf_backup, 2, 'Backup/restore conference settings.\nbackup show|now|restore\nshow - show available copies\nnow - backup current conference\nrestore name_conference - restore settings name_conference in current conference')]