def edit_post(id): post = Post.find(id) if request.method == 'GET': return render_template('edit_post.html', post=post, articles=Article.all()) elif request.method == 'POST': post.name = request.form['name'] post.description = request.form['description'] post.article = Article.find(request.form['article_id']) images = request.files.getlist("file") shutil.rmtree(post.file_path) letters = string.ascii_lowercase direc_path = random.choice(letters) direc = request.form['name'] os.mkdir("static/images/" + direc + User.find_by_id(session['USERNAME']) + direc_path) for img in images: img_path = 'static/images/' + direc + User.find_by_id( session['USERNAME']) + direc_path + "/" img.save(img_path + img.filename) post.file_path = img_path post.save() #logging.info('%s with id: %s edited post %s', User.find_by_id(session['USERNAME']), session['USERNAME'], post.id) return redirect(url_for('show_post', id=post.id))
def new_post(): if request.method == 'GET': return render_template('new_post.html', articles=Article.all()) elif request.method == 'POST': letters = string.ascii_lowercase direc_path = random.choice(letters) direc = request.form['name'] images = request.files.getlist("file") os.mkdir("static/images/" + direc + User.find_by_id(session['USERNAME']) + direc_path) for img in images: img_path = 'static/images/' + direc + User.find_by_id( session['USERNAME']) + direc_path + "/" img.save(img_path + img.filename) article = Article.find(request.form['article_id']) values = (None, request.form['name'], request.form['description'], article, img_path, session['USERNAME']) Post(*values).create() #logging.info('%s with id: %s added new post', User.find_by_id(session['USERNAME']), session['USERNAME']) return redirect('/')
def get_articles(): return render_template("articles.html", articles=Article.all())