GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2024-11-01 01:09:22 +01:00
Code Issues Projects Releases Wiki Activity

Second attempt at multinet support, by wrapping threading.Thread around Irc.run()

Browse Source 
Closes #15.
This commit is contained in:
James Lu 2015-07-09 18:29:00 -07:00
parent e54e17a0ea
commit eec024bf9a
2 changed files with 34 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
main.py
View File

@ -6,16 +6,19 @@ import socket
import time
import sys
from collections import defaultdict
import threading
from log import log
import conf
import classes
import utils
class Irc():
def __init__(self, proto, conf):
def __init__(self, netname, proto, conf):
# threading.Thread.__init__(self)
# Initialize some variables
self.connected = False
self.name = conf['server']['netname']
self.name = netname
self.conf = conf
# Server, channel, and user indexes to be populated by our protocol module
self.servers = {}
@ -38,7 +41,7 @@ class Irc():
self.maxnicklen = 30
self.prefixmodes = 'ov'
self.serverdata = conf['server']
self.serverdata = conf['servers'][netname]
self.sid = self.serverdata["sid"]
self.botdata = conf['bot']
self.proto = proto
@ -49,12 +52,20 @@ class Irc():
port = self.serverdata["port"]
log.info("Connecting to network %r on %s:%s", self.name, ip, port)
self.socket = socket.socket()
self.socket.setblocking(0)
self.socket.settimeout(60)
self.socket.connect((ip, port))
self.proto.connect(self)
self.loaded = []
self.load_plugins()
reading_thread = threading.Thread(target = self.run)
self.connected = True
self.run()
reading_thread.start()
def disconnect(self):
self.connected = False
self.socket.shutdown()
self.socket.close()
def run(self):
buf = ""
@ -67,19 +78,19 @@ class Irc():
break
while '\n' in buf:
line, buf = buf.split('\n', 1)
log.debug("<- %s", line)
log.debug("(%s) <- %s", self.name, line)
proto.handle_events(self, line)
except socket.error as e:
log.error('Received socket.error: %s, exiting.', str(e))
break
sys.exit(1)
except (socket.error, classes.ProtocolError) as e:
log.error('Disconnected from network %r: %s: %s, exiting.',
self.name, type(e).__name__, str(e))
self.disconnect()
def send(self, data):
# Safeguard against newlines in input!! Otherwise, each line gets
# treated as a separate command, which is particularly nasty.
data = data.replace('\n', ' ')
data = data.encode("utf-8") + b"\n"
log.debug("-> %s", data.decode("utf-8").strip("\n"))
log.debug("(%s) -> %s", self.name, data.decode("utf-8").strip("\n"))
self.socket.send(data)
def load_plugins(self):
@ -103,9 +114,9 @@ if __name__ == '__main__':
if conf.conf['login']['password'] == 'changeme':
log.critical("You have not set the login details correctly! Exiting...")
sys.exit(2)
protoname = conf.conf['server']['protocol']
protocols_folder = [os.path.join(os.getcwd(), 'protocols')]
for network in conf.conf['servers']:
protoname = conf.conf['servers'][network]['protocol']
try:
moduleinfo = imp.find_module(protoname, protocols_folder)
proto = imp.load_source(protoname, moduleinfo[1])
@ -116,4 +127,4 @@ if __name__ == '__main__':
log.critical('Failed to load protocol module: import error %s', protoname, str(e))
sys.exit(2)
else:
irc_obj = Irc(proto, conf.conf)
utils.networkobjects[network] = Irc(network, proto, conf.conf)

1
utils.py
View File

@ -8,6 +8,7 @@ global bot_commands, command_hooks
# This should be a mapping of command names to functions
bot_commands = {}
command_hooks = defaultdict(list)
networkobjects = {}
class TS6UIDGenerator():
"""TS6 UID Generator module, adapted from InspIRCd source