diff --git a/main/admin.py b/main/admin.py new file mode 100644 index 0000000..a8b7e14 --- /dev/null +++ b/main/admin.py @@ -0,0 +1,7 @@ +from django.contrib import admin +from bbs.main.models import Product + +class ProductAdmin(admin.ModelAdmin): + pass + +admin.site.register(Product, ProductAdmin) diff --git a/main/models.py b/main/models.py index 76dc50c..1fc627e 100755 --- a/main/models.py +++ b/main/models.py @@ -1,22 +1,24 @@ -from django.db import models from django.contrib.auth.models import User +from django.db import models +from django.db.models.signals import post_save +from django.dispatch import receiver class Product(models.Model): def __unicode__(self): return self.name - description = models.CharField(max_length=256, blank=True) name = models.CharField(max_length=64) sale_price_int = models.FloatField() sale_price_ext = models.FloatField() + description = models.CharField(max_length=256, blank=True) volume_liter = models.FloatField(null=True) alcohol_percent = models.FloatField(null=True) image = models.ImageField(upload_to='uploads', blank=True) - created = models.DateTimeField(auto_now_add=True) - updated = models.DateTimeField(auto_now=True) active = models.BooleanField(default=True) customer_support = models.CharField(max_length=32, blank=True) + created = models.DateTimeField(auto_now_add=True) + updated = models.DateTimeField(auto_now=True) class Order(models.Model): user = models.ForeignKey(User) @@ -34,10 +36,16 @@ class OrderLine(models.Model): class Transaction(models.Model): user = models.ForeignKey(User) - created = models.DateTimeField(auto_now_add=True) amount = models.FloatField() + created = models.DateTimeField(auto_now_add=True) class UserProfile(models.Model): user = models.ForeignKey(User, unique=True) - balance = models.FloatField() + balance = models.FloatField(default=0.0) image = models.ImageField(upload_to='uploads', blank=True) + +# Create a new UserProfile object when we create a new User. +@receiver(post_save, sender=User) +def create_user_profile(sender, instance, created, **kwargs): + if created: + UserProfile.objects.create(user=instance) diff --git a/settings-sample.py b/settings-sample.py index db3418d..c881c4b 100755 --- a/settings-sample.py +++ b/settings-sample.py @@ -150,3 +150,5 @@ LOGGING = { } } +# User profile +AUTH_PROFILE_MODULE = 'main.UserProfile'