fix several pylint issues & bug KIKIMR-13478

ref:8856573
sync: https://proxy.sandbox.yandex-team.ru/2575333352

Author: gridnevvvit

examples/access-token-credentials/main.py

@@ -9,9 +9,9 @@ def execute_query(session):
     # statement. The either way to commit transaction is using `commit` method of `TxContext` object, which is
     # not recommended.
     return session.transaction().execute(
-        'select 1 as cnt;',
+        "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),
     )
 
 
@@ -22,9 +22,11 @@ def main():
     # because the specified access token can expire. For example,
     # Yandex.Cloud IAM token lifetime is limited by 12 hours.
     driver = ydb.Driver(
-        endpoint=os.getenv('YDB_ENDPOINT'),
-        database=os.getenv('YDB_DATABASE'),
-        credentials=ydb.AccessTokenCredentials(os.getenv("YDB_ACCESS_TOKEN_CREDENTIALS")),
+        endpoint=os.getenv("YDB_ENDPOINT"),
+        database=os.getenv("YDB_DATABASE"),
+        credentials=ydb.AccessTokenCredentials(
+            os.getenv("YDB_ACCESS_TOKEN_CREDENTIALS")
+        ),
     )
 
     # Start driver context manager.

examples/aio/driver_example.py

@@ -4,11 +4,10 @@
 
 
 async def describe_database():
-    endpoint = os.getenv('YDB_ENDPOINT')
-    database = os.getenv('YDB_DATABASE')
+    endpoint = os.getenv("YDB_ENDPOINT")
+    database = os.getenv("YDB_DATABASE")
     driver = ydb.aio.Driver(
-        endpoint=endpoint,
-        database=database
+        endpoint=endpoint, database=database
     )  # Creating new database driver to execute queries
 
     await driver.wait(timeout=10)  # Wait until driver can execute calls
@@ -21,5 +20,6 @@ async def describe_database():
 
     await driver.stop()  # Stops driver and close all connections
 
+
 if __name__ == "__main__":
     asyncio.get_event_loop().run_until_complete(describe_database())

examples/aio/example_data.py

@@ -2,7 +2,7 @@
 
 
 class Series(object):
-    __slots__ = ('series_id', 'title', 'release_date', 'series_info')
+    __slots__ = ("series_id", "title", "release_date", "series_info")
 
     def __init__(self, series_id, title, release_date, series_info):
         self.series_id = series_id
@@ -12,7 +12,7 @@ def __init__(self, series_id, title, release_date, series_info):
 
 
 class Season(object):
-    __slots__ = ('series_id', 'season_id', 'title', 'first_aired', 'last_aired')
+    __slots__ = ("series_id", "season_id", "title", "first_aired", "last_aired")
 
     def __init__(self, series_id, season_id, title, first_aired, last_aired):
         self.series_id = series_id
@@ -23,7 +23,7 @@ def __init__(self, series_id, season_id, title, first_aired, last_aired):
 
 
 class Episode(object):
-    __slots__ = ('series_id', 'season_id', 'episode_id', 'title', 'air_date')
+    __slots__ = ("series_id", "season_id", "episode_id", "title", "air_date")
 
     def __init__(self, series_id, season_id, episode_id, title, air_date):
         self.series_id = series_id
@@ -35,12 +35,20 @@ def __init__(self, series_id, season_id, episode_id, title, air_date):
 
 def get_series_data():
     return [
-        Series(1, "IT Crowd", "2006-02-03",
-               "The IT Crowd is a British sitcom produced by Channel 4, written by Graham Linehan, produced by "
-               "Ash Atalla and starring Chris O'Dowd, Richard Ayoade, Katherine Parkinson, and Matt Berry."),
-        Series(2, "Silicon Valley", "2014-04-06",
-               "Silicon Valley is an American comedy television series created by Mike Judge, John Altschuler and "
-               "Dave Krinsky. The series focuses on five young men who founded a startup company in Silicon Valley.")
+        Series(
+            1,
+            "IT Crowd",
+            "2006-02-03",
+            "The IT Crowd is a British sitcom produced by Channel 4, written by Graham Linehan, produced by "
+            "Ash Atalla and starring Chris O'Dowd, Richard Ayoade, Katherine Parkinson, and Matt Berry.",
+        ),
+        Series(
+            2,
+            "Silicon Valley",
+            "2014-04-06",
+            "Silicon Valley is an American comedy television series created by Mike Judge, John Altschuler and "
+            "Dave Krinsky. The series focuses on five young men who founded a startup company in Silicon Valley.",
+        ),
     ]
 
 
@@ -54,7 +62,7 @@ def get_seasons_data():
         Season(2, 2, "Season 2", "2015-04-12", "2015-06-14"),
         Season(2, 3, "Season 3", "2016-04-24", "2016-06-26"),
         Season(2, 4, "Season 4", "2017-04-23", "2017-06-25"),
-        Season(2, 5, "Season 5", "2018-03-25", "2018-05-13")
+        Season(2, 5, "Season 5", "2018-03-25", "2018-05-13"),
     ]
 
 

examples/aio/read_table_examples.py

@@ -6,33 +6,30 @@
 
 async def create_tables():
     from session_pool_example import main
+
     await main()
 
 
 async def insert_data():
     from transaction_example import main
