Migrate to ruff

.pre-commit-config.yaml

@@ -6,15 +6,14 @@ repos:
   hooks:
     - id: black
       language_version: python3.11
-- repo: https://github.com/pycqa/isort
-  rev: v5.11.3
+
+- repo: https://github.com/charliermarsh/ruff-pre-commit
+  rev: 'v0.0.213'
   hooks:
-    - id: isort
-- repo: https://github.com/csachs/pyproject-flake8
-  rev: v6.0.0.post1
+    - id: ruff
+      args: ["--force-exclude"]
+
+- repo: https://github.com/crate-ci/typos
+  rev: v1.13.6
   hooks:
-    - id: pyproject-flake8
-      additional_dependencies: [
-        '-e', 'git+https://github.com/pycqa/pyflakes@b37f91a#egg=pyflakes',
-        '-e', 'git+https://github.com/pycqa/pycodestyle@1063db8#egg=pycodestyle',
-      ]
+    - id: typos

pyproject.toml

@@ -5,16 +5,16 @@ description = ""
 authors = ["Kurbanov Bulat <kurbanovbul@gmail.com>"]
 
 [tool.poetry.dependencies]
-python = "^3.9"
-fastapi = "^0.88.0"
-httpx = "^0.23.1"
+python = "^3.11"
+fastapi = "^0.89.0"
+httpx = "^0.23.3"
 alembic = "^1.9.1"
-uvicorn = {extras = ["standart"], version = "^0.20.0"}
+uvicorn = {extras = ["standard"], version = "^0.20.0"}
 arq = "^0.25.0"
 prometheus-fastapi-instrumentator = "^5.9.1"
 uvloop = "^0.17.0"
 gunicorn = "^20.1.0"
-orjson = "^3.8.3"
+orjson = "^3.8.4"
 sentry-sdk = "^1.12.1"
 ormar = {extras = ["postgresql"], version = "^0.12.0"}
 greenlet = "^2.0.1"
@@ -22,7 +22,8 @@ pydantic = "1.10.2"
 redis = {extras = ["hiredis"], version = "^4.4.0"}
 msgpack = "^1.0.4"
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
+pre-commit = "^2.21.0"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
@@ -39,31 +40,38 @@ exclude = '''
 )/
 '''
 
-[tool.flake8]
-ignore = [
-    # Whitespace before ':' ( https://www.flake8rules.com/rules/E203.html )
-    "E203"
-]
-max-line-length=88
-max-complexity = 15
-select = "B,C,E,F,W,T4,B9"
+[tool.ruff]
+fix = true
+target-version = "py311"
+src = ["app"]
+line-length=88
+ignore = []
+select = ["B", "C", "E", "F", "W", "B9", "I001"]
 exclude = [
     # No need to traverse our git directory
     ".git",
     # There's no value in checking cache directories
     "__pycache__",
     # The conf file is mostly autogenerated, ignore it
-    "src/app/alembic/*",
-    # The old directory contains Flake8 2.0
+    "src/app/alembic",
 ]
 
-[tool.isort]
-profile = "black"
-only_sections = true
-force_sort_within_sections = true
-lines_after_imports = 2
-lexicographical = true
-sections = ["FUTURE", "STDLIB", "BASEFRAMEWORK", "FRAMEWORKEXT", "THIRDPARTY", "FIRSTPARTY", "LOCALFOLDER"]
-known_baseframework = ["fastapi",]
-known_frameworkext = ["starlette",]
-src_paths = ["src"]
+[tool.ruff.flake8-bugbear]
+extend-immutable-calls = ["fastapi.File", "fastapi.Form", "fastapi.Security"]
+
+[tool.ruff.mccabe]
+max-complexity = 15
+
+[tool.ruff.isort]
+known-first-party = ["core", "app"]
+
+# only_sections = true
+# force_sort_within_sections = true
+# lines_after_imports = 2
+# lexicographical = true
+# sections = ["FUTURE", "STDLIB", "BASEFRAMEWORK", "FRAMEWORKEXT", "THIRDPARTY", "FIRSTPARTY", "LOCALFOLDER"]
+# known_baseframework = ["fastapi",]
+# known_frameworkext = ["starlette",]
+
+[tool.ruff.pyupgrade]
+keep-runtime-typing = true

scripts/healthcheck.py

@@ -1,6 +1,5 @@
 import httpx
 
-
 response = httpx.get("http://localhost:8080/healthcheck")
 print(f"HEALTHCHECK STATUS: {response.status_code}")
 exit(0 if response.status_code == 200 else 1)

src/app/depends.py

@@ -1,4 +1,4 @@
-from fastapi import Security, HTTPException, status
+from fastapi import HTTPException, Security, status
 
 from core.auth import default_security
 from core.config import env_config

src/app/models.py

@@ -1,6 +1,6 @@
 import ormar
 
-from core.db import metadata, database
+from core.db import database, metadata
 
 
 class BaseMeta(ormar.ModelMeta):

src/app/services/cache_updater.py

