From fbedbfe7bac894352e33df8d651445689a376e34 Mon Sep 17 00:00:00 2001
From: Chris Hines <chris.hines@monash.edu>
Date: Tue, 15 Nov 2022 13:17:58 +1100
Subject: [PATCH] more logging

---
 twsproxy/__init__.py        | 9 ++++++---
 twsproxy/server/__init__.py | 1 +
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/twsproxy/__init__.py b/twsproxy/__init__.py
index 52937e9..fe659ac 100644
--- a/twsproxy/__init__.py
+++ b/twsproxy/__init__.py
@@ -107,7 +107,7 @@ class TWSProxy(threading.Thread):
         import re
         import requests
         logger = logging.getLogger()
-        token_format = [ b'Authorization: token (?P<authtok>\w+)[\W|$]',
+        token_formats = [ b'Authorization: token (?P<authtok>\w+)[\W|$]',
                          b'token=(?P<authtok>\w+)[&|\W|$]',
                          b'twsproxyauth=(?P<authtok>\w+)[\W|$]']
 
@@ -184,6 +184,7 @@ def mainserver(port=None):
     import logging
     import sys
     import os
+    to_log=None
     try:
         from logging.handlers import TimedRotatingFileHandler
         logger = logging.getLogger()
@@ -192,12 +193,14 @@ def mainserver(port=None):
         handler.setFormattter(formatter)
         logger.addHandler(handler)
     except Exception as e:
-        print(e)
         import traceback
-        print(traceback.format_exc())
+        to_log = "{}\n".format(e)
+        to_log = to_log + traceback.format_exc()
         logging.basicConfig(filename=os.path.expanduser("~/.tws.log"),format="%(asctime)s %(levelname)s:%(process)s: %(message)s")
     logger = logging.getLogger()
     logger.setLevel(logging.DEBUG)
+    if to_log is not None:
+        logger.debug(to_log)
     logger.debug("starting TWS proxy")
     if port is None:
         try:
diff --git a/twsproxy/server/__init__.py b/twsproxy/server/__init__.py
index 2eba52f..b4631a0 100644
--- a/twsproxy/server/__init__.py
+++ b/twsproxy/server/__init__.py
@@ -34,6 +34,7 @@ class TWSServer:
         while not self.stop.is_set():
             try:
                 (clientsocket, address) = serversocket.accept()
+                logger.debug('accepted connection {}'.format(clientsocket))
             except socket.timeout:
                 continue
             clientsocket.setblocking(True)
-- 
GitLab