chore: upgrade black (#19410)

superset/databases/api.py

@@ -881,7 +881,10 @@ class DatabaseRestApi(BaseSupersetModelRestApi):
         database = DatabaseDAO.find_by_id(pk)
         if not database:
             return self.response_404()
-        return self.response(200, function_names=database.function_names,)
+        return self.response(
+            200,
+            function_names=database.function_names,
+        )
 
     @expose("/available/", methods=["GET"])
     @protect()

superset/databases/commands/exceptions.py

@@ -47,7 +47,8 @@ class DatabaseExistsValidationError(ValidationError):
 class DatabaseRequiredFieldValidationError(ValidationError):
     def __init__(self, field_name: str) -> None:
         super().__init__(
-            [_("Field is required")], field_name=field_name,
+            [_("Field is required")],
+            field_name=field_name,
         )
 
 
@@ -100,7 +101,8 @@ class DatabaseUpdateFailedError(UpdateFailedError):
 
 
 class DatabaseConnectionFailedError(  # pylint: disable=too-many-ancestors
-    DatabaseCreateFailedError, DatabaseUpdateFailedError,
+    DatabaseCreateFailedError,
+    DatabaseUpdateFailedError,
 ):
     message = _("Connection failed, please check your connection settings")
 

superset/databases/commands/validate.py

@@ -57,7 +57,8 @@ class ValidateDatabaseParametersCommand(BaseCommand):
             raise InvalidEngineError(
                 SupersetError(
                     message=__(
-                        'Engine "%(engine)s" is not a valid engine.', engine=engine,
+                        'Engine "%(engine)s" is not a valid engine.',
+                        engine=engine,
                     ),
                     error_type=SupersetErrorType.GENERIC_DB_ENGINE_ERROR,
                     level=ErrorLevel.ERROR,
@@ -101,7 +102,8 @@ class ValidateDatabaseParametersCommand(BaseCommand):
 
         # try to connect
         sqlalchemy_uri = engine_spec.build_sqlalchemy_uri(  # type: ignore
-            self._properties.get("parameters"), encrypted_extra,
+            self._properties.get("parameters"),
+            encrypted_extra,
         )
         if self._model and sqlalchemy_uri == self._model.safe_sqlalchemy_uri():
             sqlalchemy_uri = self._model.sqlalchemy_uri_decrypted

superset/databases/dao.py

@@ -42,7 +42,8 @@ class DatabaseDAO(BaseDAO):
     @staticmethod
     def validate_update_uniqueness(database_id: int, database_name: str) -> bool:
         database_query = db.session.query(Database).filter(
-            Database.database_name == database_name, Database.id != database_id,
+            Database.database_name == database_name,
+            Database.id != database_id,
         )
         return not db.session.query(database_query.exists()).scalar()
 

superset/databases/filters.py

@@ -27,7 +27,8 @@ class DatabaseFilter(BaseFilter):
     # TODO(bogdan): consider caching.
 
     def can_access_databases(  # noqa pylint: disable=no-self-use
-        self, view_menu_name: str,
+        self,
+        view_menu_name: str,
     ) -> Set[str]:
         return {
             security_manager.unpack_database_and_schema(vm).database

superset/databases/schemas.py

@@ -308,7 +308,12 @@ def get_engine_spec(engine: Optional[str]) -> Type[BaseEngineSpec]:
     engine_specs = get_engine_specs()
     if engine not in engine_specs:
         raise ValidationError(
-            [_('Engine "%(engine)s" is not a valid engine.', engine=engine,)]
+            [
+                _(
+                    'Engine "%(engine)s" is not a valid engine.',
+                    engine=engine,
+                )
+            ]
         )
     return engine_specs[engine]
 
@@ -324,7 +329,9 @@ class DatabaseValidateParametersSchema(Schema):
         description="DB-specific parameters for configuration",
     )
     database_name = fields.String(
-        description=database_name_description, allow_none=True, validate=Length(1, 250),
+        description=database_name_description,
+        allow_none=True,
+        validate=Length(1, 250),
     )
     impersonate_user = fields.Boolean(description=impersonate_user_description)
     extra = fields.String(description=extra_description, validate=extra_validator)
@@ -351,7 +358,9 @@ class DatabasePostSchema(Schema, DatabaseParametersSchemaMixin):
         unknown = EXCLUDE
 
     database_name = fields.String(
-        description=database_name_description, required=True, validate=Length(1, 250),
+        description=database_name_description,
+        required=True,
+        validate=Length(1, 250),
     )
     cache_timeout = fields.Integer(
         description=cache_timeout_description, allow_none=True
@@ -395,7 +404,9 @@ class DatabasePutSchema(Schema, DatabaseParametersSchemaMixin):
         unknown = EXCLUDE
 
     database_name = fields.String(
-        description=database_name_description, allow_none=True, validate=Length(1, 250),
+        description=database_name_description,
+        allow_none=True,
+        validate=Length(1, 250),
     )
     cache_timeout = fields.Integer(
         description=cache_timeout_description, allow_none=True
@@ -436,7 +447,9 @@ class DatabasePutSchema(Schema, DatabaseParametersSchemaMixin):
 
 class DatabaseTestConnectionSchema(Schema, DatabaseParametersSchemaMixin):
     database_name = fields.String(
-        description=database_name_description, allow_none=True, validate=Length(1, 250),
+        description=database_name_description,
+        allow_none=True,
+        validate=Length(1, 250),
     )
     impersonate_user = fields.Boolean(description=impersonate_user_description)
     extra = fields.String(description=extra_description, validate=extra_validator)