wzy/djangoProject2/movie/views.py

import os.path
import time

from django.contrib import messages
from django.db.models import Max,Count
from django.shortcuts import render,redirect,reverse
from django.views.generic import View,ListView,DetailView


from.forms import RegisterForm,LoginForm,CommentForm
from.model import User, Movie, Movie_rating, Movie_hot

BASE=os.path.dirname(os.path.abspath(__file__))


class IndexView(ListView):
    model=Movie
    template_name='movie/index.html'
    paginatee_by=15
    context_objject_name='movies'
    ordering='imdb_id'
    page_kwarg='p'


    def get_queryset(self):
        return Movie.objects.filter(imdb_id__lte=1000)


    def get_context_data(self,*,object_list=None,**kwargs):
        context=super(IndexView,self).get_context_data(**kwargs)
        paginator=context.get('paginator')
        page_obj=context.get('page_obj')
        pagination_data=self.get_pagination_data(paginator,page_obj)
        context.updata(pagination_data)
        return context


    def get_pagination_data(self,paginator,page_obj,around_count=2):
        current_page=page_obj.number
        if current_page<=around_count+2:
            left_pages=range(1,current_page)
            left_has_more=False
        else:
            left_pages=range(current_page-around_count,current_page)
            left_has_more=True


        if current_page>=paginator.num_pages-around_count-1:
            right_pages=range(current_page+1,paginator.num_pages+1)
            right_has_more=False
        else:
            right_pages=range(current_page+1,current_page+around_count+1)
            right_has_more=True
        return{
            'left_pages':left_pages,
            'right_pages':right_pages,
            'current_page':current_page,
            'left_has__more':left_has_more,
            'right_has__more':right_has_more
        }



class PopularMovieView(ListView):
    model=Movie_hot
    template_name='movie/hot.html'
    paginate_by=15
    context_object_name='movies'
    page_kwarg='p'


    def get_queryset(self):
        movies=Movie.objects.annotate(nums=Count('movie_rating__score')).order_by('-nums')[:100]
        for movie in movies:
            record=Movie_hot(movie=movie,rating_number=movie.nums)
            record.save()


        hot_movies=Movie_hot.objects.all().values("movie_id")
        movies=Movie.objects.filter(id__in=hot_movies).ammotate(nums=Max('movie_hot__num')).order_by('-nums')
        return movies


    def get_context_data(self,*,object_list=None,**kwargs):
        context=super(PopularMovieView,self).get_context_data(**kwargs)
        paginator=context.get('paginator')
        page_obj=context.get('page_obj')
        pagination_data=self.get_pagination_data(paginator,page_obj)
        context.updata(pagination_data)
        return context


    def get_pagination_data(self,paginator,page_obj,around_count=2):
        current_page=page_obj.number


        if current_page<=around_count+2:
            left_pages=range(1,current_page)
            left_has_more=False
        else:
            left_pages=range(current_page-around_count,current_page)
            left_has_more=True

        if current_page>=paginator.num_pages-around_count-1:
            right_pages=range(current_page+1,paginator.num_pages+1)
            right_has_more=False
        else:
            right_pages=range(current_page+1,current_page+around_count+1)
            right_has_more=True
        return{
            'left_pages':left_pages,
            'right_pages':right_pages,
            'current_page':current_page,
            'left_has__more':left_has_more,
            'right_has__more':right_has_more
        }
class TagView(ListView):
    model=Movie
    template_name='movie/tag.html'
    paginate_by=15
    context_object_name='movies'
    page_kwarg='p'


    def get_queryset(self):
        if 'genre' not in self.request.GET.dict().keys():
            movies=Movie.objects.all()
            return movies[100:200]

        else:
            movies=Movie.objects.filter(genre__name=self.request.GET.dict()['genre'])
            print(movies)
            return movies[:100]
    def get_context_data(self,*,object_list=None,**kwargs):
        context=super(TagView,self).get_context_data(**kwargs)
        if "genre" in self.request.GET.dict().keys():
            genre=self.request.GET.dict()['genre']
            context.update({'genre':genre})
        paginator=context.get('paginator')
        page_obj=context.get('page_obj')
        pagination_data=self.get_pagination_data(paginator,page_obj)
        context.updata(pagination_data)
        return context

    def get_pagination_data(self,paginator,page_obj,around_count=2):
        current_page=page_obj.number

        if current_page<=around_count+2:
            left_pages=range(1,current_page)
            left_has_more=False
        else:
            left_pages=range(current_page-around_count,current_page)
            left_has_more=True

        if current_page>=paginator.num_pages-around_count-1:
            right_pages=range(current_page+1,paginator.num_pages+1)
            right_has_more=False
        else:
            right_pages=range(current_page+1,current_page+around_count+1)
            right_has_more=True
        return {
            'left_pages':left_pages,
            'right_pages':right_pages,
            'current_page':current_page,
            'left_has__more':left_has_more,
            'right_has__more':right_has_more
        }
