1、定義數據模型
首先定義投票人 屬于UserProfile（它屬于User）
再次定義投票的文章它屬于Article
article = models.ForeignKey(to=Article, related_name='tickets')
定義投票的類型
VOTE_CHOICES = (
('like', 'like'),
('dislike', 'dislike'),
('normal', 'normal'),
)
choice = models.CharField(choices=VOTE_CHOICES, max_length=10)
總體代碼如下：
class UserProfile(models.Model):
belong_to = models.OneToOneField(to=User, related_name="profile1")
profile_image = models.ImageField()
class Ticket(models.Model):
voter = models.ForeignKey(to=UserProfile, related_name="voted_tickets")
article = models.ForeignKey(to=Article, related_name='tickets')
VOTE_CHOICES = (
('like', 'like'),
('dislike', 'dislike'),
('normal', 'normal'),
)
choice = models.CharField(choices=VOTE_CHOICES, max_length=10)
2、頁面結果的顯示邏輯
在views.py中引入Ticket
from firstapp.models import Article, Comment, Ticket
這就就可以使用Ticket的數據并存儲了
在詳情頁面detail中
def detail(request, id):
context = {}
if request.method == "GET":
form = CommentForm
article = Article.objects.get(id=id)
try:
voter_id = request.user.profile1.id#得到投票用戶的ID
user_ticket_for_this_article = Ticket.objects.get(voter_id=voter_id,article_id=id)# 投票到文章的ID和投票人ID
context["user_ticket"] = user_ticket_for_this_article
except:
pass
context["article"] = article
context['form'] = form
return render(request, 'detail.html', context)
此時user_ticket將可以使用models.py中Ticket的內容
在模板中
<form class="ui form" action="#" method="post">
{% if user_ticket.choice == 'like' %}
<button class="ui red tiny button" type="submit" name="vote" value="normal" >
<i class="icon checkmark"></i>
Get it!
</button>
<button class="ui tiny button" type="submit" name="vote" value="dislike" >
<i class="icon bomb"></i>
Hmmm...
</button>
{% elif user_ticket.choice == 'dislike' %}
<button class="ui tiny button" type="submit" name="vote" value="like" >
<i class="icon checkmark"></i>
Get it!
</button>
<button class="ui red tiny button" type="submit" name="vote" value="normal" >
<i class="icon bomb"></i>
Hmmm...
</button>
{% else %}
<button class="ui tiny button" type="submit" name="vote" value="like" >
<i class="icon checkmark"></i>
Get it!
</button>
<button class="ui tiny button" type="submit" name="vote" value="dislike" >
<i class="icon bomb"></i>
Hmmm...
</button>
{% endif %}
<button class="ui secondary circular tiny right floated pin icon button">
<i class="pin icon"></i>
Saved
</button>
</form>
在后臺中的
設定一個choice為like
設定為like.png
刷新前臺
完成效果
3、投票的存儲邏輯
在views.py中建立投票
引入
from django.core.exceptions import ObjectDoesNotExist
建立處理邏輯
def detail_vote(request, id):
voter_id = request.user.profile1.id
try:
user_ticket_for_this_article = Ticket.objects.get(voter_id=voter_id,article_id=id)# 投票到文章的ID和投票人ID
user_ticket_for_this_article.choice = request.POST['vote']
user_ticket_for_this_article.save()
except ObjectDoesNotExist:
new_ticket = Ticket(voter_id=voter_id, article_id=id,choice=request.POST)
new_ticket.save()
return redirect(to='detail', id=id)
在urls.py中
引入detail_vote，寫入url
urlpatterns = [
url(r'^admin/', admin.site.urls),
url(r'^index/', index, name="index"),
url(r'^detail/(?P<id>\d+)/$', detail, name="detail"),
url(r'^detail/vote/(?P<id>\d+)/$', detail_vote, name="vote"),
url(r'^comment/(?P<id>\d+)/$', comment, name="comment"),
url(r'^login/$', index_login, name="login"),
url(r'^register/$', index_register, name="register"),
url(r'^logout/$', logout,{'next_page': '/login'}, name="logout"),
在模板層中寫入action處理
<form class="ui form" action="{% url 'vote' article.id %}" method="post">
4、統計功能
在views.py中添加
like_counts = Ticket.objects.filter(choice='like', article_id=id).count()
context['like_counts'] = like_counts
在模板中引用
<span style="color:#bbbbbb">{{ like_counts }} people got it</span>
