def add_new_genre(request): # return jsonify(request.get_json()) type = request.json['type'] genre = request.json['genre'] if type is not None: new_genre = Genre(type=type, genre=genre) Genre.save(new_genre) headers = {"Description": "OK"} result = { 'id': new_genre.id, 'type': new_genre.type, 'genre': new_genre.genre } response = jsonify(result) response.status_code = 200 response.headers = headers return response else: headers = {"Description": "Invalid input"} result = { "invalid_fields": [{ "field": "type", "reason": "Type field is required" }] } response = jsonify(result) response.status_code = 400 response.headers = headers return response
def save_all_data(self, movies_iterator) -> int: """ Iterate over movies saving them in postgres content tables @param movies_iterator: the movies @return: number of movies saved """ cursor = self.cursor for movie in movies_iterator: film_work = FilmWork(title=movie['title'], description=movie['description'], rating=movie['imdb_rating']) film_work.save(cursor) for role in ['actor', 'director', 'writer']: for person_name in movie[role]: person = Person(name=person_name) person.save(cursor) FilmWorkPerson(film_work_id=film_work.id, person_id=person.id, role=role).save(cursor) for genre_name in movie['genre']: genre = Genre(name=genre_name) genre.save(cursor) FilmWorkGenre(film_work_id=film_work.id, genre_id=genre.id).save(cursor) self._counter += 1 print('.', end='', flush=True) return self._counter
def genre_save(request): if request.POST.get('genre_id'): genre = Genre.objects.get(id=request.POST.get('genre_id')) genre.name = request.POST.get('genre_name') genre.description = request.POST.get('genre_description') genre.save() else: genre = Genre( name=request.POST.get('genre_name'), description=request.POST.get('genre_description'), ) genre.save() return redirect('music.views.genre_detail', genre_id=genre.id)
def test_should_save_valid_genre(self): genre = Genre(name='TestGenre', description='This is a description') genre.save() all_genres = Genre.objects.all() self.assertEqual(len(all_genres), 1)
def create(self, validated_data): slug = validated_data['name'].lower().replace('-', '_') new_genre = Genre(name=validated_data['name'], slug=slug) new_genre.save() # add genre return new_genre
def addGenretoAlbum(genre: Genre, album: Album): if genre not in album.getGenres(): album.genres.connect(genre) genre.save() album.save()