pyupgrade fixes for python 3.14

Author: asottile

pyupgrade/_main.py

@@ -382,6 +382,10 @@ def main(argv: Sequence[str] | None = None) -> int:
         '--py313-plus',
         action='store_const', dest='min_version', const=(3, 13),
     )
+    parser.add_argument(
+        '--py314-plus',
+        action='store_const', dest='min_version', const=(3, 14),
+    )
     args = parser.parse_args(argv)
 
     ret = 0

pyupgrade/_plugins/imports.py

@@ -21,7 +21,7 @@
 from pyupgrade._token_helpers import indented_amount
 
 # GENERATED VIA generate-imports
-# Using reorder-python-imports==3.14.0
+# Using reorder-python-imports==3.15.0
 REMOVALS = {
     (3,): {
         '__future__': {
@@ -38,6 +38,7 @@
         'six.moves': {'filter', 'input', 'map', 'range', 'zip'},
     },
     (3, 7): {'__future__': {'generator_stop'}},
+    (3, 14): {'__future__': {'annotations'}},
 }
 REMOVALS[(3,)]['six.moves.builtins'] = REMOVALS[(3,)]['builtins']
 REPLACE_EXACT = {
@@ -236,6 +237,7 @@
         ('typing_extensions', 'TypeVar'): 'typing',
         ('typing_extensions', 'TypeVarTuple'): 'typing',
         ('typing_extensions', 'TypedDict'): 'typing',
+        ('typing_extensions', 'deprecated'): 'warnings',
         ('typing_extensions', 'get_protocol_members'): 'typing',
         ('typing_extensions', 'is_protocol'): 'typing',
         ('typing_extensions', 'runtime_checkable'): 'typing',

pyupgrade/_plugins/typing_pep563.py

@@ -16,7 +16,10 @@
 
 
 def _supported_version(state: State) -> bool:
-    return 'annotations' in state.from_imports['__future__']
+    return (
+        state.settings.min_version >= (3, 14) or
+        'annotations' in state.from_imports['__future__']
+    )
 
 
 def _dequote(i: int, tokens: list[Token], *, new: str) -> None:

tests/features/typing_pep563_test.py

@@ -372,6 +372,13 @@ def test_fix_typing_pep563(s, expected):
     assert ret == expected
 
 
+def test_fixes_in_py314():
+    src = 'def f(x: "X") -> "Y": pass\n'
+    expected = 'def f(x: X) -> Y: pass\n'
+    ret = _fix_plugins(src, settings=Settings(min_version=(3, 14)))
+    assert ret == expected
+
+
 @pytest.mark.xfail(sys.version_info < (3, 12), reason='3.12+ syntax')
 def test_typevar_bound():
     src = '''\