def init_app(app):
Config.init_app(app)
with app.app_context():
from app.db import db
from app.models import User
from app.models import Plan
db.init_app(app)
db.create_all()
# check if plans exist
if len(Plan.query.all()) != 4:
p1 = Plan(id=1, name='free', description='All the basic features of SupportService', cost=0)
db.session.add(p1)
p2 = Plan(id=2, name='bronze', description='Everything in free and email support.', cost=25)
db.session.add(p2)
p3 = Plan(id=3, name='silver', description='Everything in bronze and chat support.', cost=50)
db.session.add(p3)
p4 = Plan(id=4, name='gold', description='Everything in silver and 99.999% uptime SLA!', cost=50)
db.session.add(p4)
db.session.commit()
# check if user exists
if User.query.filter_by(email='*****@*****.**').first() is None:
app.logger.info("Creating test user: [email protected] password: test")
u = User(email='*****@*****.**')
u.set_password('test')
db.session.add(u)
db.session.commit()
else:
app.logger.info("You can login with user: [email protected] password: test")
def create_plan():
print(request.json)
name = request.json.get('name')
week = request.json.get('week')
current = request.json.get('current')
follow = request.json.get('follow')
project_id = request.json.get('project_id')
project = Project.query.filter_by(id=project_id).first()
if project is None:
return jsonify({'success':False,'error_code':-1,'errmsg':'project_id不存在'})
user = g.current_user
plan = Plan(name=name,week=week,current=current,follow=follow,project_id=project.id,user_id=user.id,)
db.session.add(plan)
try:
db.session.commit()
except Exception as e:
db.session.rollback()
logging.error(f'添加数据库发生错误,已经回退:{e}')
return jsonify({'success': False, 'error_code': -123, 'errmsg': '数据库插入错误，请查看日志'})
return jsonify({'success':True,
'error_code':0,
})
def updatePlanControler(self, request):
if request.method == 'PUT':
status = 200
[sessionid, response] = self.ulity.isEmptySession(request.session)
if sessionid == None:
response["message"] = Constant.FAIL_UPDATEPLAN
return HttpResponse(content=simplejson.dumps(response),
status=status)
is_login = self.uservice.isLogin(sessionid)
req = simplejson.loads(request.body)
if is_login == Constant.ERROR_LOGIN_NOLOGIN:
response['message'] = Constant.FAIL_UPDATEPLAN
response['data'] = {}
response['data']['error'] = is_login
else:
partner = req["partner"]
partner_str = ','.join(partner)
plan = Plan(PLAN_ID=req["planid"],
PLAN_USER=is_login,
PLAN_NAME=req["planname"],
RUN_DATETIME=datetime.datetime.strptime(
req["runtime"], '%Y-%m-%d %H:%M'),
PARTNER=partner_str,
PLACE=req["place"])
update_plan_message = self.planservice.updatePlan(plan)
if update_plan_message != Constant.SUCCESS_UPDATEPLAN:
response["message"] = Constant.FAIL_UPDATEPLAN
response["data"] = {}
response["data"]["error"] = update_plan_message
else:
response["message"] = Constant.SUCCESS_UPDATEPLAN
response["planid"] = plan.PLAN_ID
status = 200
return HttpResponse(content=simplejson.dumps(response),
status=status)
def populate():
""" Create DB Entries for Userland Cloud Plans"""
for plan_name, plan in LIMITS.items():
p = Plan(**{"name": plan_name}, **plan)
db.session.add(p)
db.session.commit()
def populate():
""" Create DB Entries for Holepunch Plans"""
global LIMITS
for plan_name, plan in LIMITS.items():
p = Plan(**{"name": plan_name}, **plan)
db.session.add(p)
db.session.commit()
def test_ld_user_hash(self):
p = Plan(id=1, name='free')
db.session.add(p)
db.session.commit()
u = User(email='*****@*****.**', plan_id=1)
db.session.add(u)
db.session.commit()
emailHash = u.get_email_hash()
userEmailHash = u.get_ld_user()['key']
self.assertEqual(userEmailHash, emailHash)
def add_plan():
if current_user.email != '*****@*****.**':
abort(403)
form = InsertPlanForm()
if form.validate_on_submit():
plan = Plan(name=form.name.data, req_short=form.req_short.data, req_full=form.req_full.data,
benefits_short=form.benefits_short.data, benefits_full=form.benefits_full.data,
application=form.application.data, website=form.website.data, kw1=form.kw1.data, kw2=form.kw2.data,
kw3=form.kw3.data, kw4=form.kw4.data, kw5=form.kw5.data)
db.session.add(plan)
db.session.commit()
flash('Plan added successfully.', 'success')
return redirect(url_for('add_plan'))
else:
return render_template('insertplan.html', title='Insert Plan', form=form)
def addplan():
form = InsertPlanForm()
if form.validate_on_submit():
plan = Plan(name=form.name.data,
req_short=form.req_short.data,
req_full=form.req_full.data,
benefits_short=form.benefits_short.data,
benefits_full=form.benefits_full.data,
application=form.application.data,
website=form.website.data,
kw1=form.kw1.data,
kw2=form.kw2.data,
kw3=form.kw3.data,
kw4=form.kw4.data,
kw5=form.kw5.data,
kw6=form.kw6.data)
db.session.add(plan)
db.session.commit()
flash('Plan added successfully.')
return redirect(url_for('addplan'))
else:
return render_template('insertplan.html', form=form)
# -*- coding: utf-8 -*-
from app.service.planservice import PlanService
from app.models import Plan
import datetime
if __name__ == "__main__":
planservice = PlanService()
str = '2018-01-01 10:10'
date_time = datetime.datetime.strptime(str, '%Y-%m-%d %H:%M')
plan = Plan(PLAN_USER='******',
PLAN_NAME='跑步',
RUN_DATETIME=date_time,
PLACE='假草',
PARTNER='李四')
print plan.PLAN_ID
