Typing: SQLAlchemy 1.4 má svůj vlastní plugin do MyPy

... chodí zase jinak mizerně než ten původní.

.mypy.ini

 [mypy]
-plugins=sqlmypy
+plugins=sqlalchemy.ext.mypy.plugin
 python_executable=venv/bin/python3

constraints.txt

@@ -38,6 +38,7 @@ requests==2.28.1
 six==1.16.0
 SQLAlchemy==1.4.40
 sqlalchemy-stubs==0.4
+sqlalchemy2-stubs==0.0.2a25
 tomli==2.0.1
 types-bleach==5.0.3
 types-Flask-SQLAlchemy==2.5.9

mo/db.py

@@ -8,7 +8,7 @@ import locale
 import os
 import re
 from sqlalchemy import \
-    Boolean, Column, DateTime, ForeignKey, Integer, String, Text, UniqueConstraint, \
+    Boolean, Column, DateTime, Enum, ForeignKey, Integer, String, Text, UniqueConstraint, \
     text, func, \
     create_engine, inspect, select
 from sqlalchemy.engine import Engine
@@ -20,23 +20,13 @@ from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.sql.expression import CTE
 from sqlalchemy.sql.functions import ReturnTypeFromArgs
 from sqlalchemy.sql.sqltypes import Numeric
-from typing import Optional, List, Tuple
+from typing import Any, Optional, List, Tuple
 
 import mo
 import mo.config as config
 from mo.place_level import place_levels, PlaceLevel
 from mo.util_format import timeformat_short, timedelta, time_and_timedelta
 
-# HACK: Work-around for https://github.com/dropbox/sqlalchemy-stubs/issues/114
-from typing import TYPE_CHECKING, TypeVar, Type, Any
-if TYPE_CHECKING:
-    from sqlalchemy.sql.type_api import TypeEngine
-    T = TypeVar('T')
-    class Enum(TypeEngine[T]):
-        def __init__(self, enum: Type[T], **kwargs: Any) -> None: ...
-else:
-    from sqlalchemy import Enum
-
 
 # Funkce f_unaccent je definovaná v db.ddl, naučme Alchymii volat ji:
 class f_unaccent(ReturnTypeFromArgs):

setup.py

@@ -47,7 +47,7 @@ setuptools.setup(
         'python-poppler',
         'pyzbar',
         'requests',
-        'sqlalchemy',
+        'sqlalchemy[mypy]',
         'uwsgidecorators',
         # Používáme pro vývoj, ale aby je pylsp našel, musí být ve stejném virtualenvu
         # jako ostatní knihovny.