Clean up models.

Author: adamghill

example/coffee/models.py

@@ -1,29 +1,21 @@
 import uuid
 
-from django.db.models import SET_NULL, ForeignKey, Model
-from django.db.models.fields import (
-    CharField,
-    DateField,
-    DateTimeField,
-    DecimalField,
-    DurationField,
-    FloatField,
-    TimeField,
-    UUIDField,
-)
+from django.db import models
 
 
-class Flavor(Model):
-    name = CharField(max_length=255)
-    label = CharField(max_length=255)
-    parent = ForeignKey("self", blank=True, null=True, on_delete=SET_NULL)
-    float_value = FloatField(blank=True, null=True)
-    decimal_value = DecimalField(blank=True, null=True, max_digits=10, decimal_places=2)
-    uuid = UUIDField(default=uuid.uuid4)
-    datetime = DateTimeField(blank=True, null=True)
-    date = DateField(blank=True, null=True)
-    time = TimeField(blank=True, null=True)
-    duration = DurationField(blank=True, null=True)
+class Flavor(models.Model):
+    name = models.CharField(max_length=255)
+    label = models.CharField(max_length=255)
+    parent = models.ForeignKey("self", blank=True, null=True, on_delete=models.SET_NULL)
+    float_value = models.FloatField(blank=True, null=True)
+    decimal_value = models.DecimalField(
+        blank=True, null=True, max_digits=10, decimal_places=2
+    )
+    uuid = models.UUIDField(default=uuid.uuid4)
+    datetime = models.DateTimeField(blank=True, null=True)
+    date = models.DateField(blank=True, null=True)
+    time = models.TimeField(blank=True, null=True)
+    duration = models.DurationField(blank=True, null=True)
 
     def __str__(self):
         return self.name

example/unicorn/components/objects.py

@@ -4,10 +4,10 @@
 
 from django.utils.timezone import now
 
-from books.models import Book
 from pydantic import BaseModel
 
 from django_unicorn.components import UnicornField, UnicornView
+from example.books.models import Book
 
 
 class PublishDateField(UnicornField):