Merge pull request #13 from mugulmd/dev/logging

Replace print() with logging library

Author: mugulmd

claude_server/__main__.py

@@ -5,6 +5,7 @@
 from moderngl_window.resources import register_dir
 
 from pathlib import Path
+import logging
 
 
 # Command line interface
@@ -16,7 +17,30 @@
 register_dir(internal_resources_dir)
 ClaudeApp.resource_dir = internal_resources_dir
 
+# Log level coloring
+logging.addLevelName(
+    logging.INFO,
+    '\u001b[32mINFO\u001b[0m'
+)
+logging.addLevelName(
+    logging.WARNING,
+    '\u001b[33mWARNING\u001b[0m'
+)
+logging.addLevelName(
+    logging.ERROR,
+    '\u001b[31mERROR\u001b[0m'
+)
+logging.addLevelName(
+    logging.CRITICAL,
+    '\u001b[31m\u001b[1mCRITICAL\u001b[0m'
+)
+
 # Configuration
+logging.basicConfig(
+    format='%(asctime)s - [%(levelname)s]::%(filename)s::%(funcName)s::%(lineno)d - %(message)s',
+    level=args.verbose.upper()
+)
+ClaudeApp.log_level = logging.getLevelName(args.verbose.upper())
 if args.frag:
     ClaudeApp.fragment_shader = args.frag
 if args.tex:

claude_server/app.py

@@ -8,6 +8,7 @@
 
 from multiprocessing import Process, Queue
 import os
+import logging
 
 
 class ClaudeApp(mglw.WindowConfig):
@@ -97,7 +98,7 @@ def __init__(self, **kwargs):
         self.tex_locations = {}
         self.load_textures()
 
-        # Store render lopp time
+        # Store render loop time
         self.render_time = 0.0
 
     def load_textures(self):
@@ -121,7 +122,7 @@ def load_textures(self):
                     self.textures.append(tex)
                     locations.append(loc)
                 except Exception as e:
-                    print(e)
+                    logging.error('%s', e)
                     continue
 
             if len(locations) > 0:
@@ -168,7 +169,7 @@ def write_uniform(self, np_dtype, name, value, caching = True):
                 # Send value as raw bytes
                 uniform.write(np.array(value).astype(np_dtype).tobytes())
         except Exception as e:
-            print(e)
+            logging.error('%s', e)
             return
 
         # Uniform value was sent successfully
@@ -200,7 +201,7 @@ def render(self, time, frametime):
                 for name, content in self.uniform_cache.items():
                     self.write_uniform(content['np_dtype'], name, content['value'], False)
             except Exception as e:
-                print(e)
+                logging.error('%s', e)
 
         # Read messages fed from server and update uniforms accordingly
         while not self.queue.empty():
@@ -209,7 +210,7 @@ def render(self, time, frametime):
                 np_dtype, name, value = self.parse_message(message)
                 self.write_uniform(np_dtype, name, value)
             except Exception as e:
-                print(e)
+                logging.error('%s', e)
                 pass
 
         # Update time

claude_server/cli.py

@@ -42,4 +42,10 @@ def create_parser():
              "Default: %(default)s"
     )
 
+    parser.add_argument(
+        '--verbose', type=str, default='info',
+        help="Verbose level (debug, info, warning, error, critical).\n"
+             "Default: %(default)s"
+    )
+
     return parser

claude_server/server.py

@@ -1,5 +1,6 @@
 import socket
 from multiprocessing import Queue
+import logging
 
 
 def server_feed(ip: str, port: int, q: Queue):
@@ -12,7 +13,7 @@ def server_feed(ip: str, port: int, q: Queue):
         # Client connection
         conn, addr = s.accept()
         with conn:
-            print(f'Connected with {addr}')
+            logging.info('Connected with %s', addr)
 
             # Listen to messages from client
             while True:
@@ -22,8 +23,9 @@ def server_feed(ip: str, port: int, q: Queue):
                     message = data.decode('utf-8')
                     message = message.replace('\r', '')
                     message = message.replace('\n', '')
+                    logging.debug('Received message: %s', message)
                     message = message.split(' ')
                     q.put_nowait(message)
                 except Exception as e:
-                    print(e)
+                    logging.error('%s', e)
                     pass