【创新实训】 后端开发记录（2）：电影详情后台

需求

RESTful API风格
GET 传入id
返回id对应的电影信息，该电影的相关评论，并将该电影加入用户的历史记录中

models.py

用了Movie，Comments，User三个model

class Movie(Document):
    movieId=IntField()
    source=DictField()
    name = StringField()
    nameFrn = StringField()
    directors = ListField(StringField())
    writers = ListField(StringField())
    stars = ListField(StringField())
    types = ListField(StringField())
    country = ListField(StringField())
    language = ListField(StringField())
    releaseDate = ListField(StringField())
    runtime = IntField()
    imdb = StringField()
    summary = StringField()
    timestamp = LongField()
    year = StringField()
    _id = ObjectIdField(primary_key=True)

class Comments(Document):
    _id = ObjectIdField()
    movieId=StringField()
    user=StringField()
    userId=StringField()
    rating=StringField()
    content=StringField()
    time=StringField()

class User(Document):
    _id = ObjectIdField(primary_key=True)
    phone = StringField()
    name = StringField()
    pwd = StringField()
    emb = ListField(FloatField())
    history = ListField(DictField())
    question = IntField()

urls.py

添加url

urlpatterns = [
    
    path('api/movie/<int:id>/', view=views.movie),

]

views.py

通过电影的id从movie集合中查找电影，从comments集合中查找电影的评论，并将评论加入电影的字典中，如果存在登录用户，从cookie中取出用户_id，查找用户并将电影id和访问时间存入用户的history，实现history的去重、排序，将旧记录去除以维护history长度为20

@api_view(['GET'])
def movie(request, id):
    movie = Movie.objects.exclude('_id').filter(movieId=id).first()
    if movie is None:
        return JsonResponse({'data': None, 'msg': '不存在该电影'}, json_dumps_params={'ensure_ascii': False})
    movie = movie.to_mongo()
    
    if int(datetime.datetime.now()+datetime.timedelta(days=-2))>movie['timestamp']:
        data = craviews.updateFromDouban(movie['source']['douban']['sourceId'])
        for key in ["cover", "rating", "rateNum"]:
            if key in data:
                movie['source']['douban'][key] = data[key]
        Movie.objects.filter(movieId=movie['movieId']).update_one(source=movie['source'], timestamp=data['timestamp'])
    
    movie['comments']=[]
    if "mtime" in movie['source']:
        comments = Comments.objects.exclude('_id').filter(movieId=id)
        movie['comments'] = json.loads(comments.to_json()) if comments else []

    if '_id' in request.COOKIES:
        _id = request.COOKIES['_id']
        user = User.objects.with_id(_id)
        if user:
            if 'history' not in user:
                user.history = []

            flag = True
            for history in user.history:
                if history['id'] == id:
                    history['timestamp'] = int(time.time())
                    flag = False
            if flag:
                user.history.append({'id': id, 'timestamp': int(time.time())})
            user.history.sort(key=historySortOrder, reverse=True)
            if len(user.history) > 19:
                user.history.pop()

            User.objects.filter(_id=ObjectId(_id)).update_one(history=user.history)

    request.session.set_expiry(3600)
    request.session.setdefault('sess', [])
    request.session['sess'] += [id]

    resp = JsonResponse({'data': movie}, json_dumps_params={'ensure_ascii': False})
    return resp

history数据库实例