Finish the timezine feature

1 files changed, 3 insertions, 2 deletions

diff --git a/famille.py b/famille.py
index 47e98ea..6bae72f 100644
--- a/famille.py
+++ b/famille.py
@@ -188,10 +188,11 @@ def edit_user():
                      '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=?, notify=? " \
+        "address_line1=?, address_line2=?, address_city_line=?, timezone=?, notify=? " \
         "where id=?"
         g.db.execute(sqlstr, args)
         g.db.commit()
+        session["timezone"] = request.form["timezone"]
         return redirect(url_for('view_user', user_id=session['user_id']))
 
 @app.route('/login/', methods=['GET', 'POST'])
@@ -208,7 +209,7 @@ def login():
             if user['password'] == password:
                 session['user_name'] = user['user_name']
                 session['user_id'] = user['id']
-                session['timezone'] = user['timezone']
+                session['timezone'] = user['timezone'] or "UTC"
                 g.db.execute("UPDATE users SET last_seen=? WHERE id=?",
                 (datetime.utcnow(),session['user_id']))
                 g.db.commit()