updated to use env variables passed in constructor

Author: kafkasl

README.md

@@ -22,8 +22,14 @@ Get the host & token by creating a new source at https://telemetry.betterstack.c
 
 3. **Initialize once** in your main file
    ```python
-    from fewsats_logger.core import setup_logging
-   setup_logging()
+   import os
+   from dotenv import load_dotenv
+   from fewsats_logger.core import setup_logging
+
+   load_dotenv()   
+   setup_logging(env=os.environ.get("ENV", "dev"),
+                 host=os.environ.get("BETTER_STACK_HOST"),
+                 token=os.environ.get("BETTER_STACK_SOURCE_TOKEN"))
    ```
 
 4. **Use standard logging** anywhere in your code
@@ -52,43 +58,26 @@ That's it. Logs will appear in Better Stack dashboard and console.
 
 ```python
 logger.info("Processing order", extra={
-    "order_id": "12345",
+    "external_id": "12345",
     "customer_id": "67890"
 })
 ```
 
-### Integration Examples
-
-#### FastAPI
+you can add it to multiple statements like this:
 
 ```python
-from fastapi import FastAPI
-from fewsats_logger import setup_logging
-
-setup_logging()
-app = FastAPI()
-```
-
-#### Flask
 
-```python
-from flask import Flask
-from fewsats_logger import setup_logging
+with logtail.context(order={"external_id": "12345" }):
+    logger.info('Processing order')
+    # More code here ...
+    logger.info('Done.')
 
-setup_logging()
-app = Flask(__name__)
 ```
 
