fix: imported logger and version number

Author: jhirschibar

curator/data_pipeline/download.py

@@ -1,4 +1,3 @@
-import logging
 from datetime import datetime
 
 from aiomultiprocess import Pool
@@ -23,11 +22,9 @@
 from db_tools.queries import lookup_multi_ticker_ids
 from db_tools.utils import OptionTicker
 
-from curator.proj_constants import POLYGON_BASE_URL
+from curator.proj_constants import POLYGON_BASE_URL, log
 from curator.utils import pool_kwarg_config
 
-log = logging.getLogger(__name__)
-
 planned_exceptions = (
     InvalidArgs,
     ProjClientConnectionError,

curator/data_pipeline/orchestrator.py

@@ -1,4 +1,3 @@
-import logging
 from datetime import datetime
 
 from data_pipeline.download import (
@@ -21,7 +20,7 @@
 from db_tools.utils import generate_o_ticker_lookup, pull_tickers_from_db
 from pandas import DataFrame
 
-log = logging.getLogger(__name__)
+from curator.proj_constants import log
 
 
 async def import_all(tickers: list, start_date: datetime, end_date: datetime, months_hist: int):

curator/data_pipeline/path_runner.py

@@ -1,4 +1,3 @@
-import logging
 import os
 from abc import ABC, abstractmethod
 from datetime import datetime
@@ -15,7 +14,7 @@
 )
 from db_tools.utils import OptionTicker
 
-from curator.proj_constants import BASE_DOWNLOAD_PATH, POSTGRES_BATCH_MAX, logger_setup
+from curator.proj_constants import BASE_DOWNLOAD_PATH, POSTGRES_BATCH_MAX, log
 from curator.utils import (
     clean_o_ticker,
     months_ago,
@@ -24,8 +23,6 @@
     timestamp_to_datetime,
 )
 
-log = logging.getLogger(__name__)
-
 
 class PathRunner(ABC):
     """Base class for runner that will traverse the directory structure and retrieve/clean raw data.

curator/data_pipeline/polygon_utils.py

@@ -1,5 +1,4 @@
 import asyncio
-import logging
 import os
 from abc import ABC, abstractmethod
 from datetime import date, datetime
@@ -18,7 +17,7 @@
 from dateutil.relativedelta import relativedelta
 from db_tools.utils import OptionTicker
 
-from curator.proj_constants import BASE_DOWNLOAD_PATH, POLYGON_API_KEY, POLYGON_BASE_URL
+from curator.proj_constants import BASE_DOWNLOAD_PATH, POLYGON_API_KEY, POLYGON_BASE_URL, log
 from curator.utils import (
     extract_underlying_from_o_ticker,
     first_weekday_of_month,
@@ -29,8 +28,6 @@
     write_api_data_to_file,
 )
 
-log = logging.getLogger(__name__)
-
 
 class Timespans(Enum):
     second = "second"

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "curator"
-version = "0.1.0"
+version = "1.0.4"
 description = "Package for curating historical market data from Polygon.io to backfill into a postgres database"
 authors = ["Jody Hirschi <jody@heylittlejohn.com>"]
 readme = "README.md"