Merge pull request #6 from ThetaData-API/3.1

Added get_req method. Allows to send a raw request to the terminal. D…

Author: ThetaData-API

dev/dev.py

@@ -50,21 +50,23 @@ def strikes_crash() -> pd.DataFrame:
 
 def crash() -> pd.DataFrame:
     # Create a ThetaClient
-    client = ThetaClient(username="my_theta_data_email", passwd="my_thetadata_passwd")
+    client = ThetaClient()
 
     # Connect to the Terminal
     with client.connect():
-        # Make the request
-        data = client.get_hist_option(
-            req=OptionReqType.QUOTE,
-            root="AAPL",
-            exp=date(2022, 7, 15),
-            strike=150,
-            right=OptionRight.CALL,
-            date_range=DateRange(date(2022, 7, 1), date(2022, 7, 7)),
-            progress_bar=False,
-        )
+        with open("raw.txt") as f:
+            # Make the request
+            s = f.readline()
+            count = 0
 
+            while s is not None:
+                try:
+                    if count == 109:
+                        data = client.get_req(s)
+                except thetadata.NoData:
+                    x = 0
+                count += 1
+                s = f.readline()
     return data
 
 

thetadata/client.py

@@ -59,7 +59,7 @@ def __init__(self, port: int = 11000, timeout: Optional[float] = 60,
             Thread(target=launch_terminal, args=[username, passwd]).start()
         else:
             warnings.warn("You must specify a username and passwd to access data!"
-                          " If you have a terminal already running, you can ignore this message")
+                          " If you have a terminal already running, you can ignore this message.")
 
     @contextmanager
     def connect(self):
@@ -247,3 +247,22 @@ def get_last_option(
         )
 
         return body
+
+    def get_req(
+        self,
+        req: str,
+    ) -> pd.DataFrame:
+        assert self._server is not None, _NOT_CONNECTED_MSG
+        # send request
+        print(req)
+        self._server.sendall(req.encode("utf-8"))
+
+        # parse response header
+        header_data = self._server.recv(20)
+        header: Header = Header.parse(req, header_data)
+
+        # parse response body
+        body_data = self._recv(header.size, progress_bar=False)
+        body: DataFrame = TickBody.parse(req, header, body_data)
+        return body
+