use line-length=120 for better code readability

Author: Valeriya Popova

examples/_sqlalchemy_example/example.py

@@ -24,9 +24,7 @@ def simple_select(conn):
 
 
 def simple_insert(conn):
-    stm = Episodes.__table__.insert().values(
-        series_id=3, season_id=6, episode_id=1, title="TBD"
-    )
+    stm = Episodes.__table__.insert().values(series_id=3, season_id=6, episode_id=1, title="TBD")
     conn.execute(stm)
 
 
@@ -51,9 +49,7 @@ def test_types(conn):
     conn.execute(stm)
 
     # GROUP BY
-    stm = sa.select(types_tb.c.str, sa.func.max(types_tb.c.num)).group_by(
-        types_tb.c.str
-    )
+    stm = sa.select(types_tb.c.str, sa.func.max(types_tb.c.num)).group_by(types_tb.c.str)
     rs = conn.execute(stm)
     for x in rs:
         print(x)
@@ -171,9 +167,9 @@ def run_example_core(engine):
         simple_insert(conn)
 
         # simple join
-        stm = sa.select(
-            [Episodes.__table__.join(Series, Episodes.series_id == Series.series_id)]
-        ).where(sa.and_(Series.series_id == 1, Episodes.season_id == 1))
+        stm = sa.select([Episodes.__table__.join(Series, Episodes.series_id == Series.series_id)]).where(
+            sa.and_(Series.series_id == 1, Episodes.season_id == 1)
+        )
         rs = conn.execute(stm)
         for row in rs:
             print(f"{row.series_title}({row.episode_id}): {row.title}")

examples/access-token-credentials/main.py

@@ -24,9 +24,7 @@ def main():
     driver = ydb.Driver(
         endpoint=os.getenv("YDB_ENDPOINT"),
         database=os.getenv("YDB_DATABASE"),
-        credentials=ydb.AccessTokenCredentials(
-            os.getenv("YDB_ACCESS_TOKEN_CREDENTIALS")
-        ),
+        credentials=ydb.AccessTokenCredentials(os.getenv("YDB_ACCESS_TOKEN_CREDENTIALS")),
     )
 
     # Start driver context manager.

examples/aio/driver_example.py

@@ -6,9 +6,7 @@
 async def describe_database():
     endpoint = os.getenv("YDB_ENDPOINT")
     database = os.getenv("YDB_DATABASE")
-    driver = ydb.aio.Driver(
-        endpoint=endpoint, database=database
-    )  # Creating new database driver to execute queries
+    driver = ydb.aio.Driver(endpoint=endpoint, database=database)  # Creating new database driver to execute queries
 
     await driver.wait(timeout=10)  # Wait until driver can execute calls
 

examples/asyncio/main.py

@@ -10,31 +10,23 @@ async def execute_query(pool):
     with pool.async_checkout() as session_holder:
         try:
             # wait for the session checkout to complete.
-            session = await asyncio.wait_for(
-                asyncio.wrap_future(session_holder), timeout=5
-            )
+            session = await asyncio.wait_for(asyncio.wrap_future(session_holder), timeout=5)
         except asyncio.TimeoutError:
             raise ydb.SessionPoolEmpty()
 
         return await asyncio.wrap_future(
             session.transaction().async_execute(
                 "select 1 as cnt;",
                 commit_tx=True,
-                settings=ydb.BaseRequestSettings()
-                .with_timeout(3)
-                .with_operation_timeout(2),
+                settings=ydb.BaseRequestSettings().with_timeout(3).with_operation_timeout(2),
             )
         )
 
 
 async def main():
-    with ydb.Driver(
-        endpoint=os.getenv("YDB_ENDPOINT"), database=os.getenv("YDB_DATABASE")
-    ) as driver:
+    with ydb.Driver(endpoint=os.getenv("YDB_ENDPOINT"), database=os.getenv("YDB_DATABASE")) as driver:
         # Wait for the driver to become active for requests.
-        await asyncio.wait_for(
-            asyncio.wrap_future(driver.async_wait(fail_fast=True)), timeout=5
-        )
+        await asyncio.wait_for(asyncio.wrap_future(driver.async_wait(fail_fast=True)), timeout=5)
 
         # Create the session pool instance to manage YDB session.
         session_pool = ydb.SessionPool(driver)

examples/basic_example_v1/__main__.py

@@ -9,9 +9,7 @@
         formatter_class=argparse.RawDescriptionHelpFormatter,
         description="""\033[92mYandex.Database examples binary.\x1b[0m\n""",
     )
