shetts: fix logging

TheShellLand · Oct 25, 2023 · 8d42964 · 8d42964
1 parent 451100d
commit 8d42964
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 8 deletions.
diff --git a/automon/integrations/google/sheets/client.py b/automon/integrations/google/sheets/client.py
@@ -1,9 +1,10 @@
-from automon.log import Logging
+from automon.log import logger
 from automon.integrations.google.auth import GoogleAuthClient
 
 from .config import GoogleSheetsConfig
 
-log = Logging(name='GoogleSheetsClient', level=Logging.DEBUG)
+log = logger.logging.getLogger(__name__)
+log.setLevel(logger.DEBUG)
 
 
 class Fields:
@@ -94,7 +95,7 @@ def get(
                 **kwargs,
             ).execute()
         except Exception as e:
-            log.error(f'{e}', enable_traceback=False)
+            log.error(f'{e}')
 
         return self
 
@@ -112,7 +113,7 @@ def get_values(
                 **kwargs,
             ).execute()
         except Exception as e:
-            log.error(f'{e}', enable_traceback=False)
+            log.error(f'{e}')
 
         return self
 

diff --git a/automon/integrations/google/sheets/config.py b/automon/integrations/google/sheets/config.py
@@ -1,8 +1,9 @@
-from automon.log import Logging
+from automon.log import logger
 from automon.helpers.osWrapper import environ
 from automon.integrations.google.auth import GoogleAuthConfig
 
-log = Logging(name='SheetsConfig', level=Logging.DEBUG)
+log = logger.logging.getLogger(__name__)
+log.setLevel(logger.DEBUG)
 
 
 class GoogleSheetsConfig(GoogleAuthConfig):

diff --git a/automon/integrations/google/sheets/tests/test_google_sheets.py b/automon/integrations/google/sheets/tests/test_google_sheets.py
@@ -6,7 +6,7 @@
 import pandas as pd
 import numpy as np
 
-from automon import Logging
+from automon import logger
 from automon.integrations.facebook import FacebookGroups
 from automon.integrations.google.sheets import GoogleSheetsClient
 
@@ -25,7 +25,8 @@
 
 tracemalloc.start()
 
-log = Logging(level=Logging.INFO)
+log = logger.logging.getLogger(__name__)
+log.setLevel(logger.INFO)
 
 sheets_client = GoogleSheetsClient(
     worksheet='Automated Count WIP',