class SearchView(ListView):
    model=Movie
    template_name='movie/search.html'
    paginate_by=15
    context_object_name='movies'
    page_kwarg='p'

    def get_queryset(self):
        movies=Movie.objects.filter(name__icontains=self.request.GET.dict()['keyword'])
        return movies
    def get_context_data(self,*,object_list=None,**kwargs):
        context=super(SearchView,self).get_context_data(**kwargs)
        paginator=context.get('paginator')
        page_obj=context.get('page_obj')
        pagination_data=self.get_pagination_data(paginator,page_obj)
        context.updata(pagination_data)
        context.updata({'keyword':self.request.GET.dict()['keyword']})
        return context
    def get_pagination_data(self,paginator,page_obj,around_count=2):
        current_page=page_obj.number
        if current_page<=around_count+2:
            left_pages=range(1,current_page)
            left_has_more=False
        else:
            left_pages=range(current_page-around_count,current_page)
            left_has_more=True

        if current_page>=paginator.num_pages-around_count-1:
            right_pages=range(current_page+1,paginator.num_pages+1)
            right_has_more=False
        else:
            right_pages=range(current_page+1,current_page+around_count+1)
            right_has_more=True

        return {
                'left_pages':left_pages,
                'right_pages':right_pages,
                'current_page':current_page,
                'left_has__more':left_has_more,
                'right_has__more':right_has_more
               }



class RegisterView(View):
    def get(self,request):
        return render(request,'movie/register.html')

    def post(self,request):
        form=RegisterForm(request.POST)
        if form.is_valid():
            form.save()
            return redirect(reverse('movie.index'))
        else:
            errors=form.get_errors()
            for error in errors:
                messages.info(request,error)
            print(form.errors.get_json_data())
            return redirect(reverse('movie:register'))

class LoginView(View):
    def get(self,request):
        return render(request,'movie/login.html')

    def post(self,request):
        form=LoginForm(request.POST)
        if form.is_valid():
            name=form.cleaned_data.get('name')
            pwd=form.cleaned_data.get('password')
            remember=form.cleaned_data.get('remember')
            user=User.objects.filter(name=name,password=pwd).first()
            if user:
                if remember:
                    request.session.set_expiry(None)
                else:
                    request.session.set_expiry(0)
                request.session['user_id']=user.id
                return redirect(reverse('movie.index'))
            else:
                messages.info(request,'用户名或密码错误')
                return redirect(reverse('movie:login'))
        else:
            errors=form.get_errors()
            for error in errors:
                messages.info(request,error)
            return redirect(reverse('movie:login'))
def UserLogout(request):
        request.session.set_expiry(-1)
        return redirect(reverse('movie:index'))

class MovieDetailView(DetailView):
    model=Movie
    template_name='movie/detail.html'
    context_object_name = 'movie'
    def get_context_data(self,**kwargs):
        global user_id
        context=super().get_context_data(**kwargs)
        login=True
        try:
            user_id=self.request.session['user_id']
        except KeyError as e:
            login=False

        pk=self.kwargs['pk']
        movie=Movie.objects.get(pk=pk)
        if login:
            user=User.objects.get(pk=user_id)
            rating=Movie_rating.objects.filter(user=user,movie=movie).first()
            score=0
            comment=''
            if rating:
                score=rating.score
                comment=rating.comment
            context.update({'score':score,'comment':comment})
        similary_movies=movie.get_similarity()
        context.update({'similary_movies':similary_movies})
        context.update({'login':login})
        return context

    def post(self,request,pk):
        form=CommentForm(request.POST)
        if form.is_valid():
            score=form.cleaned_data.get('score')
            comment=form.cleaned_data.get('comment')
            user_id=request.session['user_id']
            user=User.objects.get(pk=user_id)
            movie=Movie.objects.get(pk=pk)
            rating=Movie_rating.objects.filter(user=user,movie=movie).first()
            if rating:
                rating.score=score
                rating.comment=comment
                rating.save()
            else:
                rating=Movie_rating(user=user,movie=movie,score=score,comment=comment)
                rating.save()
            messages.info(request,"评论成功！")
        else:
            messages.info(request,"评论不能为空！")
        return redirect(reverse('movie:detail',args=(pk,)))

