zipline_polygon_bundle 0.2.0.dev1__tar.gz → 0.2.3__tar.gz

{zipline_polygon_bundle-0.2.0.dev1 → zipline_polygon_bundle-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zipline_polygon_bundle
-Version: 0.2.0.dev1
+Version: 0.2.3
 Summary: A zipline-reloaded data provider bundle for Polygon.io
 License: GNU AFFERO GENERAL PUBLIC LICENSE
                                 Version 3, 19 November 2007
@@ -666,31 +666,36 @@ License: GNU AFFERO GENERAL PUBLIC LICENSE
 Keywords: zipline,data-bundle,finance
 Author: Jim White
 Author-email: jim@fovi.com
-Requires-Python: >=3.9,<4.0
+Requires-Python: >= 3.10,<4.0
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: GNU Affero General Public License v3
 Classifier: Operating System :: OS Independent
 Requires-Dist: bcolz-zipline (>=1.2.11)
+Requires-Dist: filelock (>=3.16.0)
 Requires-Dist: fsspec (>=2024.10)
 Requires-Dist: numpy (<2)
 Requires-Dist: pandas (>=2.2,<3)
-Requires-Dist: pandas-market-calendars (>=4.4.2)
-Requires-Dist: pandas_ta (>=0.3)
 Requires-Dist: polygon-api-client (>=1.14.2)
 Requires-Dist: pyarrow (>=18.1.0,<19)
 Requires-Dist: pytz (>=2018.5)
 Requires-Dist: requests (>=2.9.1)
 Requires-Dist: toolz (>=0.8.2)
-Requires-Dist: zipline-reloaded (>=3.1)
+Requires-Dist: zipline-arrow (>=3.2.2)
 Project-URL: Repository, https://github.com/fovi-llc/zipline-polygon-bundle
 Description-Content-Type: text/markdown
 
 # zipline-polygon-bundle
-`zipline-polygon-bundle` is a `zipline-reloaded` (https://github.com/stefan-jansen/zipline-reloaded) data ingestion bundle for [Polygon.io](https://polygon.io/).
+`zipline-polygon-bundle` is a `zipline-arrow` (https://github.com/fovi-llc/zipline-arrow) data ingestion bundle for [Polygon.io](https://polygon.io/).
+
+Zipline Arrow is a fork of Zipline Reloaded `zipline-reloaded` (https://github.com/stefan-jansen/zipline-reloaded) which is only required if you want to use Polygon.io trades flatfiles.  So if you only need to use Polygon daily or minute agg flatfiles then you may want to use `zipline-polygon-bundle<0.2` which depends on `zipline-reloaded>=3.1`.
 
 ## GitHub
 https://github.com/fovi-llc/zipline-polygon-bundle
 
+## PyPi
+
+https://pypi.org/project/zipline_polygon_bundle
+
 ## Resources
 
 Get a subscription to https://polygon.io/ for an API key and access to flat files.
@@ -707,7 +712,25 @@ Code from *Trading Evolved* with some small updates for convenience: https://git
 
 One of the modifications I've made to that code is so that some of the notebooks can be run on Colab with a minimum of fuss: https://github.com/fovi-llc/trading_evolved/blob/main/Chapter%207%20-%20Backtesting%20Trading%20Strategies/First%20Zipline%20Backtest.ipynb
 
-# Ingest data from Polygon.io into Zipline
+# Zipline Reloaded (`zipline-reloaded`) or Zipline Arrow (`zipline-arrow`)?
+
+This bundle supports Polygon daily and minute aggregates and now trades too (quotes coming).  The trades are converted to minute and daily aggregates for all trading hours (extended both pre and post, as well as regular market).  But in order to support those extended hours I needed to change how Zipline handles `get_calendar` for Exchange Calendar (`exchange-calendar`) initialization.  To make that work I've forked `zipline-reloaded` as `zipline-arrow`.  The versions of this package before 0.2 depend on `zipline-reloaded>=3.1` and only support daily and minute flatfiles.  Versions >= 0.2 of `zipline-polygon-bundle` depend on `zipline-arrow` and will work with daily and minute flatfiles as well as trades flatfiles.
+
+# Ingest data from Polygon.io into Zipline using `aws s3` CLI
+Get AWS S3 CLI in the usual way: https://docs.aws.amazon.com/cli/latest/reference/s3/
+
+This will get everything which is currently around 12TB.
+```bash
+aws s3 sync s3://flatfiles/us_stocks_sip $POLYGON_DATA_DIR/flatfiles/us_stocks_sip --checksum-mode ENABLED  --endpoint-url https://files.polygon.io
+```
+
+If you don't need quotes yet (and this bundle doesn't use them yet) then this will be faster (quotes about twice as big as trades):
+```bash
+aws s3 sync s3://flatfiles/us_stocks_sip/{subdir} $POLYGON_DATA_DIR/flatfiles/us_stocks_sip/{subdir} --checksum-mode ENABLED  --endpoint-url https://files.polygon.io
+```
+
+# Alternative: Ingest data using `rclone`.
+I've had problems with `rclone` on the larger files for trades and quotes so I recommend using `aws s3` CLI instead.
 
 ## Set up your `rclone` (https://rclone.org/) configuration
 ```bash
@@ -742,9 +765,23 @@ register_polygon_equities_bundle(
 )
 ```
 
+## Cython build setup
+
+```bash
+sudo apt-get update
+sudo apt-get install python3-dev python3-poetry
+
+CFLAGS=$(python3-config --includes) pip install git+https://github.com/fovi-llc/zipline-arrow.git
+```
+
+
 ## Install the Zipline Polygon.io Bundle PyPi package and check that it works.
 Listing bundles will show if everything is working correctly.
 ```bash
+
+pip install -U git+https://github.com/fovi-llc/zipline-reloaded.git@calendar
+pip install -U git+https://github.com/fovi-llc/zipline-polygon-bundle.git
+
 pip install zipline_polygon_bundle
 zipline -e extension.py bundles
 ```
@@ -759,7 +796,7 @@ quantopian-quandl <no ingestions>
 
 ## Ingest the Polygon.io data.  The API key is needed for the split and dividend data.
 
-Note that ingest currently stores cached API data and shuffled agg data in the `POLYGON_DATA_DIR` directory (`flatfiles/us_stocks_sip/api_cache` and `flatfiles/us_stocks_sip/day_by_ticker_v1` respectively) so write access is needed at this stage.  After ingestion the data in `POLYGON_DATA_DIR` is not accessed.
+Note that ingest currently stores cached API data and shuffled agg ("by ticker") data in the `$CUSTOM_ASSET_FILES_DIR` directory which is `$ZIPLINE_ROOT/data/polygon_custom_assets` by default.
 
 ```bash
 export POLYGON_API_KEY=<your API key here>
@@ -793,6 +830,51 @@ This ingestion for 10 years of minute bars took around 10 hours on my Mac using
 zipline ingest -b polygon-minute
 ```
 
+## Using trades flat files.
+This takes a lot of space for the trades flatfiles (currently the 22 years of trades take around 4TB) and a fair bit of time to convert to minute aggregates.  The benefit though is the whole trading day is covered from premarket open to after hours close.  Also the current conversion logic ignores trade corrections, official close updates, and the TRF "dark pool" trades (because they are not reported when they occurred nor were they offered on the exchanges).  That is to make the aggregates be as good of a simulation of real-time as we can do for algo training and backtesting.  Details in the `trades_to_custom_aggs` function in `zipline_polygon_bundle/trades.py`.
+
+The conversion process creates `.csv.gz` files in the same format as Polygon flatfiles in the custom assets dir, which is `$ZIPLINE_ROOT/data/polygon_custom_assets` by default.  So while `$ZIPLINE_ROOT` needs to be writable, the Polygon flatfiles (`$POLYGON_DATA_DIR`) can be read-only.
+
+Get AWS S3 CLI in the usual way: https://docs.aws.amazon.com/cli/latest/reference/s3/
+
+```bash
+aws s3 sync s3://flatfiles/us_stocks_sip/trades_v1 $POLYGON_DATA_DIR/flatfiles/us_stocks_sip/trades_v1 --checksum-mode ENABLED  --endpoint-url https://files.polygon.io
+```
+
+## `extension.py`
+
+If you set the `ZIPLINE_ROOT` environment variable (recommended and likely necessary because the default of `~/.zipline` is probably not what you'll want) and copy your `extension.py` config there then you don't need to put `-e extension.py` on the `zipline` command line.
+
+If you leave out the `start_date` and/or `end_date` args then `register_polygon_equities_bundle` will scan for the dates of the first and last trade file in `$POLYGON_DATA_DIR` and use them respectively.
+
+The `NYSE_ALL_HOURS` calendar (defined in `zipline_polygon_bundle/nyse_all_hours_calendar.py`) uses open and close times for the entire trading day from premarket open to after hours close.
+
+Right now `agg_time="1min"` is the only supported aggregate duration because Zipline can only deal with day or minute duration aggregates.
+
+```python
+from zipline_polygon_bundle import register_polygon_equities_bundle, register_nyse_all_hours_calendar, NYSE_ALL_HOURS
+from exchange_calendars.calendar_helpers import parse_date
+# from zipline.utils.calendar_utils import get_calendar
+
+# Register the NYSE_ALL_HOURS ExchangeCalendar.
+register_nyse_all_hours_calendar()
+
+register_polygon_equities_bundle(
+    "polygon-trades",
+    calendar_name=NYSE_ALL_HOURS,
+    # start_date=parse_date("2020-01-03", raise_oob=False),
+    # end_date=parse_date("2021-01-29", raise_oob=False),
+    agg_time="1min",
+    minutes_per_day=16 * 60,
+)
+```
+
+As with the daily and minute aggs, the POLYGON_API_KEY is needed for the split and dividend data.  Also coming is SID assignment across ticker changes using the Polygon tickers API data.
+
+```bash
+zipline ingest -b polygon-trades
+```
+
 # License is Affero General Public License v3 (AGPL v3)
 The content of this project is Copyright (C) 2024 Fovi LLC and authored by James P. White (https://www.linkedin.com/in/jamespaulwhite/).  It is distributed under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE (AGPL) Version 3 (See LICENSE file).
 

zipline_polygon_bundle-0.2.3/README.md

@@ -0,0 +1,195 @@
+# zipline-polygon-bundle
+`zipline-polygon-bundle` is a `zipline-arrow` (https://github.com/fovi-llc/zipline-arrow) data ingestion bundle for [Polygon.io](https://polygon.io/).
+
+Zipline Arrow is a fork of Zipline Reloaded `zipline-reloaded` (https://github.com/stefan-jansen/zipline-reloaded) which is only required if you want to use Polygon.io trades flatfiles.  So if you only need to use Polygon daily or minute agg flatfiles then you may want to use `zipline-polygon-bundle<0.2` which depends on `zipline-reloaded>=3.1`.
+
+## GitHub
+https://github.com/fovi-llc/zipline-polygon-bundle
+
+## PyPi
+
+https://pypi.org/project/zipline_polygon_bundle
+
+## Resources
+
+Get a subscription to https://polygon.io/ for an API key and access to flat files.
+
+https://polygon.io/knowledge-base/article/how-to-get-started-with-s3
+
+Quantopian's Zipline backtester revived by Stefan Jansen: https://github.com/stefan-jansen/zipline-reloaded
+
+Stefan's excellent book *Machine Learning for Algorithmic Trading*: https://ml4trading.io/
+
+*Trading Evolved* by Andreas Clenow is a gentler introduction to Zipline Reloaded: https://www.followingthetrend.com/trading-evolved/
+
+Code from *Trading Evolved* with some small updates for convenience: https://github.com/fovi-llc/trading_evolved
+
+One of the modifications I've made to that code is so that some of the notebooks can be run on Colab with a minimum of fuss: https://github.com/fovi-llc/trading_evolved/blob/main/Chapter%207%20-%20Backtesting%20Trading%20Strategies/First%20Zipline%20Backtest.ipynb
+
+# Zipline Reloaded (`zipline-reloaded`) or Zipline Arrow (`zipline-arrow`)?
+
+This bundle supports Polygon daily and minute aggregates and now trades too (quotes coming).  The trades are converted to minute and daily aggregates for all trading hours (extended both pre and post, as well as regular market).  But in order to support those extended hours I needed to change how Zipline handles `get_calendar` for Exchange Calendar (`exchange-calendar`) initialization.  To make that work I've forked `zipline-reloaded` as `zipline-arrow`.  The versions of this package before 0.2 depend on `zipline-reloaded>=3.1` and only support daily and minute flatfiles.  Versions >= 0.2 of `zipline-polygon-bundle` depend on `zipline-arrow` and will work with daily and minute flatfiles as well as trades flatfiles.
+
+# Ingest data from Polygon.io into Zipline using `aws s3` CLI
+Get AWS S3 CLI in the usual way: https://docs.aws.amazon.com/cli/latest/reference/s3/
+
+This will get everything which is currently around 12TB.
+```bash
+aws s3 sync s3://flatfiles/us_stocks_sip $POLYGON_DATA_DIR/flatfiles/us_stocks_sip --checksum-mode ENABLED  --endpoint-url https://files.polygon.io
+```
+
+If you don't need quotes yet (and this bundle doesn't use them yet) then this will be faster (quotes about twice as big as trades):
+```bash
+aws s3 sync s3://flatfiles/us_stocks_sip/{subdir} $POLYGON_DATA_DIR/flatfiles/us_stocks_sip/{subdir} --checksum-mode ENABLED  --endpoint-url https://files.polygon.io
+```
+
+# Alternative: Ingest data using `rclone`.
+I've had problems with `rclone` on the larger files for trades and quotes so I recommend using `aws s3` CLI instead.
+
+## Set up your `rclone` (https://rclone.org/) configuration
+```bash
+export POLYGON_FILE_ENDPOINT=https://files.polygon.io/
+rclone config create s3polygon s3 env_auth=false endpoint=$POLYGON_FILE_ENDPOINT \
+  access_key_id=$POLYGON_S3_Access_ID secret_access_key=$POLYGON_Secret_Access_Key
+```
+
+## Get flat files (`*.csv.gz`) for US Stock daily aggregates.
+The default asset dir is `us_stock_sip` but that can be overriden with the `POLYGON_ASSET_SUBDIR` 
+environment variable if/when Polygon.io adds other markets to flat files.
+
+```bash
+export POLYGON_DATA_DIR=`pwd`/data/files.polygon.io
+for year in 2024 2023 2022 2021; do \
+    rclone copy -P s3polygon:flatfiles/us_stocks_sip/day_aggs_v1/$year \
+    $POLYGON_DATA_DIR/flatfiles/us_stocks_sip/day_aggs_v1/$year; \
+done
+```
+
+## `extension.py`
+
+```python
+from zipline_polygon_bundle import register_polygon_equities_bundle
+
+# All tickers (>20K) are ingested.  Filtering is TBD.
+# `start_session` and `end_session` can be set to ingest a range of dates (which must be market days).
+register_polygon_equities_bundle(
+    "polygon",
+    calendar_name="XNYS",
+    agg_time="day"
+)
+```
+
+## Cython build setup
+
+```bash
+sudo apt-get update
+sudo apt-get install python3-dev python3-poetry
+
+CFLAGS=$(python3-config --includes) pip install git+https://github.com/fovi-llc/zipline-arrow.git
+```
+
+
+## Install the Zipline Polygon.io Bundle PyPi package and check that it works.
+Listing bundles will show if everything is working correctly.
+```bash
+
+pip install -U git+https://github.com/fovi-llc/zipline-reloaded.git@calendar
+pip install -U git+https://github.com/fovi-llc/zipline-polygon-bundle.git
+
+pip install zipline_polygon_bundle
+zipline -e extension.py bundles
+```
+stdout:
+```
+csvdir <no ingestions>
+polygon <no ingestions>
+polygon-minute <no ingestions>
+quandl <no ingestions>
+quantopian-quandl <no ingestions>
+```
+
+## Ingest the Polygon.io data.  The API key is needed for the split and dividend data.
+
+Note that ingest currently stores cached API data and shuffled agg ("by ticker") data in the `$CUSTOM_ASSET_FILES_DIR` directory which is `$ZIPLINE_ROOT/data/polygon_custom_assets` by default.
+
+```bash
+export POLYGON_API_KEY=<your API key here>
+zipline -e extension.py ingest -b polygon
+```
+
+### Cleaning up bad ingests
+After a while you may wind up with old (or empty because of an error during ingestion) bundles cluttering
+up the list and could waste space (although old bundles may be useful for rerunning old backtests).
+To remove all but the last ingest (say after your first successful ingest after a number of false starts) you could use:
+```bash
+zipline -e extension.py clean -b polygon --keep-last 1
+```
+
+## Using minute aggregate flat files.
+Minute aggs work too but everything takes more space and a lot longer to do.  
+
+```bash
+export POLYGON_DATA_DIR=`pwd`/data/files.polygon.io
+for year in 2024 2023 2022 2021; do \
+    rclone copy -P s3polygon:flatfiles/us_stocks_sip/minute_aggs_v1/$year \
+    $POLYGON_DATA_DIR/flatfiles/us_stocks_sip/minute_aggs_v1/$year; \
+done
+```
+
+If you set the `ZIPLINE_ROOT` environment variable (recommended and likely necessary because the default of `~/.zipline` is probably not what you'll want) and copy your `extension.py` config there then you don't need to put `-e extension.py` on the `zipline` command line.
+
+This ingestion for 10 years of minute bars took around 10 hours on my Mac using an external hard drive (not SSD).  A big chunk of that was copying from the default tmp dir to the Zipline root (6.3million files for 47GB actual, 63GB used).  I plan to change that `shutil.copy2` to use `shutil.move` and to use a `tmp` dir in Zipline root for temporary files instead of the default which should save an hour or two.  Also the ingestion process is single threaded and could be sped up with some concurrency.
+
+```bash
+zipline ingest -b polygon-minute
+```
+
+## Using trades flat files.
+This takes a lot of space for the trades flatfiles (currently the 22 years of trades take around 4TB) and a fair bit of time to convert to minute aggregates.  The benefit though is the whole trading day is covered from premarket open to after hours close.  Also the current conversion logic ignores trade corrections, official close updates, and the TRF "dark pool" trades (because they are not reported when they occurred nor were they offered on the exchanges).  That is to make the aggregates be as good of a simulation of real-time as we can do for algo training and backtesting.  Details in the `trades_to_custom_aggs` function in `zipline_polygon_bundle/trades.py`.
+
+The conversion process creates `.csv.gz` files in the same format as Polygon flatfiles in the custom assets dir, which is `$ZIPLINE_ROOT/data/polygon_custom_assets` by default.  So while `$ZIPLINE_ROOT` needs to be writable, the Polygon flatfiles (`$POLYGON_DATA_DIR`) can be read-only.
+
+Get AWS S3 CLI in the usual way: https://docs.aws.amazon.com/cli/latest/reference/s3/
+
+```bash
+aws s3 sync s3://flatfiles/us_stocks_sip/trades_v1 $POLYGON_DATA_DIR/flatfiles/us_stocks_sip/trades_v1 --checksum-mode ENABLED  --endpoint-url https://files.polygon.io
+```
+
+## `extension.py`
+
+If you set the `ZIPLINE_ROOT` environment variable (recommended and likely necessary because the default of `~/.zipline` is probably not what you'll want) and copy your `extension.py` config there then you don't need to put `-e extension.py` on the `zipline` command line.
+
+If you leave out the `start_date` and/or `end_date` args then `register_polygon_equities_bundle` will scan for the dates of the first and last trade file in `$POLYGON_DATA_DIR` and use them respectively.
+
+The `NYSE_ALL_HOURS` calendar (defined in `zipline_polygon_bundle/nyse_all_hours_calendar.py`) uses open and close times for the entire trading day from premarket open to after hours close.
+
+Right now `agg_time="1min"` is the only supported aggregate duration because Zipline can only deal with day or minute duration aggregates.
+
+```python
+from zipline_polygon_bundle import register_polygon_equities_bundle, register_nyse_all_hours_calendar, NYSE_ALL_HOURS
+from exchange_calendars.calendar_helpers import parse_date
+# from zipline.utils.calendar_utils import get_calendar
+
+# Register the NYSE_ALL_HOURS ExchangeCalendar.
+register_nyse_all_hours_calendar()
+
+register_polygon_equities_bundle(
+    "polygon-trades",
+    calendar_name=NYSE_ALL_HOURS,
+    # start_date=parse_date("2020-01-03", raise_oob=False),
+    # end_date=parse_date("2021-01-29", raise_oob=False),
+    agg_time="1min",
+    minutes_per_day=16 * 60,
+)
+```
+
+As with the daily and minute aggs, the POLYGON_API_KEY is needed for the split and dividend data.  Also coming is SID assignment across ticker changes using the Polygon tickers API data.
+
+```bash
+zipline ingest -b polygon-trades
+```
+
+# License is Affero General Public License v3 (AGPL v3)
+The content of this project is Copyright (C) 2024 Fovi LLC and authored by James P. White (https://www.linkedin.com/in/jamespaulwhite/).  It is distributed under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE (AGPL) Version 3 (See LICENSE file).
+
+The AGPL doesn't put any restrictions on personal use but people using this in a service for others have obligations.  If you have commerical purposes and those distribution requirements don't work for you, feel free to contact me (mailto:jim@fovi.com) about other licensing terms.

{zipline_polygon_bundle-0.2.0.dev1 → zipline_polygon_bundle-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = 'zipline_polygon_bundle'
-version = '0.2.0dev1'
+version = '0.2.3'
 description = 'A zipline-reloaded data provider bundle for Polygon.io'
 authors = [
     { name = 'Jim White', email = 'jim@fovi.com' },
@@ -14,44 +14,37 @@ classifiers = [
     'Operating System :: OS Independent',
 ]
 
-[project.urls]
-Repository = 'https://github.com/fovi-llc/zipline-polygon-bundle'
+requires-python = ">= 3.10,<4.0"
 
-[tool.poetry]
-name = 'zipline-polygon-bundle'
-version = '0.2.0dev1'
-description = 'A zipline-reloaded data provider bundle for Polygon.io'
-authors = ['Jim White <jim@fovi.com>']
-license = 'AGPL-3.0'
-readme = 'README.md'
-keywords = ['zipline', 'data-bundle', 'finance']
-classifiers = [
-    'Programming Language :: Python :: 3',
-    'License :: OSI Approved :: GNU Affero General Public License v3',
-    'Operating System :: OS Independent',
+dependencies = [
+    "fsspec>=2024.10",
+    "filelock>=3.16.0",
+    "polygon-api-client>=1.14.2",
+    "pandas>=2.2,<3",
+    # "pandas-market-calendars>=4.4.2",
+    # "pandas-ta>=0.3", # pandas-ta install doesn't work with poetry for some reason.
+    # It is used in compute_signals.py which we're not using yet.
+    "pytz>=2018.5",
+    "requests>=2.9.1",
+    "bcolz-zipline>=1.2.11",
+    # There was an issue in PyArrow 19 which is probably fixed but don't remember how to test it.
+    "pyarrow>=18.1.0,<19",
+    "numpy<2",
+    "toolz>=0.8.2",
+    "zipline-arrow>=3.2.2",
+    # "zipline-arrow = { git = 'https://github.com/fovi-llc/zipline-arrow.git' }"
 ]
 
-[tool.poetry.dependencies]
-fsspec = ">=2024.10"
-python = ">=3.9,<4.0"
-polygon-api-client = ">=1.14.2"
-pandas = ">=2.2,<3"
-pandas-market-calendars = ">=4.4.2"
-pandas_ta = ">=0.3"
-pytz = ">=2018.5"
-requests = ">=2.9.1"
-bcolz-zipline = ">=1.2.11"
-pyarrow = ">=18.1.0,<19"
-numpy = "<2"
-toolz = ">=0.8.2"
-zipline-reloaded = ">=3.1"
+[project.urls]
+Repository = 'https://github.com/fovi-llc/zipline-polygon-bundle'
+
 
-[tool.poetry.dev-dependencies]
+[poetry.group.dev.dependencies]
 pytest = "*"
 
 [build-system]
 build-backend = "poetry.core.masonry.api"
-requires = ["poetry_core>=1.0.0"]
+requires = ["poetry_core>=2.1.0"]
 
 [tool.pytest.ini_options]
 # https://docs.pytest.org/en/stable/how-to/capture-warnings.html#controlling-warnings

{zipline_polygon_bundle-0.2.0.dev1 → zipline_polygon_bundle-0.2.3}/zipline_polygon_bundle/__init__.py

@@ -1,17 +1,17 @@
 from .bundle import (
     register_polygon_equities_bundle,
     symbol_to_upper,
-    polygon_equities_bundle_day,
-    polygon_equities_bundle_minute,
+    ingest_polygon_equities_bundle
 )
 
 from .config import PolygonConfig
 from .nyse_all_hours_calendar import NYSE_ALL_HOURS, register_nyse_all_hours_calendar
 from .concat_all_aggs import concat_all_aggs_from_csv, generate_csv_agg_tables
 from .adjustments import load_splits, load_dividends, load_conditions
-from .trades import trades_schema, trades_dataset, cast_trades, date_to_path
+from .trades import trades_schema, trades_dataset, cast_trades
 from .trades import custom_aggs_partitioning, custom_aggs_schema, trades_to_custom_aggs, convert_trades_to_custom_aggs
-from .trades import get_custom_aggs_dates, generate_csv_trades_tables, compute_signals_for_all_custom_aggs
+from .trades import get_aggs_dates, generate_csv_trades_tables
+# from .compute_signals import compute_signals_for_all_custom_aggs
 from .quotes import quotes_schema, quotes_dataset, cast_quotes
 # from .tickers_and_names import load_all_tickers, merge_tickers, ticker_names_from_merged_tickers, get_ticker_universe
 from .tickers_and_names import PolygonAssets, get_ticker_universe
@@ -22,8 +22,7 @@ __all__ = [
     "register_nyse_all_hours_calendar",
     "NYSE_ALL_HOURS",
     "symbol_to_upper",
-    "polygon_equities_bundle_day",
-    "polygon_equities_bundle_minute",
+    "ingest_polygon_equities_bundle",
     "PolygonConfig",
     "concat_all_aggs_from_csv",
     "generate_csv_agg_tables",
@@ -33,14 +32,13 @@ __all__ = [
     "trades_schema",
     "trades_dataset",
     "cast_trades",
-    "date_to_path",
-    "get_custom_aggs_dates",
+    "get_aggs_dates",
     "generate_csv_trades_tables",
     "custom_aggs_partitioning",
     "custom_aggs_schema",
     "trades_to_custom_aggs",
     "convert_trades_to_custom_aggs",
-    "compute_signals_for_all_custom_aggs",
+    # "compute_signals_for_all_custom_aggs",
     "quotes_schema",
     "quotes_dataset",
     "cast_quotes",

{zipline_polygon_bundle-0.2.0.dev1 → zipline_polygon_bundle-0.2.3}/zipline_polygon_bundle/adjustments.py

@@ -10,19 +10,19 @@ from urllib3 import HTTPResponse
 
 
 def load_polygon_splits(
-    config: PolygonConfig, first_start_end: datetime.date, last_end_date: datetime.date
+    config: PolygonConfig, first_day: pd.Timestamp, last_day: pd.Timestamp
 ) -> pd.DataFrame:
     # N.B. If the schema changes then the filename should change.  We're on v3 now.
     splits_path = config.api_cache_path(
-        start_date=first_start_end, end_date=last_end_date, filename="list_splits"
+        first_day=first_day, last_day=last_day, filename="list_splits"
     )
-    expected_split_count = (last_end_date - first_start_end).days * 3
+    expected_split_count = (last_day - first_day).days * 3
     if not os.path.exists(splits_path):
         client = polygon.RESTClient(api_key=config.api_key)
         splits = client.list_splits(
             limit=1000,
-            execution_date_gte=first_start_end,
-            execution_date_lt=last_end_date + datetime.timedelta(days=1),
+            execution_date_gte=first_day.date(),
+            execution_date_lt=last_day.date() + datetime.timedelta(days=1),
         )
         if splits is HTTPResponse:
             raise ValueError(f"Polygon.list_splits bad HTTPResponse: {splits}")
@@ -32,7 +32,7 @@ def load_polygon_splits(
         splits.to_parquet(splits_path)
         if len(splits) < expected_split_count:
             logging.warning(
-                f"Only got {len(splits)=} from Polygon list_splits (expected {expected_split_count=}).  "
+                f"Only got {len(splits)=} from Polygon list_splits ({expected_split_count=}).  "
                 "This is probably fine if your historical range is short."
             )
         # We will always load from the file to avoid any chance of weird errors.
@@ -41,7 +41,7 @@ def load_polygon_splits(
         print(f"Loaded {len(splits)=} from {splits_path}")
         if len(splits) < expected_split_count:
             logging.warning(
-                f"Only got {len(splits)=} from Polygon list_splits (expected {expected_split_count=}).  "
+                f"Only got {len(splits)=} from Polygon list_splits ({expected_split_count=}).  "
                 "This is probably fine if your historical range is short."
             )
         return splits
@@ -50,11 +50,11 @@ def load_polygon_splits(
 
 def load_splits(
     config: PolygonConfig,
-    first_start_end: datetime.date,
-    last_end_date: datetime.date,
+    first_day: pd.Timestamp,
+    last_day: pd.Timestamp,
     ticker_to_sid: dict[str, int],
 ) -> pd.DataFrame:
-    splits = load_polygon_splits(config, first_start_end, last_end_date)
+    splits = load_polygon_splits(config, first_day=first_day, last_day=last_day)
     splits["sid"] = splits["ticker"].apply(lambda t: ticker_to_sid.get(t, pd.NA))
     splits.dropna(inplace=True)
     splits["sid"] = splits["sid"].astype("int64")
@@ -70,18 +70,18 @@ def load_splits(
 
 
 def load_polygon_dividends(
-    config: PolygonConfig, first_start_date: datetime.date, last_end_date: datetime.date
+    config: PolygonConfig, first_day: pd.Timestamp, last_day: pd.Timestamp
 ) -> pd.DataFrame:
     # N.B. If the schema changes then the filename should change.  We're on v3 now.
     dividends_path = config.api_cache_path(
-        start_date=first_start_date, end_date=last_end_date, filename="list_dividends"
+        first_day=first_day, last_day=last_day, filename="list_dividends"
     )
     if not os.path.exists(dividends_path):
         client = polygon.RESTClient(api_key=config.api_key)
         dividends = client.list_dividends(
             limit=1000,
-            record_date_gte=first_start_date,
-            pay_date_lt=last_end_date + datetime.timedelta(days=1),
+            record_date_gte=first_day.date(),
+            pay_date_lt=last_day.date() + datetime.timedelta(days=1),
         )
         if dividends is HTTPResponse:
             raise ValueError(f"Polygon.list_dividends bad HTTPResponse: {dividends}")
@@ -104,35 +104,30 @@ def load_polygon_dividends(
 
 
 def load_chunked_polygon_dividends(
-    config: PolygonConfig, first_start_end: datetime.date, last_end_date: datetime.date
+    config: PolygonConfig, first_day: pd.Timestamp,
+    last_day: pd.Timestamp
 ) -> pd.DataFrame:
     dividends_list = []
-    next_start_end = first_start_end
-    while next_start_end < last_end_date:
+    next_start_end = first_day
+    while next_start_end < last_day:
         # We want at most a month of dividends at a time.  They should end on the last day of the month.
-        # So the next_end_date is the day before the first day of the next month.
-        first_of_next_month = datetime.date(
-            next_start_end.year + (next_start_end.month // 12),
-            (next_start_end.month % 12) + 1,
-            1,
-        )
-        next_end_date = first_of_next_month - datetime.timedelta(days=1)
-        if next_end_date > last_end_date:
-            next_end_date = last_end_date
+        next_end_date = next_start_end + pd.offsets.MonthEnd()
+        if next_end_date > last_day:
+            next_end_date = last_day
         dividends_list.append(
-            load_polygon_dividends(config, next_start_end, next_end_date)
+            load_polygon_dividends(config, first_day=next_start_end, last_day=next_end_date)
         )
-        next_start_end = next_end_date + datetime.timedelta(days=1)
+        next_start_end = next_end_date + pd.Timedelta(days=1)
     return pd.concat(dividends_list)
 
 
 def load_dividends(
     config: PolygonConfig,
-    first_start_end: datetime.date,
-    last_end_date: datetime.date,
+    first_day: pd.Timestamp,
+    last_day: pd.Timestamp,
     ticker_to_sid: dict[str, int],
 ) -> pd.DataFrame:
-    dividends = load_chunked_polygon_dividends(config, first_start_end, last_end_date)
+    dividends = load_chunked_polygon_dividends(config, first_day=first_day, last_day=last_day)
     dividends["sid"] = dividends["ticker"].apply(lambda t: ticker_to_sid.get(t, pd.NA))
     dividends.dropna(how="any", inplace=True)
     dividends["sid"] = dividends["sid"].astype("int64")
@@ -159,7 +154,7 @@ def load_conditions(config: PolygonConfig) -> pd.DataFrame:
     # The API doesn't use dates for the condition codes but this is a way to provide control over caching.
     # Main thing is to get the current conditions list but we don't want to call more than once a day.
     conditions_path = config.api_cache_path(
-        start_date=config.start_timestamp.date(), end_date=config.end_timestamp.date(), filename="conditions"
+        first_day=config.start_timestamp, last_day=config.end_timestamp, filename="conditions"
     )
     expected_conditions_count = 100
     if not os.path.exists(conditions_path):