Rewrite async_timeout

Author: mxr

README.md

@@ -310,6 +310,11 @@ some examples:
 +from typing import Concatenate
 ```
 
+```diff
+-from async_timeout import timeout, timeout_at
++from asyncio import timeout, timeout_at
+```
+
 ### rewrite `mock` imports
 
 Availability:

pyupgrade/_plugins/imports.py

@@ -196,6 +196,8 @@
         ('typing_extensions', 'is_typeddict'): 'typing',
     },
     (3, 11): {
+        ('async_timeout', 'timeout'): 'asyncio',
+        ('async_timeout', 'timeout_at'): 'asyncio',
         ('typing_extensions', 'Any'): 'typing',
         ('typing_extensions', 'LiteralString'): 'typing',
         ('typing_extensions', 'Never'): 'typing',

tests/features/import_replaces_test.py

@@ -30,6 +30,11 @@
             (3,),
             id='inline import-import',
         ),
+        pytest.param(
+            'if True: from async_timeout import timeout, other_name\n',
+            (3, 11),
+            id='inline async_timeout mixed imports',
+        ),
         pytest.param(
             'import xml.etree.cElementTree',
             (3,),
@@ -45,6 +50,11 @@
             (3, 9),
             id='skip rewriting of Callable in 3.9 since it is broken',
         ),
+        pytest.param(
+            'from async_timeout import timeout, timeout_at\n',
+            (3, 10),
+            id='async_timeout timeout imports below 3.11',
+        ),
     ),
 )
 def test_import_replaces_noop(s, min_version):
@@ -77,6 +87,24 @@ def test_mock_noop_keep_mock():
             'from collections.abc import Mapping as MAP\n',
             id='one-name replacement with alias',
         ),
+        pytest.param(
+            'from async_timeout import timeout\n',
+            (3, 11),
+            'from asyncio import timeout\n',
+            id='async_timeout timeout',
+        ),
+        pytest.param(
+            'from async_timeout import timeout as t\n',
+            (3, 11),
+            'from asyncio import timeout as t\n',
+            id='async_timeout timeout alias',
+        ),
+        pytest.param(
+            'from async_timeout import timeout, timeout_at\n',
+            (3, 11),
+            'from asyncio import timeout, timeout_at\n',
+            id='async_timeout timeout names',
+        ),
         pytest.param(
             'from collections import Mapping, Sequence\n',
             (3,),
@@ -90,6 +118,13 @@ def test_mock_noop_keep_mock():
             'from collections.abc import Mapping\n',
             id='one name rewritten to new module',
         ),
+        pytest.param(
+            'from async_timeout import timeout, other_name\n',
+            (3, 11),
+            'from async_timeout import other_name\n'
+            'from asyncio import timeout\n',
+            id='async_timeout mixed imports',
+        ),
         pytest.param(
             'from collections import Counter, Mapping',
             (3,),