Remove extra layer of indirection in 'logs.py'

This removes an extra layer of indirection

Author: yuvipanda

src/jupyterhub_cost_monitoring/app.py

@@ -1,13 +1,13 @@
 from datetime import timedelta
 
 import requests
+import logging
 from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import Response
 from prometheus_client import CONTENT_TYPE_LATEST, generate_latest
 
 from .const_usage import USAGE_MAP
 from .date_utils import get_now_date, parse_from_to_in_query_params
-from .logs import get_logger
 from .metrics import MetricsMiddleware
 from .query_cost_aws import (
     query_hub_names,
@@ -26,7 +26,7 @@
 
 app = FastAPI()
 app.add_middleware(MetricsMiddleware)
-logger = get_logger(__name__)
+logger = logging.getLogger(__name__)
 
 
 @app.get("/")

src/jupyterhub_cost_monitoring/logs.py

@@ -22,10 +22,10 @@
     SERVICE_COMPONENT_MAP,
 )
 from .date_utils import DateRange
-from .logs import get_logger
+import logging
 from .query_usage import _filter_json, query_usage, query_user_groups
 
-logger = get_logger(__name__)
+logger = logging.getLogger(__name__)
 aws_ce_client = boto3.client("ce")
 
 

src/jupyterhub_cost_monitoring/query_cost_aws.py

@@ -13,9 +13,9 @@
 from .cache import ttl_lru_cache
 from .const_usage import USAGE_MAP, USER_GROUP_INFO
 from .date_utils import DateRange, get_now_date
-from .logs import get_logger
+import logging
 
-logger = get_logger(__name__)
+logger = logging.getLogger(__name__)
 
 prometheus_host = os.environ.get("SUPPORT_PROMETHEUS_SERVER_SERVICE_HOST", "localhost")
 prometheus_port = int(os.environ.get("SUPPORT_PROMETHEUS_SERVER_SERVICE_PORT", 9090))

src/jupyterhub_cost_monitoring/query_usage.py

@@ -1,9 +1,9 @@
 import pandas as pd
 
-from src.jupyterhub_cost_monitoring.logs import get_logger
+import logging
 from src.jupyterhub_cost_monitoring.query_usage import _calculate_daily_cost_factors
 
-logger = get_logger(__name__)
+logger = logging.getLogger(__name__)
 
 
 def test_cost_factor_by_hub(input_data_usage, output_data_hub):

tests/test_cost.py

@@ -7,12 +7,12 @@
     METRICS_UNBLENDED_COST,
 )
 from src.jupyterhub_cost_monitoring.date_utils import parse_from_to_in_query_params
-from src.jupyterhub_cost_monitoring.logs import get_logger
+import logging
 from src.jupyterhub_cost_monitoring.query_cost_aws import (
     query_total_costs_per_component,
 )
 
-logger = get_logger(__name__)
+logger = logging.getLogger(__name__)
 
 date_range = parse_from_to_in_query_params("2025-09-01", "2025-09-02")