Commit fedb3cd8 authored by PidgeyL's avatar PidgeyL
Browse files

gracious shutdown

parent 78fbdc41
Loading
Loading
Loading
Loading
+11 −0
Original line number Diff line number Diff line
@@ -59,6 +59,7 @@ class IRCBot(irc.bot.SingleServerIRCBot):
    if args.v:
      print("Server welcomed us")
    for chan in self.channel:
      if not chan.startswith('#'):chan=("#%s"%chan)
      if args.v:
        print("joining %s"%chan)
      c.join(chan)
@@ -125,6 +126,13 @@ class IRCBot(irc.bot.SingleServerIRCBot):
    else:
      self.reply(e, "Not understood: " + cmd)

import signal

# signal handlers
def sig_handler(sig, frame):
    print('Caught signal: %s\nShutting down' % sig)
    bot.die()

def main():
  server = args.s
  port = args.p
@@ -132,7 +140,10 @@ def main():
  password = args.w
  user = args.u
  chans = args.c
  global bot
  bot=IRCBot(chans, nick, server, port, password=password,username=user)
  signal.signal(signal.SIGTERM, sig_handler)
  signal.signal(signal.SIGINT, sig_handler)
  if args.v:
    print("Connecting to server")
  bot.start()