blackify repo (#10)

Author: techgerm

setup.py

@@ -14,18 +14,18 @@
 # limitations under the License.
 """spanner_orm setup file."""
 from setuptools import setup
+
 setup(
-    name='spanner-orm',
-    version='0.2.8',
-    description='Basic ORM for Spanner',
-    maintainer='Aniruddha Maru',
-    maintainer_email='[email protected]',
-    url='https://github.com/maroux/python-spanner-orm',
-    packages=['spanner_orm', 'spanner_orm.admin'],
+    name="spanner-orm",
+    version="0.2.8",
+    description="Basic ORM for Spanner",
+    maintainer="Aniruddha Maru",
+    maintainer_email="[email protected]",
+    url="https://github.com/maroux/python-spanner-orm",
+    packages=["spanner_orm", "spanner_orm.admin"],
     include_package_data=True,
-    python_requires='~=3.7',
-    install_requires=['google-cloud-spanner >= 1.6, <2.0.0dev'],
-    tests_require=['absl-py'],
-    entry_points={
-        'console_scripts': ['spanner-orm = spanner_orm.admin.scripts:main']
-    })
+    python_requires="~=3.7",
+    install_requires=["google-cloud-spanner >= 1.6, <2.0.0dev"],
+    tests_require=["absl-py"],
+    entry_points={"console_scripts": ["spanner-orm = spanner_orm.admin.scripts:main"]},
+)

spanner_orm/__init__.py

@@ -106,6 +106,6 @@
 MigrationExecutor = migration_executor.MigrationExecutor
 
 try:
-  __import__('pkg_resources').declare_namespace('spanner_orm')
+    __import__("pkg_resources").declare_namespace("spanner_orm")
 except ImportError:
-  __path__ = __import__('pkgutil').extend_path(__path__, 'spanner_orm')
+    __path__ = __import__("pkgutil").extend_path(__path__, "spanner_orm")

spanner_orm/admin/api.py

@@ -24,59 +24,62 @@
 
 
 class SpannerAdminApi(api.SpannerReadApi, api.SpannerWriteApi):
-  """Manages table schema information on Spanner."""
+    """Manages table schema information on Spanner."""
 
-  def __init__(self, connection: api.SpannerConnection):
-    self._spanner_connection = connection
+    def __init__(self, connection: api.SpannerConnection):
+        self._spanner_connection = connection
 
-  @property
-  def _connection(self) -> spanner_database.Database:
-    return self._spanner_connection.database
+    @property
+    def _connection(self) -> spanner_database.Database:
+        return self._spanner_connection.database
 
-  def create_database(self) -> None:
-    operation = self._connection.create()
-    operation.result()
+    def create_database(self) -> None:
+        operation = self._connection.create()
+        operation.result()
 
-  def drop_database(self) -> None:
-    self._connection.drop()
+    def drop_database(self) -> None:
+        self._connection.drop()
 
-  def update_schema(self, change: str) -> None:
-    operation = self._connection.update_ddl([change])
-    operation.result()
+    def update_schema(self, change: str) -> None:
+        operation = self._connection.update_ddl([change])
+        operation.result()
 
 
 _admin_api = None
 
 
-def connect(instance: str,
-            database: str,
-            project: Optional[str] = None,
-            credentials: Optional[auth_credentials.Credentials] = None,
-            pool: Optional[spanner_pool.AbstractSessionPool] = None,
-            create_ddl: Optional[Iterable[str]] = None) -> SpannerAdminApi:
-  """Connects the global Spanner admin API to a Spanner database."""
-  connection = api.SpannerConnection(
-      instance,
-      database,
-      project=project,
-      credentials=credentials,
-      pool=pool,
-      create_ddl=create_ddl)
-  return from_connection(connection)
+def connect(
+    instance: str,
+    database: str,
+    project: Optional[str] = None,
+    credentials: Optional[auth_credentials.Credentials] = None,
+    pool: Optional[spanner_pool.AbstractSessionPool] = None,
+    create_ddl: Optional[Iterable[str]] = None,
+) -> SpannerAdminApi:
+    """Connects the global Spanner admin API to a Spanner database."""
+    connection = api.SpannerConnection(
+        instance,
+        database,
+        project=project,
+        credentials=credentials,
+        pool=pool,
+        create_ddl=create_ddl,
+    )
+    return from_connection(connection)
 
 
 def from_connection(connection: api.SpannerConnection) -> SpannerAdminApi:
-  global _admin_api
-  _admin_api = SpannerAdminApi(connection)
-  return _admin_api
+    global _admin_api
+    _admin_api = SpannerAdminApi(connection)
+    return _admin_api
 
 
 def hangup() -> None:
-  global _admin_api
-  _admin_api = None
+    global _admin_api
+    _admin_api = None
 
 
 def spanner_admin_api() -> SpannerAdminApi:
