2011-01-07 20:37:24 -06:00
"""
Seen - track when a person speaks , and allow data to be queried
Copyright ( C ) 2010 Brian S . Stephan
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 / > .
"""
2010-07-28 23:47:29 -05:00
2010-07-29 00:42:44 -05:00
import re
2011-02-25 23:09:41 -06:00
import sqlite3
2010-07-29 00:18:20 -05:00
2010-07-27 20:35:01 -05:00
from datetime import datetime
from dateutil . tz import *
2010-07-30 18:34:10 -05:00
from extlib import irclib
2010-07-27 20:35:01 -05:00
2010-07-29 00:18:20 -05:00
from Module import Module
2010-07-27 20:35:01 -05:00
class Seen ( Module ) :
2011-01-07 20:37:24 -06:00
""" Track when people say things in public channels, and report on it. """
2011-02-25 21:54:09 -06:00
def db_init ( self ) :
""" Create the table to store seen data. """
version = self . db_module_registered ( self . __class__ . __name__ )
if version == None :
db = self . get_db ( )
try :
db . execute ( '''
CREATE TABLE seen_nicks (
nick TEXT NOT NULL PRIMARY KEY ,
host TEXT NOT NULL ,
time TEXT DEFAULT CURRENT_TIMESTAMP ,
what TEXT NOT NULL
) ''' )
sql = ' INSERT INTO drbotzo_modules VALUES (?,?) '
db . execute ( sql , ( self . __class__ . __name__ , 1 ) )
db . commit ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2011-02-25 21:54:09 -06:00
version = 1
except sqlite3 . Error as e :
db . rollback ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2012-07-15 21:32:12 -05:00
self . log . error ( " sqlite error: " + str ( e ) )
2011-02-25 21:54:09 -06:00
raise
2011-06-15 11:07:32 -05:00
if version < 2 :
db = self . get_db ( )
try :
version = 2
db . execute ( ''' DROP TABLE seen_nicks ''' )
db . execute ( '''
CREATE TABLE seen_nicks (
nick TEXT NOT NULL ,
location TEXT NOT NULL ,
host TEXT NOT NULL ,
time TEXT DEFAULT CURRENT_TIMESTAMP ,
what TEXT NOT NULL
) ''' )
db . execute ( '''
CREATE UNIQUE INDEX seen_nicks_nick_and_location_index
ON seen_nicks ( nick , location ) ''' )
db . commit ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2011-06-15 11:07:32 -05:00
self . db_register_module_version ( self . __class__ . __name__ , version )
except sqlite3 . Error as e :
db . rollback ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2012-07-15 21:32:12 -05:00
self . log . error ( " sqlite error: " + str ( e ) )
2011-06-15 11:07:32 -05:00
raise
2011-02-25 21:54:09 -06:00
2011-01-07 23:09:07 -06:00
def do ( self , connection , event , nick , userhost , what , admin_unlocked ) :
2011-02-25 21:54:09 -06:00
""" Track pubmsg/privmsg events, and if asked, report on someone. """
2010-07-27 20:35:01 -05:00
2011-06-15 11:07:32 -05:00
where = event . target ( )
2011-02-25 21:54:09 -06:00
# whatever it is, store it
try :
2011-06-16 21:26:13 -05:00
# if there's no where, this is probably a sub-command. don't learn it
if where :
db = self . get_db ( )
cur = db . cursor ( )
statement = ' REPLACE INTO seen_nicks (nick, location, host, what) VALUES (?, ?, ?, ?) '
cur . execute ( statement , ( nick , where , userhost , what . decode ( ' utf-8 ' , ' replace ' ) ) )
db . commit ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2011-02-25 21:54:09 -06:00
except sqlite3 . Error as e :
db . rollback ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2012-07-15 21:32:12 -05:00
self . log . error ( " sqlite error: " + str ( e ) )
2011-02-25 21:54:09 -06:00
raise
2010-07-27 20:35:01 -05:00
2011-01-07 20:37:24 -06:00
match = re . search ( ' ^!seen \ s+( \ S+)$ ' , what )
if match :
2011-02-25 21:54:09 -06:00
nick = match . group ( 1 )
try :
db = self . get_db ( )
2011-06-15 11:07:32 -05:00
query = ' SELECT * FROM seen_nicks WHERE nick = ? AND location = ? '
cursor = db . execute ( query , ( nick , where ) )
2011-02-25 21:54:09 -06:00
result = cursor . fetchone ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2011-02-25 21:54:09 -06:00
if result :
seentime = datetime . strptime ( result [ ' time ' ] , ' % Y- % m- %d % H: % M: % S ' ) . replace ( tzinfo = tzutc ( ) )
2012-07-10 17:15:52 -05:00
replystr = ' last saw {0:s} in {3:s} at {1:s} saying \' {2:s} \' . ' . format ( result [ ' nick ' ] , seentime . astimezone ( tzlocal ( ) ) . strftime ( ' % Y/ % m/ %d % H: % M: % S % Z ' ) , result [ ' what ' ] . encode ( ' utf-8 ' , ' ignore ' ) , result [ ' location ' ] . encode ( ' utf-8 ' , ' ignore ' ) )
rewrite recursion/alias code for the 500th time.
more of a moving of the code, actually, it now exists in (an overridden)
_handle_event, so that recursions happen against irc events directly,
rather than an already partially interpreted object.
with this change, modules don't need to implement do() nor do we have a
need for the internal_bus, which was doing an additional walk of the
modules after the irc event was already handled and turned into text. now
the core event handler does the recursion scans.
to support this, we bring back the old replypath trick and use it again,
so we know when to send a privmsg reply and when to return text so that
it may be chained in recursion. this feels old hat by now, but if you
haven't been following along, you should really look at the diff.
that's the meat of the change. the rest is updating modules to use
self.reply() and reimplementing (un)register_handlers where appropriate
2011-02-17 01:08:45 -06:00
return self . reply ( connection , event , replystr )
2012-07-10 17:15:52 -05:00
else :
return self . reply ( connection , event , ' i have not seen {0:s} in {1:s} . ' . format ( nick , where ) )
2011-02-25 21:54:09 -06:00
except sqlite3 . Error as e :
db . rollback ( )
2011-06-20 22:34:27 -05:00
db . close ( )
2012-07-15 21:32:12 -05:00
self . log . error ( " sqlite error: " + str ( e ) )
2011-02-25 21:54:09 -06:00
raise
2010-07-27 20:35:01 -05:00
2010-07-28 23:48:47 -05:00
# vi:tabstop=4:expandtab:autoindent
2010-07-28 00:11:58 -05:00
# kate: indent-mode python;indent-width 4;replace-tabs on;