def collect(token):
if request.method == 'POST':
dramaid = request.form.get('dramaid')
if Drama.query.get(dramaid) is None:
return jsonify(Event1002())
if Collectdrama.query.filter_by(user_id=r.get(token),
drama_id=dramaid).first() is not None:
return jsonify(Event1003())
collect = Collectdrama(user_id=r.get(token), drama_id=dramaid)
db.session.add(collect)
db.session.commit()
if Collectdrama.query.get(collect.id) is not None:
return jsonify(Event0(token=token))
return jsonify(Event1005('创建失败'))
elif request.method == 'DELETE':
dramaid = request.form.get('dramaid')
collect = Collectdrama.query.filter_by(user_id=r.get(token),
drama_id=dramaid).first()
if collect is None:
return jsonify(Event1002())
db.session.delete(collect)
db.session.commit()
if Collectdrama.query.get(collect.id) is not None:
return jsonify(Event1005('删除失败'))
print('aaa')
return jsonify(Event0(token=token))
return jsonify(Event1004())
def create():
if request.method == 'POST':
name = request.form.get('name')
password = request.form.get('password')
repeatpd = request.form.get('repeatpd')
avatar = request.files.get('avatar')
email = request.form.get('email')
if User.query.filter_by(name=name).first() is not None: #用户名存在，返回错误
return jsonify(Event1003())
if User.query.filter_by(email=email).first() is not None: #邮箱存在，返回错误
return jsonify(Event1005('邮箱已被注册'))
if password == repeatpd:
user = User(name=name, password=password, email=email)
db.session.add(user)
db.session.commit()
if avatar and avatar.filename != '':
avatar.filename = random_filename(avatar.filename)
avatar.save(
os.path.join(current_app.config['UPLOAD_PATH'],
avatar.filename))
user.avatar = 'http://127.0.0.1:5000/user/image/' + avatar.filename
db.session.commit()
send_email('注册成功', user.email, '注册成功欢迎加入我们')
token = user.make_token()
r.set(token, str(user.id), ex=3600)
return jsonify(Event0(token=token))
return jsonify(Event1005('密码不一致'))
return jsonify(Event1004())
def star(token):
if request.method == 'POST':
dramaid = request.form.get('dramaid')
if Likedrama.query.filter_by(user_id=r.get(token),drama_id=dramaid).first() is not None:
return jsonify(Event1003())
if Drama.query.filter(Drama.id==dramaid,Drama.solution==None).first():
star = Likedrama(user_id=r.get(token), drama_id=dramaid)
db.session.add(star)
db.session.commit()
if Likedrama.query.get(star.id) is not None:
return jsonify(Event0(token=token))
return jsonify(Event1005('创建失败'))
elif Drama.query.filter(Drama.id==dramaid,Drama.solution!=None).first():
star = Likedrama(user_id=r.get(token), drama_id=dramaid,follow=True)
db.session.add(star)
db.session.commit()
if Likedrama.query.get(star.id) is not None:
return jsonify(Event0(token=token))
return jsonify(Event1005('创建失败'))
else:
return jsonify(Event1002())
elif request.method == 'DELETE':
dramaid = request.form.get('dramaid')
star = Likedrama.query.filter_by(user_id=r.get(token),drama_id=dramaid).first()
if star is None:
return jsonify(Event1002())
db.session.delete(star)
db.session.commit()
if Likedrama.query.get(star.id) is not None:
return jsonify(Event1005('删除失败'))
return jsonify(Event0(token=token))
def updata(token):
name = request.form.get('name')
avatar = request.files.get('avatar')
password = request.form.get('password')
email = request.form.get('email')
user = User.query.filter_by(id=r.get(token)).first()
if name is not None:
if User.query.filter_by(name=name).first() is not None: #用户名存在，返回错误
return jsonify(Event1003())
user.name = name
if avatar is not None:
avatar.filename = random_filename('user' + str(user.id) + os.path.splitext(avatar.filename)[1])
avatar.save(os.path.join(current_app.config['UPLOAD_PATH'], avatar.filename))
user.avatar = 'http://101.200.157.252/user/image/' + avatar.filename
if password is not None:
user.password = password
if email is not None:
if User.query.filter_by(email=email).first() is not None: #邮箱存在，返回错误
return jsonify(Event1005('邮箱已被注册'))
user.email = email
db.session.commit()
return jsonify(Event0(token=token))