def new_post():
form = PostForm()
if form.validate_on_submit():
if form.type.data == "blog":
new_post = BlogPost()
new_post.text = form.text.data
elif form.type.data == "image":
new_post = ImagePost()
new_post.image_url = form.image.data
elif form.type.data == "video":
new_post = VideoPost()
new_post.video_object = form.video.data
elif form.type.data == "quote":
new_post = QuotePost()
new_post.text = form.text.data
new_post.author = form.author.data
new_post.title = form.title.data
new_post.user = User.objects(
username=current_user.username
).one()
new_post.save()
return render_template('new.html', form=form)
def login():
form = LoginForm()
if form.validate_on_submit():
user = User.objects(username=form.username.data).first()
login_user(user, remember=form.remember.data)
identity_changed.send(current_app._get_current_object(),
identity=Identity(user.username))
flash("You have been logged in", category="success")
return redirect(url_for('stock.home'))
return render_template('main/login.html', form=form)
def validate(self):
check_validate = super(RegisterForm, self).validate()
if not check_validate:
return False
user = User.objects(username=self.username.data).first()
if user:
self.username.errors.append("User with that name already exists")
return False
return True
def new_post():
form = PostForm()
print("new_post")
if form.validate_on_submit():
new_post = Post()
new_post.title = form.title.data
new_post.text = form.text.data
new_post.publish_date = datetime.datetime.now()
user = User.objects(id=current_user.id).first()
new_post.user = user
new_post.save()
return render_template('new.html', form=form)
def validate(self):
check_validate = super(LoginForm, self).validate()
if not check_validate:
return False
user = User.objects(username=self.username.data).first()
if not user:
self.username.errors.append('Invalid username or password')
return False
if not user.check_password(self.password.data):
self.username.errors.append('Invalid username or password')
return False
return True
def login():
if current_user.is_authenticated:
return redirect(url_for('home'))
form = LoginForm()
if form.validate_on_submit():
user = User.objects(username=form.username.data).first()
if user is None or not user.check_password(form.password.data):
return redirect(url_for('login', page=1))
login_user(user, remember=form.remember_me.data)
next_page = request.args.get('next')
if not next_page or url_parse(next_page).netloc != '':
next_page = url_for('home', page=1)
return redirect(next_page)
return render_template('login.html', form=form)
def login():
if current_user.is_authenticated:
return redirect(url_for('home'))
form = LoginForm()
if form.validate_on_submit():
user = User.objects(username=form.username.data).first()
if user is None or not user.check_password(form.password.data):
return redirect(url_for('login'))
login_user(user, remember=form.remember_me.data)
next_page = request.args.get('next')
if not next_page or url_parse(next_page).netloc != '':
next_page = url_for('home')
return redirect(next_page)
return render_template('login.html', form=form)
def validate_email(self, email):
user = User.objects(email=email.data).first()
if user is not None:
raise ValidationError('Please use a different email address.')
def validate_username(self, username):
user = User.objects(username=username.data).first()
if user is not None:
raise ValidationError('Please use a different username.')
def validate_username(self, username):
form = LoginForm()
user = User.objects(username=username.data).first()
if user is None or not user.check_password(form.password.data):
raise ValidationError('Invalid username or password')
def validate_email(self, email):
user = User.objects(email=email.data).first()
if user is not None:
raise ValidationError('Please use a different email address.')
def validate_username(self, username):
user = User.objects(username=username.data).first()
if user is not None:
raise ValidationError('Please use a different username.')
def validate_username(self, username):
form = LoginForm()
user = User.objects(username=username.data).first()
if user is None or not user.check_password(form.password.data):
raise ValidationError('Invalid username or password')
sqlite_db.connect()
## Ad-hoc migration
# move all persons
for person in LegacyPerson.select():
if Person.objects(name=person.name).count():
continue
Person(name=person.name).save()
print 'Migrated Person: '+person.name
# move all users
for user in LegacyUser.select():
if User.objects(name=user.name).count():
continue
# transfer user data
User(
name=user.name,
email=user.email,
password=user.password,
is_admin=bool(user.is_admin)
).save()
print 'Migrated User: '+user.name
# move groups
for group in LegacyInventoryGroup.select():
if InventoryGroup.objects(name=group.name).count():
continue
InventoryGroup(