more generics

Author: AlexWaygood

crates/red_knot_python_semantic/src/types/class.rs

@@ -1970,6 +1970,15 @@ pub(crate) enum KnownClass {
     SupportsAbs,
     SupportsRound,
     Awaitable,
+    Iterable,
+    Iterator,
+    Generator,
+    Reversible,
+    AsyncIterable,
+    AsyncIterator,
+    AsyncGenerator,
+    Container,
+    Collection,
     // Collections
     ChainMap,
     Counter,
@@ -2044,6 +2053,15 @@ impl<'db> KnownClass {
             | Self::SupportsAbs
             | Self::SupportsRound
             | Self::Awaitable
+            | Self::Iterable
+            | Self::Iterator
+            | Self::Generator
+            | Self::Reversible
+            | Self::AsyncIterable
+            | Self::AsyncIterator
+            | Self::AsyncGenerator
+            | Self::Container
+            | Self::Collection
 
             | Self::Set
             | Self::Tuple
@@ -2124,6 +2142,15 @@ impl<'db> KnownClass {
             Self::SupportsAbs => "SupportsAbs",
             Self::SupportsRound => "SupportsRound",
             Self::Awaitable => "Awaitable",
+            Self::Iterable => "Iterable",
+            Self::Iterator => "Iterator",
+            Self::Generator => "Generator",
+            Self::Reversible => "Reversible",
+            Self::AsyncIterable => "AsyncIterable",
+            Self::AsyncIterator => "AsyncIterator",
+            Self::AsyncGenerator => "AsyncGenerator",
+            Self::Container => "Container",
+            Self::Collection => "Collection",
 
             Self::ChainMap => "ChainMap",
             Self::Counter => "Counter",
@@ -2311,6 +2338,15 @@ impl<'db> KnownClass {
             | Self::SupportsAbs
             | Self::SupportsRound
             | Self::Awaitable
+            | Self::Iterable
+            | Self::Iterator
+            | Self::Generator
+            | Self::Reversible
+            | Self::AsyncIterable
+            | Self::AsyncIterator
+            | Self::AsyncGenerator
+            | Self::Container
+            | Self::Collection
             | Self::Sized => KnownModule::Typing,
             Self::TypeAliasType
             | Self::TypeVarTuple
@@ -2392,7 +2428,6 @@ impl<'db> KnownClass {
             | Self::DefaultDict
             | Self::Deque
             | Self::OrderedDict
-
             | Self::SupportsIndex
             | Self::SupportsInt
             | Self::SupportsFloat
@@ -2403,7 +2438,15 @@ impl<'db> KnownClass {
             | Self::SupportsAbs
             | Self::SupportsRound
             | Self::Awaitable
-
+            | Self::Iterable
+            | Self::Iterator
+            | Self::Generator
+            | Self::Reversible
+            | Self::AsyncIterable
+            | Self::AsyncIterator
+            | Self::AsyncGenerator
+            | Self::Container
+            | Self::Collection
             | Self::StdlibAlias
             | Self::TypeVar
             | Self::ParamSpec
@@ -2470,6 +2513,15 @@ impl<'db> KnownClass {
             | Self::SupportsAbs
             | Self::SupportsRound
             | Self::Awaitable
+            | Self::Iterable
+            | Self::Iterator
+            | Self::Generator
+            | Self::Reversible
+            | Self::AsyncIterable
+            | Self::AsyncIterator
+            | Self::AsyncGenerator
+            | Self::Container
+            | Self::Collection
             | Self::BaseException
             | Self::BaseExceptionGroup
             | Self::Classmethod
@@ -2549,6 +2601,15 @@ impl<'db> KnownClass {
             "SupportsAbs" => Self::SupportsAbs,
             "SupportsRound" => Self::SupportsRound,
             "Awaitable" => Self::Awaitable,
+            "Iterable" => Self::Iterable,
+            "Iterator" => Self::Iterator,
+            "Generator" => Self::Generator,
+            "Reversible" => Self::Reversible,
+            "AsyncIterable" => Self::AsyncIterable,
+            "AsyncIterator" => Self::AsyncIterator,
+            "AsyncGenerator" => Self::AsyncGenerator,
+            "Container" => Self::Container,
+            "Collection" => Self::Collection,
 
             "Enum" => Self::Enum,
             "ABCMeta" => Self::ABCMeta,
@@ -2628,6 +2689,15 @@ impl<'db> KnownClass {
             | Self::SupportsAbs
             | Self::SupportsRound
             | Self::Awaitable
+            | Self::Iterable
+            | Self::Iterator
+            | Self::Generator
+            | Self::Reversible
+            | Self::AsyncIterable
+            | Self::AsyncIterator
+            | Self::AsyncGenerator
+            | Self::Container
+            | Self::Collection
 
             | Self::ParamSpec
             | Self::ParamSpecArgs
@@ -2648,6 +2718,15 @@ impl<'db> KnownClass {
             | Self::SupportsAbs
             | Self::SupportsRound
             | Self::Awaitable
+            | Self::Iterable
+            | Self::Iterator
+            | Self::Generator
+            | Self::Reversible
+            | Self::AsyncIterable
+            | Self::AsyncIterator
+            | Self::AsyncGenerator
+            | Self::Container
+            | Self::Collection
             | Self::Sized => true,
 
             Self::Any