1
Files
isida-4/plugins/watcher.py
2014-05-01 14:07:44 +04:00

109 lines
4.6 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/>. #
# #
# --------------------------------------------------------------------------- #
watch_time = time.time()
watch_count = 0
watch_reset = True
watch_last_activity = {}
watch_activity = {}
def connect_watch():
global iq_request, watch_time, game_over, watch_count, bot_exit_type, watch_reset, plugins_reload
if GT('watcher_self_ping') and (time.time() - watch_time) > GT('watch_size'):
watch_time = time.time()
watch_count += 1
watch_reset = True
iqid = get_id()
i = xmpp.Node('iq', {'id': iqid, 'type': 'get', 'to':selfjid}, payload = [xmpp.Node('ping', {'xmlns': xmpp.NS_URN_PING},[])])
iq_request[iqid]=(time.time(),watcher_reset,['chat',god,'',''],xmpp.NS_URN_PING)
sender(i)
to = GT('timeout') - 10
if to <= 10: to = 600
while to > 0 and not game_over and not plugins_reload:
to -= 1
time.sleep(1)
if game_over or plugins_reload: return
if watch_reset:
pprint('Restart by watcher\'s timeout!','red')
bot_exit_type, game_over = 'restart', True
time.sleep(2)
def watch_room_activity():
if not GT('watcher_room_activity'): return
global watch_last_activity
to = int(time.time())-GT('watch_activity_timeout')
cnf = cur_execute_fetchall('select room from conference;')
if cnf:
for t in cnf:
tmp = t[0]
try: cw = watch_last_activity[getRoom(tmp)]
except: cw = to
if cw < to:
watch_last_activity[getRoom(tmp)] = int(time.time())
text = tmp
if '\n' in text: text,passwd = text.split('\n',2)
else: passwd = ''
zz = join(text, passwd)
while unicode(zz)[:3] == '409':
time.sleep(1)
text += '_'
zz = join(text, passwd)
time.sleep(1)
pprint('Low activity! Try rejoin into %s' % text,'white')
def watcher_reset(a,b,c,d,e):
global watch_reset
watch_reset = None
def c_watcher(type, jid, nick): send_msg(type, jid, nick, L('Timeout for ask: %s | Timeout for answer: %s | Last ask: %s | Total checks: %s','%s/%s'%(jid,nick)) % (GT('watch_size'),GT('timeout'),un_unix(int(time.time() - watch_time),'%s/%s'%(jid,nick)),watch_count))
def activity_watch(type, jid, nick):
rooms = cur_execute_fetchall("select split_part(room,'/',1) from conference;")
cnf = [[watch_activity[t],t] for t in watch_activity.keys()]
cnf.sort(reverse=1)
tt = int(time.time())
msg = '\n'.join(['%s - %s' % (t[1],un_unix(tt-t[0],'%s/%s'%(jid,nick))) for t in cnf])
rooms = ['%s - %s' % (t[0],L('Unknown','%s/%s'%(jid,nick))) for t in rooms if t[0] not in watch_activity.keys()]
if rooms: msg = '%s\n%s' % (msg,'\n'.join(rooms))
msg = L('Last activity in room(s):\n%s','%s/%s'%(jid,nick)) % msg
if type == 'groupchat':
send_msg('chat', jid, nick, msg)
msg = L('Send for you in private','%s/%s'%(jid,nick))
send_msg(type, jid, nick, msg)
def connect_watch_uni(room,jid,nick,type,mass):
global watch_last_activity
if jid != 'None':
gr = getRoom(room)
watch_last_activity[gr] = int(time.time())
if jid != Settings['jid']: watch_activity[gr] = int(time.time())
global execute, timer
if GT('iq_version_enable'): timer = [connect_watch,watch_room_activity]
presence_control = [connect_watch_uni]
message_control = [connect_watch_uni]
execute = [(6,'watcher',c_watcher,1,'Connection activity control.'),
(9,'activity',activity_watch,1,'Show activity of conferences.')]