Fix type representation of DateTimeField.__set__ types (#67)

Author: Viicos

src/django_autotyping/stubbing/codemods/_model_creation.py

@@ -69,7 +69,7 @@ class FieldType(TypedDict):
     # For datetime related fields, we use `datetime.x` because `datetime`
     # is already imported in `db/models/manager.pyi`:
     DateTimeField: {
-        "type": "str | datetime.datetime | datetime.Date | Combinable",
+        "type": "str | datetime.datetime | datetime.date | Combinable",
         "extra_imports": [ImportItem("datetime")],
     },
     DateField: {

tests/stubbing/testfiles/djas002_003.py

@@ -1,6 +1,5 @@
 # Test with `MODEL_FIELDS_OPTIONAL=True` (the default).
 
-
 from datetime import date, datetime, time
 from decimal import Decimal
 from uuid import UUID
@@ -11,6 +10,7 @@
     AltNameModel,
     CharFieldsModel,
     DateFieldsModel,
+    DateTimeFieldsModel,
     ForeignKeyModel,
     ModelOne,
 )
@@ -31,6 +31,12 @@
 DateFieldsModel(date_field=date.today())
 DateFieldsModel.objects.create(date_field=date.today())
 
+DateTimeFieldsModel(datetime_field="")
+DateTimeFieldsModel.objects.create(datetime_field="")
+DateTimeFieldsModel(datetime_field=datetime.now())
+DateTimeFieldsModel.objects.create(datetime_field=datetime.now())
+DateTimeFieldsModel(datetime_field=date.today())
+DateTimeFieldsModel.objects.create(datetime_field=date.today())
 
 AllFieldsModel(integer_field=1)
 AllFieldsModel.objects.create(integer_field=1)

tests/stubstestproj/firstapp/models.py

@@ -32,20 +32,22 @@ class DuplicateModel(models.Model):
 
 
 class CharFieldsModel(models.Model):
-
     char_field = models.CharField()
     char_field_blank = models.CharField(blank=True)
     char_field_null = models.CharField(null=True)
     char_field_default = models.CharField(default="")
 
 class DateFieldsModel(models.Model):
-
     date_field = models.DateField()
     date_field_auto_now = models.DateField(auto_now=True)
     date_field_auto_now_add = models.DateField(auto_now_add=True)
 
-class AllFieldsModel(models.Model):
+class DateTimeFieldsModel(models.Model):
+    datetime_field = models.DateTimeField()
+    datetime_field_auto_now = models.DateTimeField(auto_now=True)
+    datetime_field_auto_now_add = models.DateTimeField(auto_now_add=True)
 
+class AllFieldsModel(models.Model):
     integer_field = models.IntegerField()
     float_field = models.FloatField()
     decimal_field = models.DecimalField()
@@ -57,7 +59,6 @@ class AllFieldsModel(models.Model):
     uuid_field = models.UUIDField()
 
 class ForeignKeyModel(models.Model):
-
     model_one = models.ForeignKey("firstapp.ModelOne", on_delete=models.CASCADE)
     model_one_null = models.ForeignKey("firstapp.ModelOne", on_delete=models.CASCADE, null=True)