chore: migrate packages from black to ruff

mypy.ini

@@ -42,6 +42,9 @@ ignore_missing_imports = True
 [mypy-google.oauth2,google.oauth2.*]
 ignore_missing_imports = True
 
+[mypy-google_auth_oauthlib,google_auth_oauthlib.*]
+ignore_missing_imports = True
+
 [mypy-google.protobuf,google.protobuf.*]
 ignore_missing_imports = True
 

packages/bigquery-magics/bigquery_magics/_versions_helpers.py

@@ -16,8 +16,8 @@
 
 from typing import Any
 
-from google.cloud.bigquery import exceptions
 import packaging.version
+from google.cloud.bigquery import exceptions
 
 _MIN_BQ_STORAGE_VERSION = packaging.version.Version("2.0.0")
 

packages/bigquery-magics/bigquery_magics/bigquery.py

@@ -106,33 +106,33 @@
 from __future__ import print_function
 
 import ast
-from concurrent import futures
 import copy
 import json
 import re
 import sys
 import threading
 import time
-from typing import Any, List, Tuple
 import warnings
+from concurrent import futures
+from typing import Any, List, Tuple
 
 import IPython  # type: ignore
-from IPython.core import magic_arguments  # type: ignore
-from IPython.core.getipython import get_ipython
+import pandas
 from google.api_core.exceptions import NotFound
 from google.cloud import bigquery
 from google.cloud.bigquery import exceptions
 from google.cloud.bigquery.dataset import DatasetReference
 from google.cloud.bigquery.dbapi import _helpers
 from google.cloud.bigquery.job import QueryJobConfig
-import pandas
+from IPython.core import magic_arguments  # type: ignore
+from IPython.core.getipython import get_ipython
 
-from bigquery_magics import core
-from bigquery_magics import line_arg_parser as lap
 import bigquery_magics._versions_helpers
 import bigquery_magics.config
 import bigquery_magics.graph_server as graph_server
 import bigquery_magics.pyformat
+from bigquery_magics import core
+from bigquery_magics import line_arg_parser as lap
 
 try:
     from google.cloud import bigquery_storage  # type: ignore
@@ -471,8 +471,9 @@ def _parse_magic_args(line: str) -> Tuple[List[Any], Any]:
 
     except lap.ParseError as exc:
         raise ValueError(
-            "Unrecognized input, are option values correct? "
-            "Error details: {}".format(exc.args[0])
+            "Unrecognized input, are option values correct? Error details: {}".format(
+                exc.args[0]
+            )
         ) from exc
 
     params = []

packages/bigquery-magics/bigquery_magics/config.py

@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import warnings
 from dataclasses import dataclass
 from typing import Optional
-import warnings
 
 import google.api_core.client_options as client_options
 import google.cloud.bigquery as bigquery

packages/bigquery-magics/bigquery_magics/core.py

@@ -18,9 +18,9 @@
 from google.api_core import client_info
 from google.cloud import bigquery
 
-from bigquery_magics import environment
 import bigquery_magics.config
 import bigquery_magics.version
+from bigquery_magics import environment
 
 context = bigquery_magics.config.context
 

packages/bigquery-magics/bigquery_magics/line_arg_parser/lexer.py

@@ -12,10 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from collections import OrderedDict, namedtuple
 import enum
 import itertools
 import re
+from collections import OrderedDict, namedtuple
 
 Token = namedtuple("Token", ("type_", "lexeme", "pos"))
 StateTransition = namedtuple("StateTransition", ("new_state", "total_offset"))

packages/bigquery-magics/noxfile.py

@@ -151,10 +151,24 @@ def lint(session):
     Returns a failure if the linters find linting errors or sufficiently
     serious code quality issues.
     """
-    session.install("setuptools", FLAKE8_VERSION, BLACK_VERSION)
+    session.install("setuptools", FLAKE8_VERSION, RUFF_VERSION)
+    # 1. Check imports
     session.run(
-        "black",
+        "ruff",
+        "check",
+        "--select",
+        "I",
+        f"--target-version=py{UNIT_TEST_PYTHON_VERSIONS[0].replace('.', '')}",
+        "--line-length=88",
+        *LINT_PATHS,
+    )
+    # 2. Check formatting
+    session.run(
+        "ruff",
+        "format",
         "--check",
+        f"--target-version=py{UNIT_TEST_PYTHON_VERSIONS[0].replace('.', '')}",
+        "--line-length=88",
         *LINT_PATHS,
     )
     session.run("flake8", "bigquery_magics", "tests")