Fix linter violations surfaced by updated ruff and mypy

psqlpy-stress:
- migrations/versions: add `# ruff: noqa: INP001` (Alembic version files are
  intentionally outside a package); modernize Union[X, None] → X | None
- mocker.py: correct generate_random_array return type list[str] → list[int];
  add # noqa: S311 for non-cryptographic random use; wrap overlong INSERT string
- models/sqlalchemy.py: add type: ignore[misc, valid-type] on declarative_base
  subclasses — Base is typed as Any because sqlalchemy lacks installed stubs
- models/piccolo.py: add type: ignore[call-arg] — tablename kwarg goes to
  Table.__init_subclass__ which mypy can't see through Any
- settings.py: remove `: str` annotations from StrEnum members (mypy 2.x
  requires enum members to be unannotated)

python/tests:
- conftest.py: fix PLW1508 — os.environ.get default must be str, not int
- test_binary_copy.py: extend existing noqa comments with ASYNC240 (os.path
  use inside async functions; the paths are compile-time constants so anyio
  would add no value here)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Dev-iL

psqlpy-stress/psqlpy_stress/migrations/env.py

@@ -1,4 +1,16 @@
 # noqa: INP001
+"""Alembic migration environment.
+
+To enable autogenerate support, point ``target_metadata`` at your models::
+
+    from myapp import mymodel
+    target_metadata = mymodel.Base.metadata
+
+Additional values from alembic.ini can be read via ``config.get_main_option``::
+
+    my_important_option = config.get_main_option("my_important_option")
+"""
+
 from logging.config import fileConfig
 
 from alembic import context
@@ -17,20 +29,11 @@
 if config.config_file_name is not None:
     fileConfig(config.config_file_name)
 
-# add your model's MetaData object here
-# for 'autogenerate' support
-# from myapp import mymodel
-# target_metadata = mymodel.Base.metadata
 target_metadata = Base.metadata
 
 database_url = settings.database_url
 config.set_main_option("sqlalchemy.url", database_url)
 
-# other values from the config, defined by the needs of env.py,
-# can be acquired:
-# my_important_option = config.get_main_option("my_important_option")
-# ... etc.
-
 
 def run_migrations_offline() -> None:
     """Run migrations in 'offline' mode.

psqlpy-stress/psqlpy_stress/migrations/versions/06d989926550_basic_user_table.py

@@ -1,3 +1,4 @@
+# ruff: noqa: INP001
 """basic user table
 
 Revision ID: 06d989926550
@@ -7,17 +8,16 @@
 """
 
 from collections.abc import Sequence
-from typing import Union
 
 import sqlalchemy as sa
 from alembic import op
 
 
 # revision identifiers, used by Alembic.
 revision: str = "06d989926550"
-down_revision: Union[str, None] = None
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = None
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

psqlpy-stress/psqlpy_stress/migrations/versions/d162c084f522_big_af_table.py

@@ -1,3 +1,4 @@
+# ruff: noqa: INP001
 """big af table
 
 Revision ID: d162c084f522
