phenny-1/__init__.py

88 lines
1.9 KiB
Python
Raw Normal View History

2012-12-16 17:09:43 -05:00
#!/usr/bin/env python3
"""
__init__.py - Phenny Init Module
Copyright 2008, Sean B. Palmer, inamidst.com
Licensed under the Eiffel Forum License 2.
http://inamidst.com/phenny/
"""
import os
import signal
import sys
import threading
import time
class Watcher(object):
2012-05-31 02:24:21 -04:00
# Cf. http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/496735
def __init__(self):
self.child = os.fork()
if self.child != 0:
signal.signal(signal.SIGTERM, self.sig_term)
self.watch()
2012-05-31 02:24:21 -04:00
def watch(self):
try:
os.wait()
2012-05-31 02:24:21 -04:00
except KeyboardInterrupt:
self.kill()
sys.exit()
2012-05-31 02:24:21 -04:00
def kill(self):
try:
os.kill(self.child, signal.SIGKILL)
except OSError:
pass
2012-05-31 02:24:21 -04:00
def sig_term(self, signum, frame):
self.kill()
sys.exit()
def run_phenny(config):
if hasattr(config, 'delay'):
2012-05-31 02:24:21 -04:00
delay = config.delay
else:
delay = 20
def connect(config):
2012-05-31 02:24:21 -04:00
import bot
p = bot.Phenny(config)
ssl_context = p.get_ssl_context(config.ca_certs)
if config.ssl_cert and config.ssl_key:
ssl_context.load_cert_chain(config.ssl_cert, config.ssl_key)
p.run(config.host, config.port, config.ssl, config.ipv6, None,
ssl_context)
try:
Watcher()
2012-05-31 02:24:21 -04:00
except Exception as e:
print('Warning:', e, '(in __init__.py)', file=sys.stderr)
while True:
try:
connect(config)
2012-05-31 02:24:21 -04:00
except KeyboardInterrupt:
sys.exit()
2008-03-02 10:50:05 -05:00
2012-05-31 02:24:21 -04:00
if not isinstance(delay, int):
break
msg = "Warning: Disconnected. Reconnecting in {0} seconds..."
print(msg.format(delay), file=sys.stderr)
2012-05-31 02:24:21 -04:00
time.sleep(delay)
def run(config):
2012-05-31 02:24:21 -04:00
t = threading.Thread(target=run_phenny, args=(config,))
if hasattr(t, 'run'):
t.run()
else:
t.start()
2012-05-31 02:24:21 -04:00
if __name__ == '__main__':
print(__doc__)