Allow filtering by genre

1 files changed, 14 insertions, 0 deletions

diff --git a/src/models.py b/src/models.py
index d271b62..0297551 100644
--- a/src/models.py
+++ b/src/models.py
@@ -1,2 +1,16 @@
 all_progress = ['Unread', 'Reading', 'Read', 'Stopped']
 langs = ['fr', 'en', 'de']
+
+# Genres
+
+all_genres = 'Tous les genres'
+no_genre = 'Aucun genre'
+
+def get_genres(books):
+    genres = {all_genres}
+    for data in books.values():
+        for genre in data['genres']:
+            genres.add(genre)
+        if len(data['genres']) == 0:
+            genres.add(no_genre)
+    return list(genres)