Merge pull request #1 from ThetaData-API/3.1

3.1

Author: baileydanseglio

dev/test.py

@@ -0,0 +1,30 @@
+import pandas as pd
+from datetime import date
+from thetadata import ThetaClient, OptionReqType, OptionRight, DateRange
+
+
+def end_of_day() -> pd.DataFrame:
+    """Request end-of-day data between 7/18/2022 and 7/22/2022."""
+    # Create a ThetaClient
+    client = ThetaClient(timeout=180)
+
+    # Connect to the Terminal
+    with client.connect():
+
+        # Make the request
+        data = client.get_hist_option(
+            req=OptionReqType.EOD,
+            root="AAPL",
+            exp=date(2022, 8, 12),
+            strike=160,
+            right=OptionRight.CALL,
+            date_range=DateRange(date(2022, 8, 3), date(2022, 8, 3)),
+            progress_bar=True,
+        )
+
+    return data
+
+
+if __name__ == "__main__":
+    data = end_of_day()
+    print(data)

thetadata/client.py

@@ -1,13 +1,13 @@
 """Module that contains Theta Client class."""
-from typing import Optional, Callable
+from typing import Optional
 from contextlib import contextmanager
-from functools import wraps
-from datetime import datetime, date
+
 import socket
+
+from pandas import DataFrame
 from tqdm import tqdm
 import pandas as pd
 
-from . import exceptions
 from .enums import *
 from .parsing import (
     Header,
@@ -101,6 +101,7 @@ def get_hist_option(
         strike: float,
         right: OptionRight,
         date_range: DateRange,
+        interval_size: int = 0,
         progress_bar: bool = False,
     ) -> pd.DataFrame:
         """
@@ -112,6 +113,7 @@ def get_hist_option(
         :param strike:         The strike price in USD, rounded to 1/10th of a cent.
         :param right:          The right of an option.
         :param date_range:     The dates to fetch.
+        :param interval_size:        The interval size in milliseconds. Applicable only to OHLC & QUOTE requests.
         :param progress_bar:   Print a progress bar displaying download progress.
 
         :return:               The requested data as a pandas DataFrame.
@@ -125,7 +127,7 @@ def get_hist_option(
         end_fmt = _format_date(date_range.end)
 
         # send request
-        hist_msg = f"MSG_CODE={MessageType.HIST.value}&START_DATE={start_fmt}&END_DATE={end_fmt}&root={root}&exp={exp_fmt}&strike={strike}&right={right.value}&sec={SecType.OPTION.value}&req={req.value}\n"
+        hist_msg = f"MSG_CODE={MessageType.HIST.value}&START_DATE={start_fmt}&END_DATE={end_fmt}&root={root}&exp={exp_fmt}&strike={strike}&right={right.value}&sec={SecType.OPTION.value}&req={req.value}&IVL={interval_size}\n"
         self._server.sendall(hist_msg.encode("utf-8"))
 
         # parse response header

thetadata/parsing.py

@@ -208,6 +208,7 @@ def _post_process(cls, df: DataFrame) -> None:
 
         if DataType.PRICE_TYPE in df.columns:
             # replace price type column with price multipliers
+
             df[DataType.PRICE_TYPE] = _to_price_mul(df[DataType.PRICE_TYPE])
 
             # multiply prices by price multipliers