Qubx 0.1.82__tar.gz → 0.1.84__tar.gz

{qubx-0.1.82 → qubx-0.1.84}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: Qubx
-Version: 0.1.82
+Version: 0.1.84
 Summary: Qubx - quantitative trading framework
 Home-page: https://github.com/dmarienko/Qubx
 Author: Dmitry Marienko

{qubx-0.1.82 → qubx-0.1.84}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "Qubx"
-version = "0.1.82"
+version = "0.1.84"
 description = "Qubx - quantitative trading framework"
 authors = ["Dmitry Marienko <dmitry@gmail.com>"]
 readme = "README.md"

{qubx-0.1.82 → qubx-0.1.84}/src/qubx/__init__.py

@@ -10,13 +10,20 @@ def formatter(record):
     end = record["extra"].get("end", "\n")
     fmt = "<lvl>{message}</lvl>%s" % end
     if record["level"].name in {"WARNING", "SNAKY"}:
-        fmt = "<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - %s" % fmt
+        fmt = (
+            "<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - %s" % fmt
+        )
 
-    prefix = "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> [ <level>%s</level> ] " % record["level"].icon
+    prefix = (
+        "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> [ <level>%s</level> ] "
+        % record["level"].icon
+    )
 
     if record["exception"] is not None:
         # stackprinter.set_excepthook(style='darkbg2')
-        record["extra"]["stack"] = stackprinter.format(record["exception"], style="darkbg")
+        record["extra"]["stack"] = stackprinter.format(
+            record["exception"], style="darkbg"
+        )
         fmt += "\n{extra[stack]}\n"
 
     if record["level"].name in {"TEXT"}:
@@ -25,24 +32,43 @@ def formatter(record):
     return prefix + fmt
 
 
-config = {
-    "handlers": [ {"sink": sys.stdout, "format": "{time} - {message}"}, ],
-    "extra": {"user": "someone"},
-}
+class QubxLogConfig:
 
+    @staticmethod
+    def get_log_level():
+        return os.getenv("QUBX_LOG_LEVEL", "DEBUG")
+
+    @staticmethod
+    def set_log_level(level: str):
+        os.environ["QUBX_LOG_LEVEL"] = level
+        QubxLogConfig.setup_logger(level)
+
+    @staticmethod
+    def setup_logger(level: str | None = None):
+        global logger
+        config = {
+            "handlers": [
+                {"sink": sys.stdout, "format": "{time} - {message}"},
+            ],
+            "extra": {"user": "someone"},
+        }
+        logger.configure(**config)
+        logger.remove(None)
+        level = level or QubxLogConfig.get_log_level()
+        logger.add(sys.stdout, format=formatter, colorize=True, level=level)
+        logger = logger.opt(colors=True)
+
+
+QubxLogConfig.setup_logger()
 
-logger.configure(**config)
-logger.remove(None)
-logger.add(sys.stdout, format=formatter, colorize=True)
-logger = logger.opt(colors=True)
 
 # - global lookup helper
 lookup = GlobalLookup(InstrumentsLookup(), FeesLookup())
 
 
 # registering magic for jupyter notebook
-if runtime_env() in ['notebook', 'shell']:
-    from IPython.core.magic import (Magics, magics_class, line_magic, line_cell_magic)
+if runtime_env() in ["notebook", "shell"]:
+    from IPython.core.magic import Magics, magics_class, line_magic, line_cell_magic
     from IPython import get_ipython
 
     @magics_class
@@ -52,11 +78,11 @@ if runtime_env() in ['notebook', 'shell']:
 
         @line_magic
         def qubxd(self, line: str):
-            self.qubx_setup('dark' + ' ' + line)
+            self.qubx_setup("dark" + " " + line)
 
         @line_magic
         def qubxl(self, line: str):
-            self.qubx_setup('light' + ' ' + line)
+            self.qubx_setup("light" + " " + line)
 
         @line_magic
         def qubx_setup(self, line: str):