@@ -1,15 +1,14 @@
 import collections
-from datetime import timedelta
 import logging
 import random
+from datetime import timedelta
 from tempfile import SpooledTemporaryFile
 from typing import Optional, cast
 
-from fastapi import UploadFile
-
+import httpx
 from arq.connections import ArqRedis
 from arq.worker import Retry
-import httpx
+from fastapi import UploadFile
 from redis import asyncio as aioredis
 from redis.exceptions import LockError
 
@@ -17,8 +16,7 @@ from app.models import CachedFile
 from app.services.caption_getter import get_caption
 from app.services.downloader import download
 from app.services.files_client import upload_file
-from app.services.library_client import get_books, get_book, Book
-
+from app.services.library_client import Book, get_book, get_books
 
 logger = logging.getLogger("telegram_channel_files_manager")
 
@@ -61,7 +59,7 @@ async def check_books(ctx: dict, *args, **kwargs) -> None:  # NOSONAR
     try:
         books_page = await get_books(1, PAGE_SIZE)
     except httpx.ConnectError:
-        raise Retry(defer=15)
+        raise Retry(defer=15)  # noqa: B904
 
     for i, page_number in enumerate(range(books_page.total_pages, 0, -1)):
         await arq_pool.enqueue_job(
@@ -143,7 +141,9 @@ async def cache_file_by_book_id(
                 if by_request:
                     return result
         except LockError:
-            raise Retry(defer=timedelta(minutes=15).seconds * random.random())
+            raise Retry(  # noqa: B904
+                defer=timedelta(minutes=15).seconds * random.random()
+            )
     except Retry as e:
         if by_request:
             return None

src/app/services/library_client.py

@@ -1,12 +1,11 @@
 from datetime import date
-from typing import Generic, TypeVar, Optional
+from typing import Generic, Optional, TypeVar
 
 import httpx
 from pydantic import BaseModel
 
 from core.config import env_config
 
-
 T = TypeVar("T")
 
 

src/app/views.py

@@ -1,12 +1,10 @@
 import asyncio
 import base64
 
-from fastapi import APIRouter, Depends, HTTPException, status, Request
-from fastapi.responses import StreamingResponse
-
-from starlette.background import BackgroundTask
-
 from arq.connections import ArqRedis
+from fastapi import APIRouter, Depends, HTTPException, Request, status
+from fastapi.responses import StreamingResponse
+from starlette.background import BackgroundTask
 
 from app.depends import check_token
 from app.models import CachedFile as CachedFileDB
@@ -17,7 +15,6 @@ from app.services.downloader import get_filename
 from app.services.files_client import download_file as download_file_from_cache
 from app.services.library_client import get_book
 
-
 router = APIRouter(
     prefix="/api/v1", tags=["files"], dependencies=[Depends(check_token)]
 )

src/core/app.py

@@ -1,13 +1,12 @@
 from fastapi import FastAPI
 from fastapi.responses import ORJSONResponse
-
 from prometheus_fastapi_instrumentator import Instrumentator
 
-from app.views import router, healthcheck_router
+import core.sentry  # noqa: F401
+from app.views import healthcheck_router, router
 from core.arq_pool import get_arq_pool
 from core.db import database
 from core.redis_client import get_client
-import core.sentry  # noqa: F401
 
 
 def start_app() -> FastAPI:

src/core/arq_pool.py

@@ -1,5 +1,5 @@
-from arq.connections import create_pool, RedisSettings, ArqRedis
 import msgpack
+from arq.connections import ArqRedis, RedisSettings, create_pool
 
 from core.config import env_config
 

src/core/auth.py

@@ -1,4 +1,3 @@
 from fastapi.security import APIKeyHeader
 
-
 default_security = APIKeyHeader(name="Authorization")

src/core/db.py

@@ -5,7 +5,6 @@ from sqlalchemy import MetaData
 
 from core.config import env_config
 
-
 DATABASE_URL = (
     f"postgresql://{env_config.POSTGRES_USER}:{quote(env_config.POSTGRES_PASSWORD)}@"
     f"{env_config.POSTGRES_HOST}:{env_config.POSTGRES_PORT}/{env_config.POSTGRES_DB}"

src/core/sentry.py

@@ -2,7 +2,6 @@ import sentry_sdk
 
 from core.config import env_config
 
-
 sentry_sdk.init(
     env_config.SENTRY_DSN,
 )

src/core/setup_arq.py

@@ -1,14 +1,14 @@
 import msgpack
 
+import core.sentry  # noqa: F401
 from app.services.cache_updater import (
-    check_books,
     cache_file_by_book_id,
+    check_books,
     check_books_page,
 )
-from core.arq_pool import get_redis_settings, get_arq_pool
+from core.arq_pool import get_arq_pool, get_redis_settings
 from core.db import database
 from core.redis_client import get_client
-import core.sentry  # noqa: F401
 
 
 async def startup(ctx):

src/main.py

@@ -1,4 +1,3 @@
 from core.app import start_app
 
-
 app = start_app()