def all_banner_post():
form = AllBannerForm(request.form)
if form.validate_on_submit():
f = request.files['image']
filename = secure_filename(f.filename)
if not filename:
flash(u'请选择图片','error')
return redirect(url_for('.all_banner'))
if not allowed_file(f.filename):
flash(u'图文件名或格式错误。','error')
return redirect(url_for('.all_banner'))
dataetime = datetime.datetime.today().strftime('%Y%m%d')
file_dir = 'users/%s/banner/%s/'%(current_user.id,dataetime)
if not os.path.isdir(current_app.config['UPLOADED_PATH']+file_dir):
os.makedirs(current_app.config['UPLOADED_PATH']+file_dir)
f.save(current_app.config['UPLOADED_PATH'] +file_dir+filename)
Banner.create(
title = form.title.data,
summary = form.summary.data,
alt = form.alt.data,
sort = form.sort.data,
image = file_dir+filename,
user = current_user.id
)
flash(u'添加成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_banner'))
def addtemplates_post():
form = AddTemplateForm(request.form)
if form.validate_on_submit():
UserTemplate.create(name=form.title.data)
flash(u'添加成功！')
else:
flash_errors(form)
return redirect(url_for('.addtemplates'))
def add_photo_post():
form = PhotoForm(request.form)
if form.validate_on_submit():
f = request.files['image']
if not f:
flash(u'请选择缩略图。','error')
return redirect(url_for('.all_photo'))
if not allowed_file(f.filename):
flash(u'缩略图文件名或格式错误。','error')
return redirect(url_for('.all_photo'))
photos_date = request.form.get('photos_date','')
from datetime import datetime as dt
if not photos_date:
photos_date = datetime.datetime.today().strftime('%Y-%m-%d')
else:
try:
photos_date = dt.strptime(photos_date,'%Y-%m-%d')
except Exception as e:
flash(u'拍摄时间格式填写错误。%s'%str(e),'error')
return redirect(url_for('.all_photo'))
filename = create_file_name(f)
dataetime = datetime.datetime.today().strftime('%Y%m%d')
file_dir = 'users/%s/photos/%s/'%(current_user.id,dataetime)
if not os.path.isdir(current_app.config['UPLOADED_PATH']+file_dir):
os.makedirs(current_app.config['UPLOADED_PATH']+file_dir)
f.save(current_app.config['UPLOADED_PATH'] +file_dir+filename)
create_thumbnail(f,80,file_dir,filename)
Photo.create(
title = form.title.data,
summary = form.summary.data,
key = form.key.data,
word = form.word.data,
count = form.count.data,
image_alt = form.image_alt.data,
image = file_dir+filename,
user = current_user.id,
attr_id = form.attr.data,
category = form.category.data,
photos_date = photos_date,
static_context = form.static_context.data
)
flash(u'添加成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_photo'))
def register():
"""Register new user."""
form = RegisterForm(request.form)
if form.validate_on_submit():
User.create(username=form.username.data, email=form.email.data, password=form.password.data, active=True)
flash('Thank you for registering. You can now log in.', 'success')
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
def edit_save_article_post():
form = ArticleForm(request.form)
article = Article.query.get_or_404(request.form.get('id',0))
if article.users != current_user:
abort(404)
if form.validate_on_submit():
f = request.files['image']
if f:
if not allowed_file(f.filename):
flash(u'缩略图文件名或格式错误。','error')
return redirect(url_for('.edit_article',id=request.form.get('id',0)))
#32位哈希名
filename = create_file_name(f)
dataetime = datetime.datetime.today().strftime('%Y%m%d')
file_dir = 'users/%s/article/%s/'%(current_user.id,dataetime)
if not os.path.isdir(current_app.config['UPLOADED_PATH']+file_dir):
os.makedirs(current_app.config['UPLOADED_PATH']+file_dir)
f.save(current_app.config['UPLOADED_PATH'] +file_dir+filename)
if not form.img_width.data:
form.img_width.data = 80
create_thumbnail(f,int(form.img_width.data),file_dir,filename)
article.update(image=file_dir+filename)
article.update(
title = form.title.data,
author = form.author.data,
source = form.source.data,
summary = form.summary.data,
key = form.key.data,
word = form.word.data,
is_enable = form.is_enable.data,
content = form.content.data,
count = form.count.data,
image_alt = form.alt.data,
user = current_user.id,
attr_id = form.attr.data,
category = form.category.data,
static_context = form.static_context.data,
)
flash(u'更新成功！','success')
else:
flash_errors(form)
return redirect(url_for('.edit_article',id=request.form.get('id',0)))
def login():
form = LoginForm(request.form)
if request.method == 'POST':
if form.validate_on_submit():
login_user(form.user)
flash(u'您已登录.', 'success')
redirect_url = request.args.get('next') or url_for('user.members')
return redirect(redirect_url)
else:
flash_errors(form)
return render_template('users/login.html',form=form)
def home():
"""Home page."""
form = LoginForm(request.form)
# Handle logging in
if request.method == 'POST':
if form.validate_on_submit():
login_user(form.user)
flash('You are logged in.', 'success')
redirect_url = request.args.get('next') or url_for('user.members')
return redirect(redirect_url)
else:
flash_errors(form)
return render_template('public/home.html', form=form)
def login():
"""login."""
form = LoginForm(request.form)
if request.method == 'POST':
if form.validate_on_submit():
login_user(form.user)
flash('You are logged in.', 'success')
redirect_url = request.args.get('next') or url_for('user.members')
return redirect(redirect_url)
else:
flash_errors(form)
return dict(form=form)
def add_article_post():
form = ArticleForm(request.form)
if form.validate_on_submit():
f = request.files['image']
if not f:
flash(u'请选择缩略图。','error')
return redirect(url_for('.all_article'))
if not allowed_file(f.filename):
flash(u'缩略图文件名或格式错误。','error')
return redirect(url_for('.all_article'))
#32位哈希名
filename = create_file_name(f)
dataetime = datetime.datetime.today().strftime('%Y%m%d')
file_dir = 'users/%s/article/%s/'%(current_user.id,dataetime)
if not os.path.isdir(current_app.config['UPLOADED_PATH']+file_dir):
os.makedirs(current_app.config['UPLOADED_PATH']+file_dir)
f.save(current_app.config['UPLOADED_PATH'] +file_dir+filename)
create_thumbnail(f,80,file_dir,filename)
Article.create(
title = form.title.data,
author = form.author.data,
source = form.source.data,
summary = form.summary.data,
key = form.key.data,
word = form.word.data,
is_enable = form.is_enable.data,
content = form.content.data,
count = form.count.data,
image_alt = form.alt.data,
image = file_dir+filename,
user = current_user.id,
attr_id = form.attr.data,
category = form.category.data,
static_context = form.static_context.data
)
flash(u'添加成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_article'))
def all_static_context_post():
form = AllStaticContextForm(request.form)
if form.validate_on_submit():
"""
StaticContext.create(
main_sort = form.main_sort.data,
html = form.html.data,
user = current_user.id
)
"""
flash(u'添加成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_static_context'))
def edit_static_context_post(id=0):
form = AllStaticContextForm(request.form)
if form.validate_on_submit():
staticcontext = StaticContext.query.get_or_404(request.form.get('id','0'))
if staticcontext.users !=current_user:
abort(404)
staticcontext.update(
main_sort = form.main_sort.data,
html = form.html.data
)
flash(u'更新成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_static_context'))
def addtemplates_attr_post():
form = AddTemplateCategoryAttrForm(request.form)
usertemplate = UserTemplate.query.get_or_404(form.user_template.data)
if form.validate_on_submit():
CategoryAttr.create(name=form.name.data,
summary=form.summary.data,
mark=form.mark.data,
url=form.url.data,
templates=form.templates.data,
usertemplate=usertemplate)
flash(u'添加成功！', 'success')
else:
flash_errors(form)
form.user_template.data = (usertemplate.id, usertemplate.name)
return redirect(url_for('.addtemplates_attr'))
def addtemplates_context_post():
form = AddTemplateContextForm(request.form)
category = CategoryAttr.query.get_or_404(request.form.get('category', '0'))
if form.validate_on_submit():
TemplatesDefault.create(
main_sort=form.main_sort.data,
html=form.html.data,
# template = usertemplate,
category_attrs=category)
flash(
u'添加成功！模板栏目：%s,排序：%s,内容：“%s”' %
(category.name, form.main_sort.data, form.html.data), 'success')
else:
flash_errors(form)
# form.category.data = (usertemplate.id,usertemplate.name)
return redirect(url_for('.addtemplates_context'))
def user_config_post():
form = WebConfigForm()
if form.validate_on_submit():
if request.form.get('state'):
state = True
else:
state = False
f = request.files['image']
if f:
if not allowed_file(f.filename):
flash(u'LOGO文件名或格式错误。','error')
return redirect(url_for('.user_config'))
#32位哈希名
filename = create_file_name(f)
dataetime = datetime.datetime.today().strftime('%Y%m%d')
file_dir = 'users/%s/logo/%s/'%(current_user.id,dataetime)
if not os.path.isdir(current_app.config['UPLOADED_PATH']+file_dir):
os.makedirs(current_app.config['UPLOADED_PATH']+file_dir)
f.save(current_app.config['UPLOADED_PATH'] +file_dir+filename)
current_user.update(web_logo_img=file_dir+filename)
current_user.update(
web_name = form.web_name.data,
web_key = form.web_key.data,
web_word = form.web_word.data,
article_page = form.article_page.data,
photo_page = form.photo_page.data,
web_logo_word = form.web_logo_word.data,
article_context = form.article_context.data,
photo_context = form.photo_context.data,
web_state = state
)
flash(u'更新成功！','success')
else:
flash_errors(form)
return redirect(url_for('.user_config'))
def edit_category_post():
form = AllCategoryForm(request.form)
id = request.form.get('id','0')
category = Category.query.get_or_404(id)
print(form)
if form.validate_on_submit():
category.update(
title = form.title.data,
summary = form.summary.data,
key = form.key.data,
word = form.word.data,
sort = form.sort.data,
main_sort = form.main_sort.data,
is_enable = form.is_enable.data,
attr_id = form.attr.data,
)
flash(u'更新成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_category'))
def add_users():
"""Register new user.
获取产品
添加用户
添加到推荐人列表
添加购买列表
"""
form = RegisterForm(request.form)
if form.validate_on_submit():
if current_user.is_admin or current_user.is_center:
tuijianren = User.query.filter_by(
phone=form.tuijianren.data).first()
if not tuijianren:
flash('已注册失败.推荐人不存在，请填写正确的手机号。', 'danger')
return dict(form=form)
baodan = User.query.filter_by(phone=form.baodan.data).first()
if not tuijianren:
flash('已注册失败.报单中心不存在，请填写正确的手机号。', 'danger')
return dict(form=form)
product = Products.query.get_or_404(int(form.products.data))
user = User.create(username=form.user.data,
password=form.password.data,
phone=form.user.data,
active=False,
parent_center=baodan.id)
ProductsBuys.create(
user=user,
product=product,
)
Recommend.create(recommender_id=tuijianren.id,
recommends_id=user.id)
flash('已注册成功.', 'success')
else:
flash('已注册失败.您不是报单中心。', 'danger')
return redirect(url_for('public.home'))
else:
flash_errors(form)
return dict(form=form)
def all_category_post():
form = AllCategoryForm(request.form)
if form.validate_on_submit():
if form.pid.data == -1:
form.pid.data = 1
category = Category.create(
title = form.title.data,
summary = form.summary.data,
key = form.key.data,
word = form.word.data,
sort = form.sort.data,
main_sort = form.main_sort.data,
is_enable = form.is_enable.data,
user = current_user.id,
attr_id = form.attr.data,
parent_id = form.pid.data,
)
templates_default = TemplatesDefault.query.filter_by(user_template_category=form.attr.data).all()
fake_static_context = []
#创建栏目默认内容
for i in templates_default:
fake_static_context.append(
StaticContext(
html = i.html,
default_html = i.html,
main_sort = i.main_sort,
users = current_user,
categorys = category
)
)
db.session.add_all(fake_static_context)
db.session.commit()
flash(u'添加成功！','success')
else:
flash_errors(form)
return redirect(url_for('.all_category'))
def register_user_post():
form = RegisterForm(request.form)
if form.validate_on_submit():
userurl = UserUrlAndTemplate.query.filter_by(user_url=request.url_root).first()
user = User.create(
username = form.username.data,
password = form.password.data,
first_name = form.first_name.data,
phone = form.phone.data,
memory_capacity = userurl.usertemplate.memory_capacity,
usertemplate = userurl.usertemplate,
)
userurl.update(users=user)
flash(u'保存设置成功！','success')
else:
flash_errors(form)
return render_template('users/adduser.html',form=form)
return render_template('users/register_success.html',index=request.url_root)
