chore: Setup ruff (#217)

* chore: Replace black and flake8 with Ruff

- Setup ruff, ran ruff scans and format
- Fixed all present issues
- Updated workflow to use ruff instead

* chore: Setup pre-commit with ruff hooks

* chore: Fix failing tests and moved conftest

* chore: modified conftest to aquire token if it is not present

Author: sean-sinclair

.flake8

@@ -1,40 +1,28 @@
-name: linting
+name: Check formatting and linting
 
 on:
   pull_request:
   push: { branches: [main] }
 
 jobs:
-  black:
+  ruff-check:
+    name: Run ruff lint and format checks
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: ["3.9"]
-    steps:
-      - uses: actions/checkout@v4
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v5
-        with:
-          python-version: ${{ matrix.python-version }}
-      - uses: psf/black@stable
-        with:
-          options: "--check --verbose --line-length 79"
-          src: "./src/sumo/wrapper"
-  flake8:
-    runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: ["3.9"]
+
     steps:
       - uses: actions/checkout@v4
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v5
+      
+      - uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          python -m pip install --upgrade pip
-          pip install flake8
-      - name: Analysing the code with flake8
-        run: |
-          flake8 src/sumo/wrapper --config .flake8
+          python-version: '3.11'
+          cache: 'pip'
+      
+      - name: Installing dependencies
+        run: pip install ruff
+                
+      - name: Run ruff lint
+        run: ruff check .
+
+      - name: Run ruff format
+        run: ruff format . --check
+

.github/workflows/linting.yml

@@ -12,4 +12,5 @@ build
 /src/testing.py
 /docs/_build
 /src/sumo/wrapper/version.py
-*_version.py
+*_version.py
+.vscode/

.gitignore

@@ -0,0 +1,16 @@
+repos:
+- repo: local 
+  hooks:
+    - id: lint
+      name: Ruff Lint
+      description: Linting using ruff
+      entry: bash -c 'ruff check .'
+      language: system
+      stages: ["pre-commit", "pre-push"]
+
+    - id: format
+      name: Ruff Format
+      description: Formatting using ruff
+      entry: bash -c 'ruff format . --check'
+      language: system
+      stages: ["pre-commit", "pre-push"]

.pre-commit-config.yaml

@@ -55,4 +55,3 @@
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
 html_static_path = ["_static"]
-

docs/conf.py

@@ -5,10 +5,6 @@ build-backend = "setuptools.build_meta"
 [tool.setuptools_scm]
 version_file = "src/sumo/wrapper/_version.py"
 
-
-[tool.black]
-line-length = 79
-
 [project]
 name = "sumo-wrapper-python"
 description = "Python wrapper for the Sumo API"
@@ -39,6 +35,7 @@ docs = [
   "sphinx-autodoc-typehints",
   "sphinxcontrib-apidoc",
 ]
+dev = ["ruff", "pre-commit"]
 
 [project.urls]
 Repository = "https://github.com/equinor/sumo-wrapper-python"
@@ -48,3 +45,32 @@ sumo_login = "sumo.wrapper.login:main"
 
 [tool.setuptools.packages.find]
 where = ["src"]
+
+[tool.ruff]
+exclude = [
+  ".env",
+  ".git",
+  ".github",
+  ".venv",
+  "venv",
+]
+
+line-length = 79
+
+[tool.ruff.lint]
+ignore = [
+  "E501",
+]
+
+extend-select = [
+  "C4",  # Flake8-comprehensions
+  "I",   # isort
+  "SIM", # Flake8-simplify
+  "TC",  # Flake8-type-checking
+  "TID", # Flake8-tidy-imports
+  "N",   # pep8-naming  
+]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]
+

pyproject.toml

@@ -1,5 +1,5 @@
-from .sumo_client import SumoClient
 from ._retry_strategy import RetryStrategy
+from .sumo_client import SumoClient
 
 try:
     from ._version import version

src/sumo/wrapper/__init__.py

@@ -1,19 +1,19 @@
-import msal
+import errno
+import json
 import os
-from datetime import datetime, timedelta
 import stat
 import sys
-import json
-import jwt
 import time
-from azure.identity import ManagedIdentityCredential
-import tenacity as tn
-from ._retry_strategy import _log_retry_info, _return_last_value
+from datetime import datetime, timedelta
 
+import jwt
+import msal
+import tenacity as tn
+from azure.identity import ManagedIdentityCredential
 from msal_extensions.persistence import FilePersistence
 from msal_extensions.token_cache import PersistedTokenCache
-import errno
 
+from ._retry_strategy import _log_retry_info, _return_last_value
 
 if not sys.platform.startswith("linux"):
     from msal_extensions import build_encrypted_persistence
@@ -422,14 +422,12 @@ def get_auth_provider(
         return AuthProviderDeviceCode(client_id, authority, resource_id)
     # ELSE
     if all(
-        [
-            os.getenv(x)
-            for x in [
-                "AZURE_FEDERATED_TOKEN_FILE",
-                "AZURE_TENANT_ID",
-                "AZURE_CLIENT_ID",
-                "AZURE_AUTHORITY_HOST",
-            ]
+        os.getenv(x)
+        for x in [
+            "AZURE_FEDERATED_TOKEN_FILE",
+            "AZURE_TENANT_ID",
+            "AZURE_CLIENT_ID",
+            "AZURE_AUTHORITY_HOST",
         ]
     ):
         return AuthProviderManaged(resource_id)

src/sumo/wrapper/_auth_provider.py

@@ -3,14 +3,19 @@
 
 
 class LogHandlerSumo(logging.Handler):
-    def __init__(self, sumoClient):
+    def __init__(self, sumo_client):
         logging.Handler.__init__(self)
-        self._sumoClient = sumoClient
+        self._sumoClient = sumo_client
         return
 
     def emit(self, record):
         try:
-            dt = datetime.utcnow().replace(microsecond=0).isoformat() + "Z"
+            dt = (
+                datetime.now(datetime.timezone.utc)
+                .replace(microsecond=0)
+                .isoformat()
+                + "Z"
+            )
             json = {
                 "severity": record.levelname,
                 "message": record.getMessage(),
@@ -20,7 +25,7 @@ def emit(self, record):
                 "funcname": record.funcName,
                 "linenumber": record.lineno,
             }
-            if "objectUuid" in record.__dict__.keys():
+            if "objectUuid" in record.__dict__:
                 json["objectUuid"] = record.__dict__.get("objectUuid")
 
             self._sumoClient.post("/message-log/new", json=json)

src/sumo/wrapper/_logging.py

@@ -1,5 +1,5 @@
-import tenacity as tn
 import httpx
+import tenacity as tn
 
 
 def _log_retry_info(retry_state):