-#### Command Line
-
-```python
-from fewsats_logger import setup_logging
-import logging
-
-def main():
-    setup_logging()
-    logger = logging.getLogger(__name__)
-    logger.info("Starting application")
+## Issues
 
-if __name__ == "__main__":
-    main()
+⚠️ **WARNING**: When using FastAPI, you may encounter server initialization issues:
+- The server may get blocked when started with `python server/main.py`
+- But works fine with `uvicorn server.main:app --reload --host 0.0.0.0`
+- This appears to be related to logging in the `__main__` scope
+- Avoid using logging statements directly in the main module scope or in `if __name__ == "__main__":` blocks

fewsats_logger/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.1"
+__version__ = "0.0.3"

fewsats_logger/core.py

@@ -6,63 +6,35 @@
 __all__ = ['setup_logging']
 
 # %% ../nbs/00_core.ipynb 3
-from logtail import LogtailHandler
-import logging
-import os
+import logging.config
+from logtail import LogtailHandler, LogtailContext
+import sys
 
 # %% ../nbs/00_core.ipynb 4
-def setup_logging():
-    """
-    Configure logging to use Better Stack's LogtailHandler
-    with env context included automatically.
-    
-    This should be called once at application startup.
-    """
-    env = os.environ.get("ENV", "development")
-    source_token = os.environ.get("BETTER_STACK_SOURCE_TOKEN")
-    host = os.environ.get("BETTER_STACK_HOST")
-    
-    if not source_token or not host:
-        # If Better Stack credentials are not available, fall back to basic config
-        logging.basicConfig(level=logging.INFO)
-        logging.warning("Better Stack credentials not found, using basic logging config")
-        return
-    
-    # Configure the root logger first to set the minimum level
+def setup_logging(env, host, token):
     root_logger = logging.getLogger()
     root_logger.setLevel(logging.INFO)
 
-    # Remove any existing handlers to avoid duplicates
-    for handler in root_logger.handlers[:]:
-        root_logger.removeHandler(handler)
+    # Clear existing handlers to avoid duplicates if called multiple times
+    root_logger.handlers = []
 
-    # Add a console handler for local development
-    console = logging.StreamHandler()
-    root_logger.addHandler(console)
-    
-    try:
-        # Set up the Logtail handler
-        handler = LogtailHandler(
-            source_token=source_token,
-            host=host
+    # Add Logtail handler for cloud logging
+    logtail_handler = LogtailHandler(
+        source_token=token, 
+        host=host,
+        context=LogtailContext().context(
+            environment={'name': env}
         )
-        
-        # Override LogRecord factory to add env to all records
-        old_factory = logging.getLogRecordFactory()
-        def record_factory(*args, **kwargs):
-            record = old_factory(*args, **kwargs)
-            record.env = env
-            return record
-        logging.setLogRecordFactory(record_factory)
-        
-        # Add the handler to the root logger
-        root_logger.addHandler(handler)
-        
-        logging.info(f"Better Stack logging configured with env: {env}")
-    except Exception as e:
-        logging.error(f"Failed to configure Better Stack logging: {e}")
-
+    )
+    root_logger.addHandler(logtail_handler)
+    
 
+    console_handler = logging.StreamHandler(sys.stdout)
+    console_formatter = logging.Formatter(
+        '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    )
+    console_handler.setFormatter(console_formatter)
+    root_logger.addHandler(console_handler)
 
 # %% ../nbs/00_core.ipynb 5
 # If this file is run directly, test the logging

nbs/00_core.ipynb

@@ -35,9 +35,9 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "from logtail import LogtailHandler\n",
-    "import logging\n",
-    "import os"
+    "import logging.config\n",
+    "from logtail import LogtailHandler, LogtailContext\n",
+    "import sys"
    ]
   },
   {
@@ -46,59 +46,32 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "\n",
     "#| export\n",
-    "def setup_logging():\n",
-    "    \"\"\"\n",
-    "    Configure logging to use Better Stack's LogtailHandler\n",
-    "    with env context included automatically.\n",
-    "    \n",
-    "    This should be called once at application startup.\n",
-    "    \"\"\"\n",
-    "    env = os.environ.get(\"ENV\", \"development\")\n",
-    "    source_token = os.environ.get(\"BETTER_STACK_SOURCE_TOKEN\")\n",
-    "    host = os.environ.get(\"BETTER_STACK_HOST\")\n",
-    "    \n",
-    "    if not source_token or not host:\n",
-    "        # If Better Stack credentials are not available, fall back to basic config\n",
-    "        logging.basicConfig(level=logging.INFO)\n",
-    "        logging.warning(\"Better Stack credentials not found, using basic logging config\")\n",
-    "        return\n",
-    "    \n",
-    "    # Configure the root logger first to set the minimum level\n",
+    "\n",
+    "def setup_logging(env, host, token):\n",
     "    root_logger = logging.getLogger()\n",
     "    root_logger.setLevel(logging.INFO)\n",
     "    \n",
-    "    # Remove any existing handlers to avoid duplicates\n",
-    "    for handler in root_logger.handlers[:]:\n",
-    "        root_logger.removeHandler(handler)\n",
-    "    \n",
-    "    # Add a console handler for local development\n",
-    "    console = logging.StreamHandler()\n",
-    "    root_logger.addHandler(console)\n",
+    "    # Clear existing handlers to avoid duplicates if called multiple times\n",
+    "    root_logger.handlers = []\n",
     "    \n",
-    "    try:\n",
-    "        # Set up the Logtail handler\n",
-    "        handler = LogtailHandler(\n",
-    "            source_token=source_token,\n",
-    "            host=host\n",
+    "    # Add Logtail handler for cloud logging\n",
+    "    logtail_handler = LogtailHandler(\n",
+    "        source_token=token, \n",
+    "        host=host,\n",
+    "        context=LogtailContext().context(\n",
+    "            environment={'name': env}\n",
     "        )\n",
-    "        \n",
-    "        # Override LogRecord factory to add env to all records\n",
-    "        old_factory = logging.getLogRecordFactory()\n",
-    "        def record_factory(*args, **kwargs):\n",
-    "            record = old_factory(*args, **kwargs)\n",
-    "            record.env = env\n",
-    "            return record\n",
-    "        logging.setLogRecordFactory(record_factory)\n",
-    "        \n",
-    "        # Add the handler to the root logger\n",
-    "        root_logger.addHandler(handler)\n",
-    "        \n",
-    "        logging.info(f\"Better Stack logging configured with env: {env}\")\n",
-    "    except Exception as e:\n",
-    "        logging.error(f\"Failed to configure Better Stack logging: {e}\")\n",
-    "\n"
+    "    )\n",
+    "    root_logger.addHandler(logtail_handler)\n",
+    "    \n",
+    "\n",
+    "    console_handler = logging.StreamHandler(sys.stdout)\n",
+    "    console_formatter = logging.Formatter(\n",
+    "        '%(asctime)s - %(name)s - %(levelname)s - %(message)s'\n",
+    "    )\n",
+    "    console_handler.setFormatter(console_formatter)\n",
+    "    root_logger.addHandler(console_handler)"
    ]
   },
   {

settings.ini

@@ -1,47 +1,39 @@
 [DEFAULT]
-# All sections below are required unless otherwise specified.
-# See https://github.com/AnswerDotAI/nbdev/blob/main/settings.ini for examples.
-
-### Python library ###
 repo = fewsats-logger-py
 lib_name = fewsats-logger
-version = 0.0.1
+version = 0.0.3
 min_python = 3.9
 license = apache2
 black_formatting = False
-
-### nbdev ###
 doc_path = _docs
 lib_path = fewsats_logger
 nbs_path = nbs
 recursive = True
 tst_flags = notest
 put_version_in_init = True
 update_pyproject = True
-
-### Docs ###
 branch = main
 custom_sidebar = False
-doc_host = https://%(user)s.github.io
-doc_baseurl = /%(repo)s
-git_url = https://github.com/%(user)s/%(repo)s
-title = %(lib_name)s
-
-### PyPI ###
+doc_host = https://Fewsats.github.io
+doc_baseurl = /fewsats-logger-py
+git_url = https://github.com/Fewsats/fewsats-logger-py
+title = fewsats-logger
 audience = Developers
 author = Pol Alvarez
 author_email = pol.avms@gmail.com
-copyright = 2025 onwards, %(author)s
+copyright = 2025 onwards, Pol Alvarez
 description = Python Logger based on BetterStack
 keywords = nbdev jupyter notebook python
 language = English
 status = 3
 user = Fewsats
-
 requirements = logtail-python
+readme_nb = index.ipynb
+allowed_metadata_keys = 
+allowed_cell_metadata_keys = 
+jupyter_hooks = False
+clean_ids = True
+clear_all = False
+cell_number = True
+skip_procs = 
 
-### Optional ###
-# dev_requirements = 
-# console_scripts =
-# conda_user = 
-# package_data =