class Users(MethodView):
decorators = [login_required, admin_permission.require(401)]
template_name = 'auth/users.html'
def get(self):
users = models.User.objects.all()
return render_template(self.template_name, users=users)
class SuUser(MethodView):
decorators = [login_required, admin_permission.require(401)]
template_name = 'auth/user.html'
def get_context(self, username, form=None):
if not form:
user = models.User.objects.get_or_404(username=username)
user.weibo = user.social_networks['weibo'].get('url')
user.weixin = user.social_networks['weixin'].get('url')
user.twitter = user.social_networks['twitter'].get('url')
user.github = user.social_networks['github'].get('url')
user.facebook = user.social_networks['facebook'].get('url')
user.linkedin = user.social_networks['linkedin'].get('url')
form = forms.SuUserForm(obj=user)
data = {'form': form}
return data
def get(self, username, form=None):
data = self.get_context(username, form)
return render_template(self.template_name, **data)
def post(self, username):
form = forms.SuUserForm(obj=request.form)
if form.validate():
user = models.User.objects.get_or_404(username=username)
user.email = form.email.data
user.is_email_confirmed = form.is_email_confirmed.data
user.display_name = form.display_name.data
user.biography = form.biography.data
user.homepage_url = form.homepage_url.data or None
user.social_networks['weibo']['url'] = form.weibo.data or None
user.social_networks['weixin']['url'] = form.weixin.data or None
user.social_networks['twitter']['url'] = form.twitter.data or None
user.social_networks['github']['url'] = form.github.data or None
user.social_networks['facebook'][
'url'] = form.facebook.data or None
user.social_networks['linkedin'][
'url'] = form.linkedin.data or None
user.save()
msg = 'Succeed to update user profile'
flash(msg, 'success')
return redirect(
url_for('auth.su_edit_user', username=user.username))
return self.get(form)
class User(MethodView):
decorators = [login_required, admin_permission.require(401)]
template_name = 'auth/user.html'
def get_context(self, username, form=None):
if not form:
user = models.User.objects.get_or_404(username=username)
form = forms.UserForm(obj=user)
data = {'form': form}
return data
def get(self, username, form=None):
data = self.get_context(username, form)
return render_template(self.template_name, **data)
def post(self, username):
form = forms.UserForm(obj=request.form)
if form.validate():
user = models.User.objects.get_or_404(username=username)
if user.email != form.email.data:
user.is_email_confirmed = False
user.email = form.email.data
# user.is_active = form.is_active.data
# user.is_superuser = form.is_superuser.data
user.role = form.role.data
user.save()
flash('Succeed to update user details', 'success')
return redirect(url_for('auth.edit_user', username=username))
return self.get(username, form)
def delete(self, username):
user = models.User.objects.get_or_404(username=username)
user.delete()
if request.args.get('ajax'):
return 'success'
msg = 'Succeed to delete user'
flash(msg, 'success')
return redirect(url_for('auth.users'))