diff --git a/src/app.py b/src/app.py index a0e522b..64c1b15 100644 --- a/src/app.py +++ b/src/app.py @@ -136,7 +136,7 @@ def write_entry(): def delete_entry(ident): if not current_user.is_authenticated: return redirect(url_for("index")) - if current_user.id == db.get_entry_by_id(ident).user_id: + if current_user.id == db.get_entry_by_id(ident).user.id: db.delete_entry(ident) return redirect(url_for("index")) diff --git a/src/content.py b/src/content.py index 467e34e..4543adc 100644 --- a/src/content.py +++ b/src/content.py @@ -33,16 +33,17 @@ def gen_arch_string(): entries = db.get_entries() if entries is None: return "" - entries.sort(key=lambda y: db.get_item_by_id(y.item_id).name) + entries.sort(key=lambda y: y.item.name) entries.reverse() - entries.sort(key=lambda y: y.date) + entries.sort(key=lambda y: y.item.date) entries.reverse() for entry in entries: ident = entry.id - title = db.get_item_by_id(entry.item_id).name - year = entry.date + title = entry.item.name + print("title", title) + year = entry.item.date rating = entry.rating - username = db.get_user_by_id(entry.user_id).name + username = entry.user.name if year != last_year: if last_year != "": content_string += "\n" @@ -70,16 +71,16 @@ def gen_user_string(name): entries = db.get_entries_by_user(name) if entries is None: return "" - entries.sort(key=lambda y: db.get_item_by_id(y.item_id).name) + entries.sort(key=lambda y: y.item.name) entries.reverse() - entries.sort(key=lambda y: y.date) + entries.sort(key=lambda y: y.item.date) entries.reverse() for entry in entries: ident = entry.id - title = db.get_item_by_id(entry.item_id).name - year = entry.date + title = entry.item.name + year = entry.item.date rating = entry.rating - username = db.get_user_by_id(entry.user_id).name + username = entry.user.name if year != last_year: if last_year != "": content_string += "\n" @@ -110,11 +111,11 @@ def gen_index_string(): entries.reverse() for entry in entries: ident = entry.id - title = db.get_item_by_id(entry.item_id).name - year = entry.date + title = entry.item.name + year = entry.item.date text = entry.text rating = entry.rating - username = db.get_user_by_id(entry.user_id).name + username = entry.user.name reviewed = entry.reviewed content_string += "