Add stocks demo scripts (#377)

* added demo scripts

Author: justinpolygon

examples/rest/stocks-aggregates_bars.py

@@ -0,0 +1,27 @@
+from polygon import RESTClient
+
+# docs
+# https://polygon.io/docs/stocks/get_v2_aggs_ticker__stocksticker__range__multiplier___timespan___from___to
+# https://polygon-api-client.readthedocs.io/en/latest/Aggs.html#polygon.RESTClient.get_aggs
+
+# API key injected below for easy use. If not provided, the script will attempt
+# to use the environment variable "POLYGON_API_KEY".
+#
+# setx POLYGON_API_KEY "<your_api_key>"   <- windows
+# export POLYGON_API_KEY="<your_api_key>" <- mac/linux
+#
+# Note: To persist the environment variable you need to add the above command
+# to the shell startup script (e.g. .bashrc or .bash_profile.
+#
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+aggs = client.get_aggs(
+    "AAPL",
+    1,
+    "day",
+    "2023-01-30",
+    "2023-02-03",
+)
+
+print(aggs)

examples/rest/stocks-aggregates_bars_extra.py

@@ -0,0 +1,78 @@
+# This code retrieves stock market data for a specific stock using the
+# Polygon REST API and writes it to a CSV file. It uses the "polygon"
+# library to communicate with the API and the "csv" library to write
+# the data to a CSV file. The script retrieves data for the stock "AAPL"
+# for the dates "2023-01-30" to "2023-02-03" in 1 hour intervals. The
+# resulting data includes the open, high, low, close, volume, vwap,
+# timestamp, transactions, and otc values for each hour. The output is
+# then printed to the console.
+from polygon import RESTClient
+from polygon.rest.models import (
+    Agg,
+)
+import csv
+import datetime
+import io
+
+# docs
+# https://polygon.io/docs/stocks/get_v2_aggs_ticker__stocksticker__range__multiplier___timespan___from___to
+# https://polygon-api-client.readthedocs.io/en/latest/Aggs.html#polygon.RESTClient.get_aggs
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+aggs = client.get_aggs(
+    "AAPL",
+    1,
+    "hour",
+    "2023-01-30",
+    "2023-02-03",
+)
+
+print(aggs)
+
+# headers
+headers = [
+    "timestamp",
+    "open",
+    "high",
+    "low",
+    "close",
+    "volume",
+    "vwap",
+    "transactions",
+    "otc",
+]
+
+# creating the csv string
+csv_string = io.StringIO()
+writer = csv.DictWriter(csv_string, fieldnames=headers)
+
+# writing headers
+writer.writeheader()
+
+# writing data
+for agg in aggs:
+
+    # verify this is an agg
+    if isinstance(agg, Agg):
+
+        # verify this is an int
+        if isinstance(agg.timestamp, int):
+
+            writer.writerow(
+                {
+                    "timestamp": datetime.datetime.fromtimestamp(agg.timestamp / 1000),
+                    "open": agg.open,
+                    "high": agg.high,
+                    "low": agg.low,
+                    "close": agg.close,
+                    "volume": agg.volume,
+                    "vwap": agg.vwap,
+                    "transactions": agg.transactions,
+                    "otc": agg.otc,
+                }
+            )
+
+# printing the csv string
+print(csv_string.getvalue())

examples/rest/stocks-conditions.py

@@ -0,0 +1,13 @@
+from polygon import RESTClient
+
+# docs
+# https://polygon.io/docs/stocks/get_v3_reference_conditions
+# https://polygon-api-client.readthedocs.io/en/latest/Reference.html#list-conditions
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+conditions = []
+for c in client.list_conditions(limit=1000):
+    conditions.append(c)
+print(conditions)

examples/rest/stocks-daily_open_close.py

@@ -0,0 +1,16 @@
+from polygon import RESTClient
+
+# docs
+# https://polygon.io/docs/stocks/get_v1_open-close__stocksticker___date
+# https://polygon-api-client.readthedocs.io/en/latest/Aggs.html#get-daily-open-close-agg
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+# make request
+request = client.get_daily_open_close_agg(
+    "AAPL",
+    "2023-02-07",
+)
+
+print(request)

examples/rest/stocks-dividends.py

@@ -0,0 +1,13 @@
+from polygon import RESTClient
+
+# docs
+# https://polygon.io/docs/stocks/get_v3_reference_dividends
+# https://polygon-api-client.readthedocs.io/en/latest/Reference.html#list-dividends
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+dividends = []
+for d in client.list_dividends("MSFT", limit=1000):
+    dividends.append(d)
+print(dividends)

examples/rest/stocks-exchanges.py

@@ -0,0 +1,27 @@
+from polygon import RESTClient
+from polygon.rest.models import (
+    Exchange,
+)
+
+# docs
+# https://polygon.io/docs/stocks/get_v3_reference_exchanges
+# https://polygon-api-client.readthedocs.io/en/latest/Reference.html#get-exchanges
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+exchanges = client.get_exchanges()
+print(exchanges)
+
+# loop over exchanges
+for exchange in exchanges:
+
+    # verify this is an exchange
+    if isinstance(exchange, Exchange):
+
+        # print exchange info
+        print(
+            "{:<15}{} ({})".format(
+                exchange.asset_class, exchange.name, exchange.operating_mic
+            )
+        )

examples/rest/stocks-grouped_daily_bars.py

@@ -0,0 +1,20 @@
+from polygon import RESTClient
+import pprint
+
+# docs
+# https://polygon.io/docs/stocks/get_v2_aggs_grouped_locale_us_market_stocks__date
+# https://polygon-api-client.readthedocs.io/en/latest/Aggs.html#get-grouped-daily-aggs
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+grouped = client.get_grouped_daily_aggs(
+    "2023-02-07",
+)
+
+# print(grouped)
+
+# pprint (short for "pretty-print") is a module that provides a more human-
+# readable output format for data structures.
+pp = pprint.PrettyPrinter(indent=2)
+pp.pprint(grouped)

examples/rest/stocks-last_quote.py

@@ -0,0 +1,14 @@
+from polygon import RESTClient
+
+# docs
+# https://polygon.io/docs/stocks/get_v2_last_nbbo__stocksticker
+# https://polygon-api-client.readthedocs.io/en/latest/Quotes.html#get-last-quote
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+quote = client.get_last_quote(
+    "AAPL",
+)
+
+print(quote)

examples/rest/stocks-last_trade.py

@@ -0,0 +1,14 @@
+from polygon import RESTClient
+
+# docs
+# https://polygon.io/docs/stocks/get_v2_last_trade__stocksticker
+# https://polygon-api-client.readthedocs.io/en/latest/Trades.html#get-last-trade
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+trade = client.get_last_trade(
+    "AAPL",
+)
+
+print(trade)

examples/rest/stocks-market_holidays.py

@@ -0,0 +1,22 @@
+from polygon import RESTClient
+from polygon.rest.models import (
+    MarketHoliday,
+)
+
+# docs
+# https://polygon.io/docs/stocks/get_v1_marketstatus_upcoming
+# https://polygon-api-client.readthedocs.io/en/latest/Reference.html#get-market-holidays
+
+# client = RESTClient("XXXXXX") # hardcoded api_key is used
+client = RESTClient()  # POLYGON_API_KEY environment variable is used
+
+holidays = client.get_market_holidays()
+# print(holidays)
+
+# print date, name, and exchange
+for holiday in holidays:
+
+    # verify this is an exchange
+    if isinstance(holiday, MarketHoliday):
+
+        print("{:<15}{:<15} ({})".format(holiday.date, holiday.name, holiday.exchange))