bert.py

@@ -38,8 +38,10 @@ y = "\x038"
 r = "\x034"
 g = "\x033"
 
-# global variable for the socket, see connect()
+context = ssl.create_default_context()
-irc = None
+with socket.socket(socket.AF_INET6, socket.SOCK_STREAM) as sock:
+    irc = context.wrap_socket(sock, server_hostname=host)
+
 
 # Currency API
 def get_exchange():
@@ -69,11 +71,11 @@ def safeexit():
 
 
 def connect(host, port):
-   global irc
    print("Connecting to "+host+":"+str(port))
-   context = ssl.create_default_context()
+   try:
-   with socket.create_connection((host, port)) as sock:
+      irc.connect((host, port))
-       irc = context.wrap_socket(sock, server_hostname=host)
+   except Exception as err:
+      print("Connection failed! "+str(err))
 
 
 def sendRaw(data):
@@ -82,17 +84,26 @@ def sendRaw(data):
 
 def register(nick, host):
    print("Registering User...")
+   try:
       sendRaw("USER "+nick+" "+host+" "+nick+" "+nick+"\n")
+   except Exception as err:
+      print("Failed! "+str(err))
 
 
 def name(nick):
    print("Setting Nickname to "+nick)
+   try:
       sendRaw("NICK "+nick+"\n")
+   except Exception as err:
+      print("Failed! "+str(err))
 
 
 def auth(nick, password):
    print("Authenticating...")
+   try:
       sendRaw("PRIVMSG NickServ :IDENTIFY "+nick+" "+password+"\n")
+   except Exception as err:
+      print(err)
 
 
 def join(chan):
@@ -105,7 +116,10 @@ def join(chan):
 
 def mode(nick):
    print("Setting modes +B-x")
+   try:
       sendRaw("MODE "+nick+" +B-x\n")
+   except Exception as err:
+      print("Failed! "+str(err))
 
 
 def part(chan):
@@ -240,6 +254,7 @@ def ping(msg):
 
 
 def start(host, port, nick, password, chans):
+   try:
       connect(host, port)
       sleep(2)
       register(nick, host)
@@ -253,6 +268,8 @@ def start(host, port, nick, password, chans):
       for chan in chans:
          join(chan)
       sleep(2)
+   except Exception as err:
+      print("FAIL: "+str(err))
 
 
 def command_loop():
@@ -492,11 +509,7 @@ class fefe_thread(threading.Thread):
       fefe_check()
 
 def main():
-   try:
    start(host, port, nick, password, set(join_chans + fefe_chans))
-   except Exception as err:
-      print("FAIL: "+str(err))
-      sys.exit(1)
    thread_command = command_thread(1)
    #thread_fefe = fefe_thread(2)
    thread_command.start()