def column_result():
"""
Fast estimate Column
"""
data = dict(doc_id=uu_id('item'))
data['tag'] = request.form.get('tag')
data['wg_tag'] = request.form.get('wg_tag')
data['category'] = Category.query.get_or_404(request.form.get('category'))
data['width'] = request.form.get('width')
data['depth'] = request.form.get('depth')
data['height'] = request.form.get('height')
data['amount'] = request.form.get('amount')
data['mainbar'] = request.form.get('main_bar_type')
data['compositebar'] = request.form.get('composite_bar_type')
data['stirup'] = request.form.get('stir_bar_type')
data['mainbar_amt'] = request.form.get('main_bar_amt')
data['compositebar_amt'] = request.form.get('composite_bar_amt')
data['stirup_spacing'] = request.form.get('stir_spacing')
column = Column(data['doc_id'], data['tag'], data['wg_tag'],
data['category'], data['width'], data['depth'],
data['height'], data['amount'], data['mainbar'],
data['compositebar'], data['stirup'],
data['stirup_spacing'], data['mainbar_amt'],
data['compositebar_amt'])
db.session.add(column)
db.session.commit()
flash('Beam {} added to project!'.format(data['tag']), 'success')
return redirect(url_for('index.getwall'))
def beam_result():
"""
Fast estimate Wall
"""
data = dict(doc_id=uu_id('item'))
data['tag'] = request.form.get('tag')
data['wall_tag'] = request.form.get('wall_tag')
data['width'] = request.form.get('width')
data['depth'] = request.form.get('depth')
data['length'] = request.form.get('length')
data['amount'] = request.form.get('amount')
data['category'] = Category.query.get_or_404(request.form.get('category'))
data['topbar'] = request.form.get('topbar')
data['midbar'] = request.form.get('midbar')
data['botbar'] = request.form.get('botbar')
data['stirup'] = request.form.get('stirup')
data['topbar_amt'] = request.form.get('topbar_amt')
data['midbar_amt'] = request.form.get('midbar_amt')
data['botbar_amt'] = request.form.get('botbar_amt')
data['stirup_spacing'] = request.form.get('stirup_spacing')
beam = Beam(data['doc_id'], data['tag'], data['wall_tag'], data['width'],
data['depth'], data['length'], data['amount'],
data['category'], data['topbar'], data['midbar'],
data['botbar'], data['stirup'], data['stirup_spacing'],
data['topbar_amt'], data['midbar_amt'], data['botbar_amt'])
db.session.add(beam)
db.session.commit()
flash('Beam {} added to project!'.format(data['tag']), 'success')
return redirect(url_for('index.getwall'))
def project_edit():
idx = request.args.get("idx")
project_name1 = request.form.get("project_name1")
project_name2 = request.form.get("project_name2")
name = request.form.get("name")
startdate = request.form.get("startdate")
enddate = request.form.get("enddate")
author = request.form.get("author")
col = mongo.db.version_list_customer
col.update_one({"_id": ObjectId(idx)}, {
"$set": {
"project_name1": project_name1,
"project_name2": project_name2,
"name": name,
"startdate": startdate,
"enddate": enddate,
"author": author,
}
})
flash("수정되었습니다.")
return redirect(url_for("project.project_write"))
def category_create():
if request.method == 'POST':
name = request.form.get('name')
category = Category(name)
db.session.add(category)
db.session.commit()
flash('Category {} created has been created!'.format(name), 'success')
return redirect(url_for('index.create_category'))
def board_delete(idx):
board = mongo.db.board
data = board.find_one({"_id": ObjectId(idx)})
if data.get("writer_id") == session.get("id"):
board.delete_one({"_id": ObjectId(idx)})
flash("삭제 되었습니다.")
else:
flash("글 삭제 권한이 없습니다.")
return redirect(url_for("board.lists"))
def board_edit(idx):
if request.method == "GET":
board = mongo.db.board
data = board.find_one({"_id": ObjectId(idx)})
if data is None:
flash("해당 게시물이 존재하지 않습니다.")
return redirect(url_for("board.lists"))
else:
if session.get("id") == data.get("writer_id"):
return render_template("edit.html", data=data, title="글 수정")
else:
flash("글 수정 권한이 없습니다.")
return redirect(url_for("board.lists"))
else:
title = request.form.get("title")
contents = request.form.get("contents")
board = mongo.db.board
data = board.find_one({"_id": ObjectId(idx)})
if data.get("writer_id") == session.get("id"):
board.update_one({"_id": ObjectId(idx)}, {
"$set": {
"title": title,
"contents": contents,
}
})
flash("수정되었습니다.")
return redirect(url_for("board.board_view", idx=idx))
else:
flash("글 수정 권한이 없습니다.")
return redirect(url_for("board.lists"))
def wall_opening():
"""
Fast estimate Wall
"""
data = dict(doc_id=uu_id('item'))
data['tag'] = request.form.get('tag')
data['wall_tag'] = request.form.get('wall_tag')
data['width'] = request.form.get('width')
data['height'] = request.form.get('height')
data['amount'] = request.form.get('amount')
data['category'] = Category.query.get_or_404(request.form.get('category'))
opening = Opening(data['tag'], data['wall_tag'], data['width'],
data['height'], data['amount'], data['category'],
data['doc_id'])
db.session.add(opening)
db.session.commit()
flash('Opening {} added to project!'.format(data['tag']), 'success')
return redirect(url_for('index.getwall'))
def foundation_result():
"""
Fast estimate Foundation
"""
data = dict(doc_id=uu_id('item'))
data['tag'] = request.form.get('tag')
data['wg_tag'] = request.form.get('wg_tag')
data['category'] = request.form.get('category')
data['width'] = request.form.get('width')
data['depth'] = request.form.get('depth')
data['thickness'] = request.form.get('thickness')
data['length'] = request.form.get('length')
data['amount'] = request.form.get('amount')
data['topbar'] = request.form.get('top_bar_type')
data['midbar'] = request.form.get('mid_bar_type')
data['botbar'] = request.form.get('bot_bar_type')
data['stirup'] = request.form.get('stir_bar_type')
data['topbar_amt'] = request.form.get('top_bar_amt')
data['midbar_amt'] = request.form.get('mid_bar_amt')
data['botbar_amt'] = request.form.get('bot_bar_amt')
data['stirup_spacing'] = request.form.get('stir_spacing')
foundation = Foundation(data['doc_id'], data['tag'], data['wg_tag'],
data['width'], data['depth'], data['thickness'],
data['length'], data['amount'], data['category'],
data['topbar'], data['midbar'], data['botbar'],
data['stirup'], data['stirup_spacing'],
data['topbar_amt'], data['midbar_amt'],
data['botbar_amt'])
db.session.add(foundation)
db.session.commit()
flash('Foundation {} added to project!'.format(data['tag']), 'success')
return redirect(url_for('index.getwall'))
def decorated_function(*args, **kwargs):
if session.get("id") is None or session.get("id") == "":
flash("로그인 후 사용 가능한 기능입니다.")
return redirect(
url_for("member.member_login", next_url=request.url))
return f(*args, **kwargs)
def project_delete(idx):
col = mongo.db.version_list_customer
col.delete_one({"_id": ObjectId(idx)})
flash("삭제되었습니다.")
return redirect(url_for("project.project_write"))