class RatingHistoryView(DetailView):
    model=User
    template_name='movie/history.html'
    qcontext_object_name='user'

    def get_context_data(self,**kwargs):
        context=super().get_context_data(**kwargs)
        user_id=self.request.session['user_id']
        user=User.objects.get(pk=user_id)
        ratings=Movie_rating.objects.filter(user=user).order_by('-score')
        context.update({'ratings':ratings})
        return context

def delete_recode(request,pk):
    movie=Movie.objects.get(pk=pk)
    user_id=request.session['user_id']
    user=User.objects.get(pk=user_id)
    rating=Movie_rating.objects.get(user=user,movie=movie)
    rating.delete()
    messages.info(request,f'删除{movie.name}评分记录成功！')
    return redirect(reverse('movie:history',args=(user_id,)))

class RecommendMovieView(ListView):
    model=Movie
    template_name='movie/recommend.html'
    paginate_by=15
    context_object_name='movies'
    ordering='movie_rating__score'
    page_kwarg='p'

    def __int__(self):
        super().__init__()
        self.K=20
        self.N=10
        self.cur_user_movie_qs=None

    def get_user_sim(self):
        user_sim_dct=dict()
        cur_user_id=self.request.session['user_id']
        cur_user=User.objects.get(pk=cur_user_id)
        other_users=User.objects.exclude(pk=cur_user_id)
        self.cur_user_movie_qs=Movie.objects.filter(user=cur_user)
        for user in other_users:
            user_sim_dct[user.id]=len(Movie.objects.filter(user=user)&self.cur_user_movie_qs)
        print("user similarity calculated!")
        return sorted(user_sim_dct.items(),key=lambda x:-x[1][:self.K])
    def get_recommend_movie(self,user_lst):
        movie_val_dct=dict()
        for user in user_lst:
            movie_set=Movie.objects.filter(user=user).exclude(id__in=self.cur_user_movie_qs).annotate(score=Movie('movie_rating__score')*Movie('movie_rating__score'))
            for movie in movie_set:
                movie_val_dct.setdefault(movie,0)
                movie_val_dct[movie]+=movie.score
        return sorted(movie_val_dct.items(),key=lambda x:-x[1])[:self.N]
    def get_queryset(self):
        s=time.time()
        user_lst=self.get_user_sim()
        movie_lst=self.get_recommend_movie(user_lst)
        result_lst=[]
        for movie, in movie_lst:
            result_lst.append(movie)
        e=time.time()
        print(f"算法推荐用时:[e-s]秒")
        return result_lst
    def get_context_data(self,*,object_list=None,**kwargs):
        context=super(RecommendMovieView,self).get_context_data(**kwargs)
        print(context)
        paginator=context.get('pagginator')
        page_obj=context.get('page_obj')
        pagination_data=self.get_pagination_data(paginator,page_obj)
        context.update(pagination_data)
        return context

    def get_pagination_data(self,paginator,page_obj,around_count=2):
        current_page=page_obj.number
        if current_page<=around_count+2:
            left_pages=range(1,current_page)
            left_has_more=False
        else:
            left_pages=range(current_page-around_count,current_page)
            left_has_more=True

        if current_page>=paginator.num_pages-around_count-1:
            right_pages=range(current_page+1,paginator.num_pages+1)
            right_has_more=False
        else:
            right_pages=range(current_page+1,current_page+around_count+1)
            right_has_more=True

        return {
                'left_pages':left_pages,
                'right_pages':right_pages,
                'current_page':current_page,
                'left_has__more':left_has_more,
                'right_has__more':right_has_more
               }