restore Thibaut's brilliant code

1 files changed, 7 insertions, 7 deletions

diff --git a/famille.py b/famille.py
index 9656ae3..4bc11f7 100644
--- a/famille.py
+++ b/famille.py
@@ -267,14 +267,14 @@ def edit_user():
                                        timezones=pytz.common_timezones)
         except KeyError:
             pass
-        args = tuple(request.form[key] for key in
-                     ['email', 'phone', 'birthday', 'nameday', 'address_line1',
-                      'address_line2', 'address_city_line', 'timezone'])
-        args += ("notify" in request.form, session['user_id'])
-        sqlstr = "UPDATE users SET email= ?, phone=?, birthday=?, nameday=?,"\
-            "address_line1=?, address_line2=?, address_city_line=?, "\
-            "timezone=?, notify=? where id=?"
+
+        result['notify'] = "notify" in request.form
+        result.update({k: v for k, v in request.form.items() if k not in
+                       ['password', 'password_confirm', 'notify']})
+        set_string = ",".join([f"{k}=?" for k in result.keys()])
+        sqlstr = f"UPDATE users SET {set_string} WHERE id=?"
         db = get_db()
+        args = (*result.values(), session['user_id'])
         db.execute(sqlstr, args)
         db.commit()
         session["timezone"] = request.form["timezone"]