-    parser.add_argument(
-        "-d", "--database", required=True, help="Name of the database to use"
-    )
+    parser.add_argument("-d", "--database", required=True, help="Name of the database to use")
     parser.add_argument("-e", "--endpoint", required=True, help="Endpoint url to use")
     parser.add_argument("-p", "--path", default="")
     parser.add_argument("-v", "--verbose", default=False, action="store_true")

examples/cloud-function-trivial-example/index.py

@@ -2,9 +2,7 @@
 import ydb
 
 # create driver in global space.
-driver = ydb.Driver(
-    endpoint=os.getenv("YDB_ENDPOINT"), database=os.getenv("YDB_DATABASE")
-)
+driver = ydb.Driver(endpoint=os.getenv("YDB_ENDPOINT"), database=os.getenv("YDB_DATABASE"))
 # Wait for the driver to become active for requests.
 driver.wait(fail_fast=True, timeout=5)
 # Create the session pool instance to manage YDB sessions.

examples/pagination/__main__.py

@@ -9,9 +9,7 @@
         formatter_class=argparse.RawDescriptionHelpFormatter,
         description="""\033[92mYandex.Database example pagination.\x1b[0m\n""",
     )
-    parser.add_argument(
-        "-d", "--database", required=True, help="Name of the database to use"
-    )
+    parser.add_argument("-d", "--database", required=True, help="Name of the database to use")
     parser.add_argument("-e", "--endpoint", required=True, help="Endpoint url to use")
     parser.add_argument("-p", "--path", default="", help="Base path for tables")
 

examples/reservations-bot-demo/cloud_function/controller.py

@@ -16,19 +16,13 @@ class Controller(object):
     def __init__(self, storage: Storage):
         self._storage = storage
 
-    def _find_available_table_id(
-        self, request: ReservationCreateRequest
-    ) -> typing.Optional[int]:
+    def _find_available_table_id(self, request: ReservationCreateRequest) -> typing.Optional[int]:
         table_ids = set(self._storage.list_table_ids(cnt=request.cnt))
-        reserved_table_ids = set(
-            self._storage.find_reserved_table_ids(cnt=request.cnt, dt=request.dt)
-        )
+        reserved_table_ids = set(self._storage.find_reserved_table_ids(cnt=request.cnt, dt=request.dt))
         for table_id in table_ids.difference(reserved_table_ids):
             return table_id
 
-    def maybe_create_reservation(
-        self, request: ReservationCreateRequest
-    ) -> ReservationCreateResponse:
+    def maybe_create_reservation(self, request: ReservationCreateRequest) -> ReservationCreateResponse:
         table_id = self._find_available_table_id(request)
         if table_id is None:
             logging.warning("reservation failed")
