fix import

src/Generator/Python.php

@@ -149,7 +149,13 @@ private function getImports(DefinitionTypeAbstract $origin): array
 
         $imports[] = 'from pydantic import BaseModel, Field, GetCoreSchemaHandler';
         $imports[] = 'from pydantic_core import CoreSchema, core_schema';
-        $imports[] = 'from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict';
+        $imports[] = 'from typing import Any, Dict, Generic, List, Optional, TypeVar';
+
+        if ($origin instanceof MapDefinitionType) {
+            $imports[] = 'from collections import UserDict';
+        } elseif ($origin instanceof ArrayDefinitionType) {
+            $imports[] = 'from collections import UserList';
+        }
 
         if (TypeUtil::contains($origin, StringPropertyType::class, Format::DATE)) {
             $imports[] = 'import datetime';

tests/Generator/resource/python/complex/any_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .property_type import PropertyType
 
 

tests/Generator/resource/python/complex/argument.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .property_type import PropertyType
 from .string_property_type import StringPropertyType
 from .integer_property_type import IntegerPropertyType

tests/Generator/resource/python/complex/array_definition_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .collection_definition_type import CollectionDefinitionType
 
 

tests/Generator/resource/python/complex/array_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .collection_property_type import CollectionPropertyType
 
 

tests/Generator/resource/python/complex/boolean_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .scalar_property_type import ScalarPropertyType
 
 

tests/Generator/resource/python/complex/collection_definition_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .map_definition_type import MapDefinitionType
 from .array_definition_type import ArrayDefinitionType
 from .definition_type import DefinitionType

tests/Generator/resource/python/complex/collection_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .property_type import PropertyType
 from .string_property_type import StringPropertyType
 from .integer_property_type import IntegerPropertyType

tests/Generator/resource/python/complex/definition_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .struct_definition_type import StructDefinitionType
 from .map_definition_type import MapDefinitionType
 from .array_definition_type import ArrayDefinitionType

tests/Generator/resource/python/complex/generic_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .property_type import PropertyType
 
 

tests/Generator/resource/python/complex/integer_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .scalar_property_type import ScalarPropertyType
 
 

tests/Generator/resource/python/complex/map_definition_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .collection_definition_type import CollectionDefinitionType
 
 

tests/Generator/resource/python/complex/map_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .collection_property_type import CollectionPropertyType
 
 

tests/Generator/resource/python/complex/number_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .scalar_property_type import ScalarPropertyType
 
 

tests/Generator/resource/python/complex/operation.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .response import Response
 from .argument import Argument
 

tests/Generator/resource/python/complex/property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .string_property_type import StringPropertyType
 from .integer_property_type import IntegerPropertyType
 from .number_property_type import NumberPropertyType

tests/Generator/resource/python/complex/reference_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .property_type import PropertyType
 
 

tests/Generator/resource/python/complex/response.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .property_type import PropertyType
 from .string_property_type import StringPropertyType
 from .integer_property_type import IntegerPropertyType

tests/Generator/resource/python/complex/scalar_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .string_property_type import StringPropertyType
 from .integer_property_type import IntegerPropertyType
 from .number_property_type import NumberPropertyType

tests/Generator/resource/python/complex/security.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .security_http_basic import SecurityHttpBasic
 from .security_http_bearer import SecurityHttpBearer
 from .security_api_key import SecurityApiKey

tests/Generator/resource/python/complex/security_api_key.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .security import Security
 
 

tests/Generator/resource/python/complex/security_http_basic.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .security import Security
 
 

tests/Generator/resource/python/complex/security_http_bearer.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .security import Security
 
 

tests/Generator/resource/python/complex/security_o_auth.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .security import Security
 
 

tests/Generator/resource/python/complex/string_property_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .scalar_property_type import ScalarPropertyType
 
 

tests/Generator/resource/python/complex/struct_definition_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .definition_type import DefinitionType
 from .reference_property_type import ReferencePropertyType
 from .property_type import PropertyType

tests/Generator/resource/python/complex/type_api.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .type_schema import TypeSchema
 from .security import Security
 from .operation import Operation

tests/Generator/resource/python/complex/type_schema.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .definition_type import DefinitionType
 
 

tests/Generator/resource/python/default/author.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .location import Location
 
 

tests/Generator/resource/python/default/location.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 
 
 # Location of the person

tests/Generator/resource/python/default/meta.py

@@ -1,6 +1,7 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
+from collections import UserDict
 
 
 class Meta(UserDict[str, str]):

tests/Generator/resource/python/default/news.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 import datetime
 from .meta import Meta
 from .author import Author

tests/Generator/resource/python/import/import.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .student_map import StudentMap
 from .student import Student
 

tests/Generator/resource/python/import/my_map.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .student import Student
 
 

tests/Generator/resource/python/namespace/import.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .student_map import StudentMap
 from .student import Student
 

tests/Generator/resource/python/namespace/my_map.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .student import Student
 
 

tests/Generator/resource/python/oop/human_map.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .map import Map
 from .human_type import HumanType
 

tests/Generator/resource/python/oop/human_type.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .human_type import HumanType
 
 

tests/Generator/resource/python/oop/map.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 
 
 P = TypeVar("P")

tests/Generator/resource/python/oop/root_schema.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .student_map import StudentMap
 
 

tests/Generator/resource/python/oop/student.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .human_type import HumanType
 
 

tests/Generator/resource/python/oop/student_map.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
-from typing import Any, Dict, Generic, List, Optional, TypeVar, UserList, UserDict
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 from .map import Map
 from .human_type import HumanType
 from .student import Student