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)