diff --git a/project1/myapp/admin.py b/project1/myapp/admin.py index 932d318..2669e4c 100644 --- a/project1/myapp/admin.py +++ b/project1/myapp/admin.py @@ -1,8 +1,10 @@ from django.contrib import admin -from myapp.models import Person, Drinks, DrinksCategory +from myapp.models import Person, Drinks, DrinksCategory, Customer, Vehicle # Register your models here. admin.site.register(Person) admin.site.register(Drinks) admin.site.register(DrinksCategory) +admin.site.register(Customer) +admin.site.register(Vehicle) diff --git a/project1/myapp/models.py b/project1/myapp/models.py index 31ab678..16e326e 100644 --- a/project1/myapp/models.py +++ b/project1/myapp/models.py @@ -16,3 +16,13 @@ class Drinks(models.Model): price = models.IntegerField() category_id = models.ForeignKey( DrinksCategory, on_delete=models.PROTECT, default=None) + + +class Customer(models.Model): + name = models.CharField(max_length=255) + + +class Vehicle(models.Model): + name = models.CharField(max_length=255) + customer = models.ForeignKey( + Customer, on_delete=models.CASCADE, related_name='Vehicle')