add user setting form and view class

Author: killpanda

firefly/forms/user.py

@@ -1,8 +1,9 @@
 # coding=utf-8
 from __future__ import absolute_import
 from flask_wtf import Form
+from flask_login import current_user
 from wtforms import StringField, PasswordField
-from wtforms.validators import ValidationError, Email, Required
+from wtforms.validators import ValidationError, Email, Required, URL
 
 from firefly.models.user import User
 
@@ -39,3 +40,17 @@ def validate_password(self, field):
             self.user = user
         else:
             raise ValidationError('邮箱或密码错误')
+
+
+class ProfileForm(Form):
+    location = StringField('Location')
+    website = StringField('URL', [URL()])
+    github_id = StringField('Github')
+
+    def save(self):
+        user = current_user
+        if user and not user.is_anonymous():
+            user.location = self.location.data
+            user.website = self.website.data
+            user.github_id = self.github_id.data
+            user.save()

firefly/views/user.py

@@ -1,8 +1,13 @@
 # coding=utf-8
 from __future__ import absolute_import
 
+from flask import url_for, redirect
+
 from flask.views import MethodView
 from flask.blueprints import Blueprint
+from flask.ext.login import login_required, current_user
+from firefly.libs.template import render_template
+from firefly.forms.user import ProfileForm
 
 
 bp = Blueprint('user', __name__, url_prefix='/user')
@@ -12,4 +17,17 @@ class UserView(MethodView):
     def get(self, id):
         return ''
 
+
+class UserSettingsView(MethodView):
+    decorators = [login_required]
+
+    def get(self):
+        return render_template('user/settings.html', user=current_user)
+
+    def post(self):
+        form = ProfileForm()
+        form.save()
+        return redirect(url_for('user.settings'))
+
 bp.add_url_rule('/<id>/', view_func=UserView.as_view('detail'))
+bp.add_url_rule('/settings', view_func=UserSettingsView.as_view('settings'))