REF: Upgrade black and ruff

Author: cjdsellers

databento/common/constants.py

@@ -21,9 +21,7 @@
 
 
 DEFINITION_TYPE_MAX_MAP: Final = {
-    x[0]: np.iinfo(x[1]).max
-    for x in InstrumentDefMsg._dtypes
-    if not isinstance(x[1], str)
+    x[0]: np.iinfo(x[1]).max for x in InstrumentDefMsg._dtypes if not isinstance(x[1], str)
 }
 
 INT64_NULL: Final = 9223372036854775807

databento/common/cram.py

@@ -1,6 +1,7 @@
 """
 Functions for handling challenge-response authentication.
 """
+
 import argparse
 import hashlib
 import os

databento/common/dbnstore.py

@@ -10,18 +10,16 @@
 from io import BytesIO
 from os import PathLike
 from pathlib import Path
-from typing import (
-    IO,
-    TYPE_CHECKING,
-    Any,
-    BinaryIO,
-    Callable,
-    Final,
-    Literal,
-    Protocol,
-    TextIO,
-    overload,
-)
+from typing import IO
+from typing import TYPE_CHECKING
+from typing import Any
+from typing import BinaryIO
+from typing import Callable
+from typing import Final
+from typing import Literal
+from typing import Protocol
+from typing import TextIO
+from typing import overload
 
 import databento_dbn
 import numpy as np

databento/common/enums.py

@@ -4,7 +4,8 @@
 from enum import Flag
 from enum import IntFlag
 from enum import unique
-from typing import Callable, TypeVar
+from typing import Callable
+from typing import TypeVar
 
 
 M = TypeVar("M", bound=Enum)

databento/common/error.py

@@ -30,8 +30,7 @@ def __init__(
                 http_body = http_body.decode("utf-8")
             except UnicodeDecodeError:
                 http_body = (
-                    "<Could not decode body as utf-8. "
-                    "Please report to [email protected]>"
+                    "<Could not decode body as utf-8. Please report to [email protected]>"
                 )
 
         self.http_status = http_status

databento/common/publishers.py

@@ -10,7 +10,6 @@
 # ruff: noqa: C901
 
 
-
 @unique
 @coercible
 class Venue(StringyMixin, str, Enum):
@@ -419,6 +418,7 @@ def description(self) -> str:
             return "Long-Term Stock Exchange, Inc."
         raise ValueError("Unexpected Venue value")
 
+
 @unique
 @coercible
 class Dataset(StringyMixin, str, Enum):
@@ -719,6 +719,7 @@ def description(self) -> str:
             return "Databento Equities Max"
         raise ValueError("Unexpected Dataset value")
 
+
 @unique
 @coercible
 class Publisher(StringyMixin, str, Enum):
@@ -1301,6 +1302,7 @@ def to_int(self) -> int:
         if self == Publisher.DBEQ_MAX_LTSE:
             return 80
         raise ValueError("Invalid Publisher")
+
     @property
     def venue(self) -> Venue:
         """
@@ -1467,6 +1469,7 @@ def venue(self) -> Venue:
         if self == Publisher.DBEQ_MAX_LTSE:
             return Venue.LTSE
         raise ValueError("Unexpected Publisher value")
+
     @property
     def dataset(self) -> Dataset:
         """

databento/common/symbology.py

@@ -10,7 +10,10 @@
 from io import TextIOWrapper
 from os import PathLike
 from pathlib import Path
-from typing import Any, ClassVar, NamedTuple, TextIO
+from typing import Any
+from typing import ClassVar
+from typing import NamedTuple
+from typing import TextIO
 
 import pandas as pd
 from databento_dbn import UNDEF_TIMESTAMP
@@ -243,9 +246,7 @@ def insert_metadata(self, metadata: Metadata) -> None:
             return
 
         stype_in = SType(metadata.stype_in) if metadata.stype_in is not None else None
-        stype_out = (
-            SType(metadata.stype_out) if metadata.stype_out is not None else None
-        )
+        stype_out = SType(metadata.stype_out) if metadata.stype_out is not None else None
 
         for symbol_in, entries in metadata.mappings.items():
             for entry in entries:

databento/common/types.py

@@ -1,4 +1,7 @@
-from typing import Callable, Generic, TypeVar, Union
+from typing import Callable
+from typing import Generic
+from typing import TypeVar
+from typing import Union
 
 import databento_dbn
 
@@ -23,6 +26,8 @@
 ExceptionCallback = Callable[[Exception], None]
 
 _T = TypeVar("_T")
+
+
 class Default(Generic[_T]):
     """
     A container for a default value. This is to be used when a callable wants

databento/common/validation.py

@@ -38,8 +38,7 @@ def validate_path(value: PathLike[str] | str, param: str) -> Path:
         return Path(value)
     except TypeError:
         raise TypeError(
-            f"The `{param}` was not a valid path type. "
-            "Use any of [str, bytes, os.PathLike].",
+            f"The `{param}` was not a valid path type. " "Use any of [str, bytes, os.PathLike].",
         ) from None
 
 
@@ -253,8 +252,7 @@ def validate_smart_symbol(symbol: str) -> str:
     if len(tokens) > 3 or not all(tokens):
         raise ValueError(
             f"value `{symbol}` is not a valid smart symbol format; ",
-            "valid formats are [ROOT], [ROOT].[ASSET_CLASS], and "
-            "[ROOT].[ROLL_RULE].[RANK].",
+            "valid formats are [ROOT], [ROOT].[ASSET_CLASS], and " "[ROOT].[ROLL_RULE].[RANK].",
         )
 
     if len(tokens) == 3:

databento/historical/api/batch.py

@@ -145,19 +145,19 @@ def submit_job(
             "stype_in": str(stype_in_valid),
             "stype_out": str(validate_enum(stype_out, SType, "stype_out")),
             "encoding": str(validate_enum(encoding, Encoding, "encoding")),
-            "compression": str(validate_enum(compression, Compression, "compression"))
-            if compression
-            else None,
+            "compression": (
+                str(validate_enum(compression, Compression, "compression")) if compression else None
+            ),
             "pretty_px": pretty_px,
             "pretty_ts": pretty_ts,
             "map_symbols": map_symbols,
             "split_symbols": split_symbols,
             "split_duration": str(
                 validate_enum(split_duration, SplitDuration, "split_duration"),
             ),
-            "packaging": str(validate_enum(packaging, Packaging, "packaging"))
-            if packaging
-            else None,
+            "packaging": (
+                str(validate_enum(packaging, Packaging, "packaging")) if packaging else None
+            ),
             "delivery": str(validate_enum(delivery, Delivery, "delivery")),
         }