+
     await main()
 
 
 async def main():
     await create_tables()
     await insert_data()
-    endpoint = os.getenv('YDB_ENDPOINT')
-    database = os.getenv('YDB_DATABASE')
-    driver = ydb.aio.Driver(
-        endpoint=endpoint,
-        database=database
-    )
-    pool = ydb.aio.SessionPool(
-        driver,
-        size=10  # Max number of available session
-    )
+    endpoint = os.getenv("YDB_ENDPOINT")
+    database = os.getenv("YDB_DATABASE")
+    driver = ydb.aio.Driver(endpoint=endpoint, database=database)
+    pool = ydb.aio.SessionPool(driver, size=10)  # Max number of available session
     session = await pool.acquire()
     async for sets in await session.read_table(database + "/episodes"):
         print(*[row[::] for row in sets.rows], sep="\n")
     await pool.release(session)
     await pool.stop()
     await driver.stop()
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     asyncio.get_event_loop().run_until_complete(main())

examples/aio/session_pool_example.py

@@ -34,7 +34,7 @@
         `air_date` Date,
         PRIMARY KEY (`series_id`, `season_id`, `episode_id`)
     )
-    """
+    """,
 ]
 
 
@@ -47,20 +47,21 @@ async def create_table(session, query):
 
 
 async def main():
-    endpoint = os.getenv('YDB_ENDPOINT')
-    database = os.getenv('YDB_DATABASE')
+    endpoint = os.getenv("YDB_ENDPOINT")
+    database = os.getenv("YDB_DATABASE")
     async with ydb.aio.Driver(endpoint=endpoint, database=database) as driver:
         await driver.wait(fail_fast=True)
 
         async with ydb.aio.SessionPool(driver, size=10) as pool:
             coros = [  # Generating coroutines to create tables concurrently
-                pool.retry_operation(create_table, query)
-                for query in queries
+                pool.retry_operation(create_table, query) for query in queries
             ]
 
             await asyncio.gather(*coros)  # Run table creating concurrently
 
-            directory = await driver.scheme_client.list_directory(database)  # Listing database to ensure that tables are created
+            directory = await driver.scheme_client.list_directory(
+                database
+            )  # Listing database to ensure that tables are created
             print("Items in database:")
             for child in directory.children:
                 print(child.type, child.name)

examples/aio/transaction_example.py

@@ -58,34 +58,31 @@
 
 async def create_tables():  # Creating tables with session_pool_example
     from session_pool_example import main
+
     await main()
 
 
 async def main():
     await create_tables()
-    endpoint = os.getenv('YDB_ENDPOINT')
-    database = os.getenv('YDB_DATABASE')
-    driver = ydb.aio.Driver(
-        endpoint=endpoint,
-        database=database
-    )
-    pool = ydb.aio.SessionPool(
-        driver,
-        size=10  # Max number of available session
-    )
+    endpoint = os.getenv("YDB_ENDPOINT")
+    database = os.getenv("YDB_DATABASE")
+    driver = ydb.aio.Driver(endpoint=endpoint, database=database)
+    pool = ydb.aio.SessionPool(driver, size=10)  # Max number of available session
     session = await pool.acquire()
     prepared_query = await session.prepare(FillDataQuery.format(database))
     await session.transaction(ydb.SerializableReadWrite()).execute(
-        prepared_query, {
-            '$seriesData': example_data.get_series_data(),
-            '$seasonsData': example_data.get_seasons_data(),
-            '$episodesData': example_data.get_episodes_data(),
+        prepared_query,
+        {
+            "$seriesData": example_data.get_series_data(),
+            "$seasonsData": example_data.get_seasons_data(),
+            "$episodesData": example_data.get_episodes_data(),
         },
-        commit_tx=True
+        commit_tx=True,
     )
     await pool.release(session)
     await driver.stop()
     await pool.stop()
 
+
 if __name__ == "__main__":
     asyncio.get_event_loop().run_until_complete(main())

examples/anonymous-credentials/main.py

@@ -4,17 +4,17 @@
 
 def execute_query(session):
     return session.transaction().execute(
-        'select 1 as cnt;',
+        "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),
     )
 
 
 def main():
     driver = ydb.Driver(
-        endpoint=os.getenv('YDB_ENDPOINT'),
-        database=os.getenv('YDB_DATABASE'),
-        credentials=ydb.AnonymousCredentials()
+        endpoint=os.getenv("YDB_ENDPOINT"),
+        database=os.getenv("YDB_DATABASE"),
+        credentials=ydb.AnonymousCredentials(),
     )
 
     with driver:

examples/asyncio/main.py

@@ -10,23 +10,31 @@ 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;',
+                "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

@@ -4,19 +4,22 @@
 import logging
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         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")
+        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("-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')
+    parser.add_argument("-p", "--path", default="")
+    parser.add_argument("-v", "--verbose", default=False, action="store_true")
 
     args = parser.parse_args()
 
     if args.verbose:
-        logger = logging.getLogger('kikimr.public.sdk.python.client.pool.Discovery')
+        logger = logging.getLogger("kikimr.public.sdk.python.client.pool.Discovery")
         logger.setLevel(logging.INFO)
         logger.addHandler(logging.StreamHandler())