bss/dr.botzo
1
0
Fork 0
Code Issues 3 Pull Requests Releases 2 Wiki Activity

move save of config to separate method in IrcAdmin, save config when receiving ^C

Browse Source
This commit is contained in:
Brian S. Stephan 2010-07-29 23:51:43 -05:00
parent 88186e1425
commit 861c343f2b
2 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
TODO
View File

@ -10,7 +10,6 @@ dr.botzo --- TODO
* karma v2 * karma v2
* some sort of cron interface (periodic events) * some sort of cron interface (periodic events)
* named pipe to send commands to the bot outside of IRC * named pipe to send commands to the bot outside of IRC
* properly save when receiving ^C
* url module * url module
* direct adds implied. also do historical urls? * direct adds implied. also do historical urls?
* last 20 or something? * last 20 or something?

18
modules/IrcAdmin.py
View File

@ -15,6 +15,7 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
from ConfigParser import NoOptionError from ConfigParser import NoOptionError
import signal
import sys import sys
from irclib import irclib from irclib import irclib
@ -29,6 +30,9 @@ class IrcAdmin(Module):
super(IrcAdmin, self).__init__(config, server, modlist) super(IrcAdmin, self).__init__(config, server, modlist)
modlist.append(self) modlist.append(self)
# we define save, so we'll bite the bullet and take SIGINT
signal.signal(signal.SIGINT, self.sigint_handler)
def register_handlers(self, server): def register_handlers(self, server):
server.add_global_handler('welcome', self.on_connect) server.add_global_handler('welcome', self.on_connect)
server.add_global_handler('pubmsg', self.on_pubmsg) server.add_global_handler('pubmsg', self.on_pubmsg)
@ -99,8 +103,7 @@ class IrcAdmin(Module):
if replypath is not None: if replypath is not None:
connection.privmsg(replypath, 'quitting') connection.privmsg(replypath, 'quitting')
connection.quit(' '.join(whats[1:])) connection.quit(' '.join(whats[1:]))
with open('dr.botzo.cfg', 'w') as cfg: self.save_config()
self.config.write(cfg)
sys.exit() sys.exit()
def sub_autojoin_manipulate(self, connection, event, nick, userhost, replypath, what, admin_unlocked): def sub_autojoin_manipulate(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
@ -158,5 +161,16 @@ class IrcAdmin(Module):
else: else:
connection.privmsg(replypath, replystr) connection.privmsg(replypath, replystr)
# Save the config file.
def save_config(self):
with open('dr.botzo.cfg', 'w') as cfg:
self.config.write(cfg)
# SIGINT signal handler
def sigint_handler(self, signal, frame):
self.save_config()
print('saved config')
sys.exit()
# vi:tabstop=4:expandtab:autoindent # vi:tabstop=4:expandtab:autoindent
# kate: indent-mode python;indent-width 4;replace-tabs on; # kate: indent-mode python;indent-width 4;replace-tabs on;