def put(self, movie_id):
json_data = request.get_json()
movie = Movie.get_by_id(movie_id=movie_id)
if movie is None:
return {'message': 'movie not found'}, HTTPStatus.NOT_FOUND
identity = get_jwt_identity()
current_user = User.get_by_id(identity)
if not current_user.is_admin:
return {'message': 'Not authorized'}, HTTPStatus.UNAUTHORIZED
movie.name = json_data['name']
movie.year = json_data['year']
movie.rating = (json_data['rating'])
movie.description = json_data['description']
movie.director = json_data['director']
movie.duration = json_data['duration']
movie.age_rating = json_data['age_rating']
movie.save()
return movie.data(), HTTPStatus.OK
def put(self, movie_id):
json_data = request.get_json()
movie = Movie.get_by_id(movie_id=movie_id)
if movie is None:
return {'message': 'movie not found'}, HTTPStatus.NOT_FOUND
identity = get_jwt_identity()
current_user = User.get_by_id(identity)
if not current_user.is_admin:
return {'message': 'Not authorized'}, HTTPStatus.UNAUTHORIZED
print(type(json_data['rating']))
new_rating_list = movie.rating
print(new_rating_list)
new_rating_list.append(json_data['rating'])
print(new_rating_list)
movie.name = movie.name
movie.year = movie.year
movie.rating = new_rating_list
movie.description = movie.description
movie.director = movie.director
movie.duration = movie.duration
movie.age_rating = movie.age_rating
movie.save()
return movie.data(), HTTPStatus.OK
def get(self, movie_id):
movie = Movie.get_by_id(movie_id=movie_id)
if movie is None:
return {'message': 'movie not found'}, HTTPStatus.NOT_FOUND
movie.rating = count_rating(movie.rating)
return movie.data(), HTTPStatus.OK
def delete(self, movie_id):
movie = Movie.get_by_id(movie_id=movie_id)
if movie is None:
return {'message': 'movie not found'}, HTTPStatus.NOT_FOUND
identity = get_jwt_identity()
current_user = User.get_by_id(identity)
if not current_user.is_admin:
return {'message': 'Not authorized'}, HTTPStatus.UNAUTHORIZED
movie.delete()
return {'message': 'deleted successfully!'}, HTTPStatus.OK
def test_get_by_id(self):
m1_id = Movie.query.filter_by(title='t1').first().id
self.assertEqual(
Movie.query.filter_by(id=m1_id).first(), Movie.get_by_id(m1_id))