@@ -7,7 +8,6 @@
 """
 
 from collections.abc import Sequence
-from typing import Union
 
 import sqlalchemy as sa
 from alembic import op
@@ -16,9 +16,9 @@
 
 # revision identifiers, used by Alembic.
 revision: str = "d162c084f522"
-down_revision: Union[str, None] = "06d989926550"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "06d989926550"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

psqlpy-stress/psqlpy_stress/mocker.py

@@ -25,12 +25,12 @@ async def fill_users() -> None:
         )
 
 
-def generate_random_array() -> list[str]:
-    return [random.randint(50, 500) for _ in range(random.randint(50, 500))]
+def generate_random_array() -> list[int]:
+    return [random.randint(50, 500) for _ in range(random.randint(50, 500))]  # noqa: S311
 
 
 def generate_random_dict() -> dict[str, str]:
-    return {str(uuid.uuid4()): str(uuid.uuid4()) for _ in range(random.randint(50, 500))}
+    return {str(uuid.uuid4()): str(uuid.uuid4()) for _ in range(random.randint(50, 500))}  # noqa: S311
 
 
 async def fill_big_table() -> None:
@@ -39,10 +39,12 @@ async def fill_big_table() -> None:
     connection = await pool.connection()
     for _ in range(big_table_amount):
         await connection.execute(
-            "INSERT INTO big_table (string_field, integer_field, json_field, array_field) VALUES($1, $2, $3, $4)",
+            "INSERT INTO big_table"
+            " (string_field, integer_field, json_field, array_field)"
+            " VALUES($1, $2, $3, $4)",
             parameters=[
                 str(uuid.uuid4()),
-                random.randint(1, 99999999),
+                random.randint(1, 99999999),  # noqa: S311
                 generate_random_dict(),
                 generate_random_array(),
             ],

psqlpy-stress/psqlpy_stress/models/piccolo.py

@@ -2,12 +2,12 @@
 from piccolo.table import Table
 
 
-class User(Table, tablename="users"):
+class User(Table, tablename="users"):  # type: ignore[call-arg]
     user_id = Serial(primary_key=True)
     username = Varchar(null=False)
 
 
-class SomeBigTable(Table, tablename="big_table"):
+class SomeBigTable(Table, tablename="big_table"):  # type: ignore[call-arg]
     big_table_id = Integer(primary_key=True)
     string_field = Varchar(null=False)
     integer_field = Integer(null=False)

psqlpy-stress/psqlpy_stress/models/sqlalchemy.py

@@ -6,14 +6,14 @@
 Base = declarative_base()
 
 
-class User(Base):
+class User(Base):  # type: ignore[misc, valid-type]
     __tablename__ = "users"
 
     user_id = Column(Integer, primary_key=True, autoincrement=True)
     username = Column(String, unique=True, nullable=False)
 
 
-class SomeBigTable(Base):
+class SomeBigTable(Base):  # type: ignore[misc, valid-type]
     __tablename__ = "big_table"
     big_table_id = Column(Integer, primary_key=True, autoincrement=True)
     string_field = Column(String, unique=False, nullable=False)

psqlpy-stress/psqlpy_stress/settings.py

@@ -4,9 +4,9 @@
 
 
 class DriversEnum(enum.StrEnum):
-    PSQLPY: str = "psqlpy"
-    ASYNCPG: str = "asyncpg"
-    PSYCOPG: str = "psycopg"
+    PSQLPY = "psqlpy"
+    ASYNCPG = "asyncpg"
+    PSYCOPG = "psycopg"
 
 
 class Settings(BaseSettings):

python/tests/conftest.py

@@ -56,7 +56,7 @@ def postgres_password() -> str:
 
 @pytest.fixture
 def postgres_port() -> int:
-    return int(os.environ.get("POSTGRES_PORT", 5432))
+    return int(os.environ.get("POSTGRES_PORT", "5432"))
 
 
 @pytest.fixture

python/tests/test_binary_copy.py

@@ -37,7 +37,7 @@ async def test_binary_copy_to_table_in_connection(
     )
 
     arrow_table = parquet.read_table(
-        f"{os.path.dirname(os.path.abspath(__file__))}/test_data/MTcars.parquet",  # noqa: PTH120, PTH100
+        f"{os.path.dirname(os.path.abspath(__file__))}/test_data/MTcars.parquet",  # noqa: PTH120, PTH100, ASYNC240
     )
     encoder = ArrowToPostgresBinaryEncoder(arrow_table.schema)
     buf = BytesIO()
@@ -90,7 +90,7 @@ async def test_binary_copy_to_table_in_transaction(
     )
 
     arrow_table = parquet.read_table(
-        f"{os.path.dirname(os.path.abspath(__file__))}/test_data/MTcars.parquet",  # noqa: PTH120, PTH100
+        f"{os.path.dirname(os.path.abspath(__file__))}/test_data/MTcars.parquet",  # noqa: PTH120, PTH100, ASYNC240
     )
     encoder = ArrowToPostgresBinaryEncoder(arrow_table.schema)
     buf = BytesIO()