fix #287

Author: Graeme22

tastytrade/__init__.py

@@ -4,7 +4,7 @@
 API_VERSION = "20251101"
 CERT_URL = "https://api.cert.tastyworks.com"
 VAST_URL = "https://vast.tastyworks.com"
-VERSION = "11.0.4"
+VERSION = "11.0.5"
 
 __version__ = VERSION
 version_str: str = f"tastyware/tastytrade:v{VERSION}"

tastytrade/account.py

@@ -546,8 +546,6 @@ def get(
         :param session: the session to use for the request.
         :param account_number: the account ID to get.
         :param include_closed: whether to include closed accounts in the results
-
-        :return: an account if an ID was provided; otherwise, a single account.
         """
         if account_number:
             data = session._get(f"/customers/me/accounts/{account_number}")

tastytrade/dxfeed/event.py

@@ -41,8 +41,6 @@ def from_stream(cls, data: list[Any]) -> list[Event]:
         a :class:`~tastyworks.streamer.DXFeedStreamer`.
 
         :param data: list of raw quote data from streamer
-
-        :return: list of event objects from data
         """
         objs: list[Event] = []
         size = len(cls.model_fields)

tastytrade/instruments.py

@@ -490,8 +490,8 @@ class Option(TradeableTastytradeData):
     stops_trading_at: datetime
     market_time_instrument_collection: str
     days_to_expiration: int
-    expires_at: datetime
     is_closing_only: bool
+    expires_at: datetime | None = None
     streamer_symbol: str = ""
     listed_market: str | None = None
     halted_at: datetime | None = None

tastytrade/order.py

@@ -133,7 +133,7 @@ class Leg(TastytradeData):
     instrument_type: InstrumentType
     symbol: str
     action: OrderAction
-    quantity: Decimal | None = None
+    quantity: Decimal | int | None = None
     remaining_quantity: Decimal | None = None
     fills: list[FillInfo] | None = None
 
@@ -149,15 +149,13 @@ class TradeableTastytradeData(TastytradeData):
     instrument_type: InstrumentType
     symbol: str
 
-    def build_leg(self, quantity: Decimal | None, action: OrderAction) -> Leg:
+    def build_leg(self, quantity: Decimal | int | None, action: OrderAction) -> Leg:
         """
         Builds an order :class:`Leg` from the dataclass.
 
         :param quantity:
             the quantity of the symbol to trade, set this as `None` for notional orders
         :param action: :class:`OrderAction` to perform, e.g. BUY_TO_OPEN
-
-        :return: a :class:`Leg` object
         """
         return Leg(
             instrument_type=self.instrument_type,

tastytrade/session.py

@@ -376,35 +376,27 @@ async def a_refresh(self) -> None:
     async def a_get_customer(self) -> Customer:
         """
         Gets the customer dict from the API.
-
-        :return: a Tastytrade 'Customer' object in JSON format.
         """
         data = await self._a_get("/customers/me")
         return Customer(**data)
 
     def get_customer(self) -> Customer:
         """
         Gets the customer dict from the API.
-
-        :return: a Tastytrade 'Customer' object in JSON format.
         """
         data = self._get("/customers/me")
         return Customer(**data)
 
     async def a_validate(self) -> bool:
         """
         Validates the current session by sending a request to the API.
-
-        :return: True if the session is valid and False otherwise.
         """
         response = await self.async_client.post("/sessions/validate")
         return response.status_code // 100 == 2
 
     def validate(self) -> bool:
         """
         Validates the current session by sending a request to the API.
-
-        :return: True if the session is valid and False otherwise.
         """
         response = self.sync_client.post("/sessions/validate")
         return response.status_code // 100 == 2
@@ -420,24 +412,19 @@ def serialize(self) -> str:
         del attrs["sync_client"]
         attrs["session_expiration"] = self.session_expiration.strftime(_fmt)
         attrs["streamer_expiration"] = self.streamer_expiration.strftime(_fmt)
+        attrs["headers"] = dict(self.sync_client.headers.copy())
         return json.dumps(attrs)
 
     @classmethod
     def deserialize(cls, serialized: str) -> Self:
         """
         Create a new Session object from a serialized string.
         """
-        deserialized = json.loads(serialized)
+        deserialized: dict[str, Any] = json.loads(serialized)
+        headers = deserialized.pop("headers")
         self = cls.__new__(cls)
         self.__dict__ = deserialized
         base_url = CERT_URL if self.is_test else API_URL
-        headers = {
-            "Accept": "application/json",
-            "Content-Type": "application/json",
-            "Authorization": self.session_token
-            if "user" in deserialized
-            else f"Bearer {self.session_token}",
-        }
         self.session_expiration = datetime.strptime(
             deserialized["session_expiration"], _fmt
         )

tastytrade/utils.py

@@ -29,17 +29,13 @@ class PriceEffect(str, Enum):
 def now_in_new_york() -> datetime:
     """
     Gets the current time in the New York timezone.
-
-    :return: current time as datetime
     """
     return datetime.now(TZ)
 
 
 def today_in_new_york() -> date:
     """
     Gets the current date in the New York timezone.
-
-    :return: current date
     """
     return now_in_new_york().date()
 
@@ -50,8 +46,6 @@ def is_market_open_on(day: date | None = None) -> bool:
     during the given day.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: whether the market opens on given day
     """
     day = day or today_in_new_york()
     date_range = NYSE.valid_days(day, day)
@@ -64,8 +58,6 @@ def get_third_friday(day: date | None = None) -> date:
     or the monthly expiration associated with today's month.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     day = (day or today_in_new_york()).replace(day=1) + timedelta(weeks=2)
     while day.weekday() != 4:  # Friday
@@ -76,8 +68,6 @@ def get_third_friday(day: date | None = None) -> date:
 def get_tasty_monthly() -> date:
     """
     Gets the monthly expiration closest to 45 days from the current date.
-
-    :return: the closest to 45 DTE monthly expiration
     """
     day = today_in_new_york()
     exp1 = get_third_friday(day + timedelta(weeks=4))
@@ -101,8 +91,6 @@ def get_future_fx_monthly(day: date | None = None) -> date:
     Wednesday.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     day = (day or today_in_new_york()).replace(day=1) + timedelta(weeks=1)
     while day.weekday() != 2:  # Wednesday
@@ -120,8 +108,6 @@ def get_future_treasury_monthly(day: date | None = None) -> date:
     business day prior.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     day = day or today_in_new_york()
     last_day = _get_last_day_of_month(day)
@@ -143,8 +129,6 @@ def get_future_metal_monthly(day: date | None = None) -> date:
     which case they expire on the prior business day.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     day = day or today_in_new_york()
     last_day = _get_last_day_of_month(day)
@@ -164,8 +148,6 @@ def get_future_grain_monthly(day: date | None = None) -> date:
     least 2 business days, the last business day of the month.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     day = day or today_in_new_york()
     last_day = _get_last_day_of_month(day)
@@ -185,8 +167,6 @@ def get_future_oil_monthly(day: date | None = None) -> date:
     they expire 7 business days prior to the 25th day of the month.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     last_day = (day or today_in_new_york()).replace(day=25)
     first_day = last_day.replace(day=1)
@@ -201,8 +181,6 @@ def get_future_index_monthly(day: date | None = None) -> date:
     month.
 
     :param day: date to check. If not provided defaults to current NY date.
-
-    :return: the associated monthly
     """
     day = day or today_in_new_york()
     last_day = _get_last_day_of_month(day)
@@ -224,8 +202,6 @@ def _dasherize(s: str) -> str:
     Converts a string from snake case to dasherized.
 
     :param s: string to convert
-
-    :return: dasherized string
     """
     return s.replace("_", "-")