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: