diff --git a/polls/models.py b/polls/models.py index 71a8362..186ddca 100644 --- a/polls/models.py +++ b/polls/models.py @@ -1,3 +1,20 @@ +import datetime + from django.db import models +from django.utils import timezone # Create your models here. +class Question(models.Model): + question_text = models.CharField(max_length=200) + pub_date = models.DateTimeField("date published") + def __str__(self): + return self.question_text + def was_published_recently(self): + return self.pub_date >= timezone.now() - datetime.timedelta(days=1) + +class Choice(models.Model): + question = models.ForeignKey(Question, on_delete=models.CASCADE) + choice_text = models.CharField(max_length=200) + votes = models.IntegerField(default=0) + def __str__(self): + return self.choice_text