Add route and button for populating userAlbumGenres

Author: CalPinSW

backend/src/app.py

@@ -25,7 +25,7 @@ def create_app():
         SESSION_COOKIE_SECURE="True",
     )
 
-    cors = CORS(
+    CORS(
         app,
         resources={
             r"/*": {

backend/src/controllers/database.py

@@ -107,13 +107,20 @@ def populate_additional_album_details():
                 )
                 update_album(db_album)
 
-        return make_response("Playlist data populated", 201)
+        return make_response("Playlist details populated", 201)
 
     @database_controller.route("populate_universal_genre_list", methods=["GET"])
     def populate_universal_genre_list():
         genre_list = musicbrainz.get_genre_list()
         [create_genre(genre) for genre in genre_list]
-        return make_response("Playlist data populated", 201)
+        return make_response("Genre data populated", 201)
+
+    @database_controller.route("populate_user_album_genres", methods=["GET"])
+    def populate_user_album_genres():
+        access_token = request.cookies.get("spotify_access_token")
+        user = spotify.get_current_user(access_token)
+        populate_album_genres_by_user_id(user.id, musicbrainz)
+        return make_response("User album genres populated", 201)
 
     return database_controller
 
@@ -124,7 +131,9 @@ def split_list(input_list, max_length=20):
     ]
 
 
-def populate_user_album_genres(user_id: str):
+def populate_album_genres_by_user_id(
+    user_id: str, musicbrainz: MusicbrainzClient = MusicbrainzClient()
+):
     albums = get_user_albums(user_id=user_id)
     print(f"processing album {0} of {len(albums)}")
     skip_count = 0
@@ -135,7 +144,7 @@ def populate_user_album_genres(user_id: str):
             skip_count += 1
             continue
         album_artists = get_album_artists(db_album)
-        genres = MusicbrainzClient().get_album_genres(
+        genres = musicbrainz.get_album_genres(
             artist_name=album_artists[0].name, album_title=db_album.name
         )
         add_genres_to_album(db_album, genres)

frontend/src/api/index.ts

@@ -145,4 +145,8 @@ export const populateAdditionalAlbumDetails = async (): Promise<Response> => {
 
 export const populateUniversalGenreList = async (): Promise<Response> => {
 	return request('database/populate_universal_genre_list', RequestMethod.GET)
+}
+
+export const populateUserAlbumGenres = async (): Promise<Response> => {
+	return request('database/populate_user_album_genres', RequestMethod.GET)
 }

frontend/src/settingsPage/SettingsPage.tsx

@@ -1,6 +1,6 @@
 import React, { FC } from "react";
 import { ButtonWithLoadingState } from "./PopulateUserDatabaseButton";
-import { populateAdditionalAlbumDetails, populateUniversalGenreList, populateUserData } from "../api";
+import { populateAdditionalAlbumDetails, populateUniversalGenreList, populateUserAlbumGenres, populateUserData } from "../api";
 
 export const SettingsPage: FC = () => {
   return (
@@ -10,6 +10,7 @@ export const SettingsPage: FC = () => {
           <ButtonWithLoadingState text="Populate user data" actionCallback={populateUserData} />
           <ButtonWithLoadingState text="Populate additional album details" actionCallback={populateAdditionalAlbumDetails} />
           <ButtonWithLoadingState text="Populate universal genre list" actionCallback={populateUniversalGenreList} />
+          <ButtonWithLoadingState text="Populate user album genres" actionCallback={populateUserAlbumGenres} />
         </div>
     </div>
   );