-  if not _admin_api:
-    raise error.SpannerError('Must connect to Spanner before calling APIs')
-  return _admin_api
+    if not _admin_api:
+        raise error.SpannerError("Must connect to Spanner before calling APIs")
+    return _admin_api

spanner_orm/admin/column.py

@@ -22,26 +22,27 @@
 
 
 class ColumnSchema(schema.InformationSchema):
-  """Model for interacting with Spanner column schema table."""
-
-  __table__ = 'information_schema.columns'
-  table_catalog = field.Field(field.String, primary_key=True)
-  table_schema = field.Field(field.String, primary_key=True)
-  table_name = field.Field(field.String, primary_key=True)
-  column_name = field.Field(field.String, primary_key=True)
-  ordinal_position = field.Field(field.Integer)
-  is_nullable = field.Field(field.String)
-  spanner_type = field.Field(field.String)
-
-  @property
-  def nullable(self) -> bool:
-    return self.is_nullable == 'YES'
-
-  @property
-  def field_type(self) -> Type[field.FieldType]:
-    for field_type in field.ALL_TYPES:
-      if self.spanner_type == field_type.ddl():
-        return field_type
-
-    raise error.SpannerError('No corresponding Type for {}'.format(
-        self.spanner_type))
+    """Model for interacting with Spanner column schema table."""
+
+    __table__ = "information_schema.columns"
+    table_catalog = field.Field(field.String, primary_key=True)
+    table_schema = field.Field(field.String, primary_key=True)
+    table_name = field.Field(field.String, primary_key=True)
+    column_name = field.Field(field.String, primary_key=True)
+    ordinal_position = field.Field(field.Integer)
+    is_nullable = field.Field(field.String)
+    spanner_type = field.Field(field.String)
+
+    @property
+    def nullable(self) -> bool:
+        return self.is_nullable == "YES"
+
+    @property
+    def field_type(self) -> Type[field.FieldType]:
+        for field_type in field.ALL_TYPES:
+            if self.spanner_type == field_type.ddl():
+                return field_type
+
+        raise error.SpannerError(
+            "No corresponding Type for {}".format(self.spanner_type)
+        )

spanner_orm/admin/index.py

@@ -19,15 +19,15 @@
 
 
 class IndexSchema(schema.InformationSchema):
-  """Model for interacting with Spanner index schema table."""
+    """Model for interacting with Spanner index schema table."""
 
-  __table__ = 'information_schema.indexes'
-  table_catalog = field.Field(field.String, primary_key=True)
-  table_schema = field.Field(field.String, primary_key=True)
-  table_name = field.Field(field.String, primary_key=True)
-  index_name = field.Field(field.String, primary_key=True)
-  index_type = field.Field(field.String)
-  parent_table_name = field.Field(field.String, nullable=True)
-  is_unique = field.Field(field.Boolean)
-  is_null_filtered = field.Field(field.Boolean)
-  index_state = field.Field(field.String)
+    __table__ = "information_schema.indexes"
+    table_catalog = field.Field(field.String, primary_key=True)
+    table_schema = field.Field(field.String, primary_key=True)
+    table_name = field.Field(field.String, primary_key=True)
+    index_name = field.Field(field.String, primary_key=True)
+    index_type = field.Field(field.String)
+    parent_table_name = field.Field(field.String, nullable=True)
+    is_unique = field.Field(field.Boolean)
+    is_null_filtered = field.Field(field.Boolean)
+    index_state = field.Field(field.String)

spanner_orm/admin/index_column.py

@@ -19,15 +19,15 @@
 
 
 class IndexColumnSchema(schema.InformationSchema):
-  """Model for interacting with Spanner index column schema table."""
+    """Model for interacting with Spanner index column schema table."""
 
-  __table__ = 'information_schema.index_columns'
-  table_catalog = field.Field(field.String, primary_key=True)
-  table_schema = field.Field(field.String, primary_key=True)
-  table_name = field.Field(field.String, primary_key=True)
-  index_name = field.Field(field.String, primary_key=True)
-  column_name = field.Field(field.String, primary_key=True)
-  ordinal_position = field.Field(field.Integer, nullable=True)
-  column_ordering = field.Field(field.String, nullable=True)
-  is_nullable = field.Field(field.String)
-  spanner_type = field.Field(field.String)
+    __table__ = "information_schema.index_columns"
+    table_catalog = field.Field(field.String, primary_key=True)
+    table_schema = field.Field(field.String, primary_key=True)
+    table_name = field.Field(field.String, primary_key=True)
+    index_name = field.Field(field.String, primary_key=True)
+    column_name = field.Field(field.String, primary_key=True)
+    ordinal_position = field.Field(field.Integer, nullable=True)
+    column_ordering = field.Field(field.String, nullable=True)
+    is_nullable = field.Field(field.String)
+    spanner_type = field.Field(field.String)