@@ -64,25 +90,26 @@ if runtime_env() in ['notebook', 'shell']:
             QUBX framework initialization
             """
             import os
-            args = [x.strip() for x in line.split(' ')]
-            
+
+            args = [x.strip() for x in line.split(" ")]
+
             # setup cython dev hooks - only if 'dev' is passed as argument
-            if line and 'dev' in args: 
+            if line and "dev" in args:
                 install_pyx_recompiler_for_dev()
 
             tpl_path = os.path.join(os.path.dirname(__file__), "_nb_magic.py")
-            with open(tpl_path, 'r', encoding="utf8") as myfile:
+            with open(tpl_path, "r", encoding="utf8") as myfile:
                 s = myfile.read()
 
             exec(s, self.shell.user_ns)
 
             # setup more funcy mpl theme instead of ugly default
             if line:
-                if 'dark' in line.lower():
-                    set_mpl_theme('dark')
+                if "dark" in line.lower():
+                    set_mpl_theme("dark")
 
-                elif 'light' in line.lower():
-                    set_mpl_theme('light')
+                elif "light" in line.lower():
+                    set_mpl_theme("light")
 
             # install additional plotly helpers
             # from qube.charting.plot_helpers import install_plotly_helpers
@@ -91,6 +118,7 @@ if runtime_env() in ['notebook', 'shell']:
         def _get_manager(self):
             if self.__manager is None:
                 import multiprocessing as m
+
                 self.__manager = m.Manager()
             return self.__manager
 
@@ -102,7 +130,7 @@ if runtime_env() in ['notebook', 'shell']:
             >>> %%proc x, y as MyProc1
             >>> x.set('Hello')
             >>> y.set([1,2,3,4])
-            
+
             """
             import multiprocessing as m
             import time, re
@@ -111,8 +139,8 @@ if runtime_env() in ['notebook', 'shell']:
             name = None
             if line:
                 # check if custom process name was provided
-                if ' as ' in line:
-                    line, name = line.split('as')
+                if " as " in line:
+                    line, name = line.split("as")
                     if not name.isspace():
                         name = name.strip()
                     else:
@@ -120,11 +148,11 @@ if runtime_env() in ['notebook', 'shell']:
                         return
 
                 ipy = get_ipython()
-                for a in [x for x in re.split('[\ ,;]', line.strip()) if x]:
+                for a in [x for x in re.split("[\ ,;]", line.strip()) if x]:
                     ipy.push({a: self._get_manager().Value(None, None)})
 
             # code to run
-            lines = '\n'.join(['    %s' % x for x in cell.split('\n')])
+            lines = "\n".join(["    %s" % x for x in cell.split("\n")])
 
             def fn():
                 result = get_ipython().run_cell(lines)
@@ -136,17 +164,18 @@ if runtime_env() in ['notebook', 'shell']:
                 if result.error_in_exec:
                     raise result.error_in_exec
 
-            t_start = str(time.time()).replace('.', '_')
-            f_id = f'proc_{t_start}' if name is None else name
+            t_start = str(time.time()).replace(".", "_")
+            f_id = f"proc_{t_start}" if name is None else name
             if self._is_task_name_already_used(f_id):
                 f_id = f"{f_id}_{t_start}"
 
             task = m.Process(target=fn, name=f_id)
             task.start()
-            print(' -> Task %s is started' % f_id)
+            print(" -> Task %s is started" % f_id)
 
         def _is_task_name_already_used(self, name):
             import multiprocessing as m
+
             for p in m.active_children():
                 if p.name == name:
                     return True
@@ -155,16 +184,17 @@ if runtime_env() in ['notebook', 'shell']:
         @line_magic
         def list_proc(self, line):
             import multiprocessing as m
+
             for p in m.active_children():
                 print(p.name)
 
         @line_magic
         def kill_proc(self, line):
             import multiprocessing as m
+
             for p in m.active_children():
                 if line and p.name.startswith(line):
                     p.terminate()
 
-
     # - registering magic here
     get_ipython().register_magics(QubxMagics)