def register():
form = RegistrationForm()
if g.user is not None:
return redirect(url_for("users.home"))
if request.method == "POST" and form.validate():
if User.query.filter_by(username=form.username.data).first():
flash("The username already exists idiot")
return render_template("users/register.html", title="register", form=form)
if User.query.filter_by(email=form.email.data).first():
flash("The email has already been registored")
return render_template("users/register.html", title="register", form=form)
file = form.image.data
profile_pic = utilities.file_save(file, "profilepics")
user = User(
username=form.username.data,
email=form.email.data,
password=generate_password_hash(form.password.data),
description=form.description.data,
profile_pic=profile_pic,
homepage=form.homepage.data,
role=form.role.data,
zodiac=form.zodiac.data,
)
psc_db.session.add(user)
psc_db.session.commit()
session["user_id"] = user.id
flash("well done f****t")
return redirect(url_for("users.profile", userId=user.id))
return render_template("users/register.html", title="register", form=form)
def create_post():
user = g.user
form = PostForm()
if request.method == 'POST' and form.validate():
user_id = g.user.id
file = form.image.data
image = utilities.file_save(file, 'images')
post = Post(title = form.title.data,body = form.body.data, image = image, timestamp = datetime.utcnow(), category = form.category.data, user_id = user_id, front_page = form.front_page.data )
psc_db.session.add(post)
psc_db.session.commit()
flash('well posted mutafuqa')
return redirect(url_for('pages.index'))
return render_template("posts/create_post.html", title="write something", user =user, form = form)
def modify(postId):
post = Post.query.get(postId)
form = PostForm()
user = g.user
if request.method == 'POST' and form.validate():
post.title = form.title.data
post.body = form.body.data
post.category = form.category.data
post.front_page = form.front_page.data
if form.image.data:
file = form.image.data
post.image = utilities.file_save(file,'images')
psc_db.session.add(post)
psc_db.session.commit()
return redirect(url_for('posts.manage_posts'))
form.title.data = post.title
form.body.data = post.body
form.category.data = str(post.category)
form.front_page.data = post.front_page
return render_template("posts/create_post.html", user = user, form = form)