@@ -47,16 +41,11 @@ def maybe_create_reservation(
             logging.warning(f"failed to reserve a table due to {repr(e)}")
             return ReservationCreateResponse(success=False)
 
-    def maybe_cancel_reservation(
-        self, request: ReservationCancelRequest
-    ) -> ReservationCancelResponse:
+    def maybe_cancel_reservation(self, request: ReservationCancelRequest) -> ReservationCancelResponse:
         try:
             self._storage.delete_reservation(phone=request.phone, dt=request.dt)
             logging.warning(f"reservation {request.phone} {request.dt} cancelled")
             return ReservationCancelResponse(success=True)
         except Exception as e:
-            logging.warning(
-                f"failed to cancel reservation for {request.phone} "
-                f"{request.dt} due to {repr(e)}"
-            )
+            logging.warning(f"failed to cancel reservation for {request.phone} " f"{request.dt} due to {repr(e)}")
             return ReservationCancelResponse(success=False)

examples/reservations-bot-demo/cloud_function/storage.py

@@ -26,9 +26,7 @@ def transaction(session):
             tables = session_pool.retry_operation_sync(transaction)
             return list(map(lambda x: getattr(x, "table_id"), tables))
 
-    def find_reserved_table_ids(
-        self, *, cnt: int, dt: datetime.datetime
-    ) -> typing.List[int]:
+    def find_reserved_table_ids(self, *, cnt: int, dt: datetime.datetime) -> typing.List[int]:
         query = f"""PRAGMA TablePathPrefix("{self._database}");
         DECLARE $dt AS DateTime;
         DECLARE $reservation_period_minutes AS Int32;

examples/reservations-bot-demo/cloud_function/utils.py

@@ -13,22 +13,17 @@ def make_driver_config(endpoint, database, path):
 
 
 @contextlib.contextmanager
-def session_pool_context(
-    driver_config: ydb.DriverConfig, size=1, workers_threads_count=1
-):
+def session_pool_context(driver_config: ydb.DriverConfig, size=1, workers_threads_count=1):
     with ydb.Driver(driver_config) as driver:
         try:
             logging.info("connecting to the database")
             driver.wait(timeout=15)
         except TimeoutError:
             logging.critical(
-                f"connection failed\n"
-                f"last reported errors by discovery: {driver.discovery_debug_details()}"
+                f"connection failed\n" f"last reported errors by discovery: {driver.discovery_debug_details()}"
             )
             raise
-        with ydb.SessionPool(
-            driver, size=size, workers_threads_count=workers_threads_count
-        ) as session_pool:
+        with ydb.SessionPool(driver, size=size, workers_threads_count=workers_threads_count) as session_pool:
             try:
                 yield session_pool
             except Exception as e:

examples/secondary_indexes_builtin/__main__.py

@@ -13,9 +13,7 @@
         formatter_class=argparse.RawDescriptionHelpFormatter,
         description="""\033[92mYandex.Database examples binary.\x1b[0m\n""",
     )
-    parser.add_argument(
-        "-d", "--database", required=True, help="Name of the database to use"
-    )
+    parser.add_argument("-d", "--database", required=True, help="Name of the database to use")
     parser.add_argument("-e", "--endpoint", required=True, help="Endpoint url to use")
     parser.add_argument("-p", "--path", default="")
     parser.add_argument("-v", "--verbose", default=False, action="store_true")

examples/secondary_indexes_builtin/secondary_indexes_builtin.py

@@ -73,9 +73,7 @@ class Series(object):
     def __init__(self, series_id, title, release_date, info, views, uploaded_user_id):
         self.series_id = series_id
         self.title = title
-        self.release_date = (
-            to_days(release_date) if isinstance(release_date, str) else release_date
-        )
+        self.release_date = to_days(release_date) if isinstance(release_date, str) else release_date
         self.info = info
         self.views = views
         self.uploaded_user_id = uploaded_user_id
@@ -138,17 +136,11 @@ def callee(session):
                 ydb.Column("series_id", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
                 ydb.Column("title", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
                 ydb.Column("info", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
-                ydb.Column(
-                    "release_date", ydb.OptionalType(ydb.PrimitiveType.Datetime)
-                ),
+                ydb.Column("release_date", ydb.OptionalType(ydb.PrimitiveType.Datetime)),
                 ydb.Column("views", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
-                ydb.Column(
-                    "uploaded_user_id", ydb.OptionalType(ydb.PrimitiveType.Uint64)
-                ),
+                ydb.Column("uploaded_user_id", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
             )
-            .with_indexes(
-                ydb.TableIndex("users_index").with_index_columns("uploaded_user_id")
-            ),
+            .with_indexes(ydb.TableIndex("users_index").with_index_columns("uploaded_user_id")),
         )
 
         session.create_table(
@@ -188,9 +180,7 @@ def get_series_data():
         ),
         Series(5, "Fifth episode", "2006-05-01", "Cercei is not good person.", 5000, 1),
         Series(6, "Sixth episode", "2006-06-01", "Tyrion is not big.", 6000, 2),
-        Series(
-            7, "Seventh episode", "2006-07-01", "Tywin should close the door.", 7000, 2
-        ),
+        Series(7, "Seventh episode", "2006-07-01", "Tywin should close the door.", 7000, 2),
         Series(
             8,
             "Eighth episode",
@@ -244,9 +234,7 @@ def callee(session):
 
 
 def run(endpoint, database, path):
-    driver_config = ydb.DriverConfig(
-        endpoint, database, credentials=ydb.credentials_from_env_variables()
-    )
+    driver_config = ydb.DriverConfig(endpoint, database, credentials=ydb.credentials_from_env_variables())
     with ydb.Driver(driver_config) as driver:
         try:
             driver.wait(timeout=5)

examples/time-series-serverless/response.py

@@ -9,9 +9,7 @@ class Response:
     body = {}
 
     def as_dict(self):
-        return dict(
-            status=self.status, headers=self.headers, body=json.dumps(self.body)
-        )
+        return dict(status=self.status, headers=self.headers, body=json.dumps(self.body))
 
 
 class Ok(Response):

examples/topic/reader_async_example.py

@@ -14,9 +14,7 @@ async def connect():
 
 
 async def create_reader_and_close_with_context_manager(db: ydb.aio.Driver):
-    async with db.topic_client.reader(
-        "/database/topic/path", consumer="consumer"
-    ) as reader:  # noqa
+    async with db.topic_client.reader("/database/topic/path", consumer="consumer") as reader:  # noqa
         ...
 
 

examples/topic/reader_example.py

@@ -13,9 +13,7 @@ def connect():
 
 
 def create_reader_and_close_with_context_manager(db: ydb.Driver):
-    with db.topic_client.reader(
-        "/database/topic/path", consumer="consumer", buffer_size_bytes=123
-    ) as reader:
+    with db.topic_client.reader("/database/topic/path", consumer="consumer", buffer_size_bytes=123) as reader:
         while True:
             message = reader.receive_message()  # noqa
             pass

examples/topic/writer_async_example.py

@@ -48,14 +48,10 @@ async def send_messages(writer: ydb.TopicWriterAsyncIO):
     )  # send few messages by one call
 
     # with meta
-    await writer.write(
-        ydb.TopicWriterMessage("asd", seqno=123, created_at=datetime.datetime.now())
-    )
+    await writer.write(ydb.TopicWriterMessage("asd", seqno=123, created_at=datetime.datetime.now()))
 
 
-async def send_message_without_block_if_internal_buffer_is_full(
-    writer: ydb.TopicWriterAsyncIO, msg
-) -> bool:
+async def send_message_without_block_if_internal_buffer_is_full(writer: ydb.TopicWriterAsyncIO, msg) -> bool:
     try:
         # put message to internal queue for send, but if buffer is full - fast return
         # without wait
@@ -99,9 +95,7 @@ async def send_messages_and_wait_all_commit_with_results(
     await writer.flush()
 
 
-async def switch_messages_with_many_producers(
-    writers: Dict[str, ydb.TopicWriterAsyncIO], messages: List[str]
-):
+async def switch_messages_with_many_producers(writers: Dict[str, ydb.TopicWriterAsyncIO], messages: List[str]):
     futures = []  # type:  List[asyncio.Future]
 
     for msg in messages:

examples/topic/writer_example.py

@@ -55,19 +55,13 @@ def send_messages(writer: ydb.TopicWriter):
     # full forms
     writer.write(ydb.TopicWriterMessage("mess"))  # send text
     writer.write(ydb.TopicWriterMessage(bytes([1, 2, 3])))  # send bytes
-    writer.write(
-        [ydb.TopicWriterMessage("mess-1"), ydb.TopicWriterMessage("mess-2")]
-    )  # send few messages by one call
+    writer.write([ydb.TopicWriterMessage("mess-1"), ydb.TopicWriterMessage("mess-2")])  # send few messages by one call
 
     # with meta
-    writer.write(
-        ydb.TopicWriterMessage("asd", seqno=123, created_at=datetime.datetime.now())
-    )
+    writer.write(ydb.TopicWriterMessage("asd", seqno=123, created_at=datetime.datetime.now()))
 
 
-def send_message_without_block_if_internal_buffer_is_full(
-    writer: ydb.TopicWriter, msg
-) -> bool:
+def send_message_without_block_if_internal_buffer_is_full(writer: ydb.TopicWriter, msg) -> bool:
     try:
         # put message to internal queue for send, but if buffer is full - fast return
         # without wait
@@ -123,9 +117,7 @@ def send_messages_and_wait_all_commit_with_results(writer: ydb.TopicWriter):
             raise future.exception()
 
 
-def switch_messages_with_many_producers(
-    writers: Dict[str, ydb.TopicWriter], messages: List[str]
-):
+def switch_messages_with_many_producers(writers: Dict[str, ydb.TopicWriter], messages: List[str]):
     futures = []  # type:  List[Future]
 
     for msg in messages:

examples/ttl/__main__.py

@@ -8,9 +8,7 @@
         formatter_class=argparse.RawDescriptionHelpFormatter,
         description="""\033[92mYandex.Database examples binary.\x1b[0m\n""",
     )
-    parser.add_argument(
-        "-d", "--database", required=True, help="Name of the database to use"
-    )
+    parser.add_argument("-d", "--database", required=True, help="Name of the database to use")
     parser.add_argument("-e", "--endpoint", required=True, help="Endpoint url to use")
     parser.add_argument("-p", "--path", default="")