udpates linting

Author: chalmerlowe

packages/sqlalchemy-spanner/create_test_config.py

@@ -18,9 +18,16 @@
 import sys
 
 
-def set_test_config(project, instance, database, user=None, password=None, host=None, port=None):
+def set_test_config(
+    project, instance, database, user=None, password=None, host=None, port=None
+):
     config = configparser.ConfigParser()
-    if user is not None and password is not None and host is not None and port is not None:
+    if (
+        user is not None
+        and password is not None
+        and host is not None
+        and port is not None
+    ):
         url = (
             f"spanner+spanner://{user}:{password}@{host}:{port}"
             f"/projects/{project}/instances/{instance}/"

packages/sqlalchemy-spanner/create_test_database.py

@@ -17,13 +17,13 @@
 import os
 import time
 
-from create_test_config import set_test_config
 from google.api_core import datetime_helpers
 from google.api_core.exceptions import AlreadyExists, ResourceExhausted
 from google.cloud.spanner_v1 import Client
-from google.cloud.spanner_v1.instance import Instance
 from google.cloud.spanner_v1.database import Database
+from google.cloud.spanner_v1.instance import Instance
 
+from create_test_config import set_test_config
 
 USE_EMULATOR = os.getenv("SPANNER_EMULATOR_HOST") is not None
 

packages/sqlalchemy-spanner/drop_test_database.py

@@ -19,13 +19,13 @@
 import re
 import time
 
-from create_test_config import set_test_config
 from google.api_core import datetime_helpers
 from google.api_core.exceptions import AlreadyExists, ResourceExhausted
 from google.cloud.spanner_v1 import Client
-from google.cloud.spanner_v1.instance import Instance
 from google.cloud.spanner_v1.database import Database
+from google.cloud.spanner_v1.instance import Instance
 
+from create_test_config import set_test_config
 
 USE_EMULATOR = os.getenv("SPANNER_EMULATOR_HOST") is not None
 
@@ -55,9 +55,9 @@ def delete_test_database():
     instance_id = re.findall(r"instances(.*?)databases", db_url)
     database_id = re.findall(r"databases(.*?)$", db_url)
 
-    instance = CLIENT.instance(
-        instance_id="".join(instance_id).replace("/", ""))
+    instance = CLIENT.instance(instance_id="".join(instance_id).replace("/", ""))
     database = instance.database("".join(database_id).replace("/", ""))
     database.drop()
 
+
 delete_test_database()

packages/sqlalchemy-spanner/migration_test_cleanup.py

@@ -21,18 +21,20 @@
 
 
 def main(argv):
-  db_url = argv[0]
+    db_url = argv[0]
 
-  project = re.findall(r"projects(.*?)instances", db_url)
-  instance_id = re.findall(r"instances(.*?)databases", db_url)
-  database_id = re.findall(r"databases(.*?)$", db_url)
+    project = re.findall(r"projects(.*?)instances", db_url)
+    instance_id = re.findall(r"instances(.*?)databases", db_url)
+    database_id = re.findall(r"databases(.*?)$", db_url)
 
-  client = spanner.Client(project="".join(project).replace("/", ""))
-  instance = client.instance(instance_id="".join(instance_id).replace("/", ""))
-  database = instance.database("".join(database_id).replace("/", ""))
+    client = spanner.Client(project="".join(project).replace("/", ""))
+    instance = client.instance(instance_id="".join(instance_id).replace("/", ""))
+    database = instance.database("".join(database_id).replace("/", ""))
 
-  database.update_ddl(["DROP TABLE IF EXISTS account", "DROP TABLE IF EXISTS alembic_version"]).result(120)
+    database.update_ddl(
+        ["DROP TABLE IF EXISTS account", "DROP TABLE IF EXISTS alembic_version"]
+    ).result(120)
 
 
 if __name__ == "__main__":
-   main(sys.argv[1:])
+    main(sys.argv[1:])

packages/sqlalchemy-spanner/test_migration_env.py

@@ -14,11 +14,9 @@
 
 from logging.config import fileConfig
 
-from sqlalchemy import engine_from_config
-from sqlalchemy import pool
-
 from alembic import context
 from alembic.ddl.impl import DefaultImpl
+from sqlalchemy import engine_from_config, pool
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.