def _save(self, file, path, filename):
buf = file.stream.read()
md5_key = md5(buf)
file_map = fetchone_dict(SelectMap.file_map, (md5_key, ),
GeneralObject)
if file_map:
return file_map.filename
file.save(path)
with open(path, "wb") as new_f:
new_f.write(buf)
execute_sql(InsertMap.file_map, (filename, md5_key), True)
return filename
def post(self):
comment = CommentTemplate()
response = post(CommentValid, parse, comment)
if response:
return jsonify(response.dict_data)
try:
response = Response()
now = datetime.datetime.now()
user_id = getattr(request, "user")["id"]
user = fetchone_dict(SelectMap.user_info_by_user_id, (user_id, ),
GeneralObject)
ret = execute_sql(InsertMap.comment,
(comment.content, now, comment.blog_id, user_id,
user.nick_name), True)
if ret == 0:
raise InvalidArgumentException("评论失败")
response.data = {
"id": ret,
"create_time": now,
"content": comment.content,
"blog_id": comment.blog_id,
"user_id": user_id,
"nick_name": user.nick_name
}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def get(self):
"""
获取一个特定的挑战的接口 根据course id 请求需要携带一个参数 id 由于挑战 不用登陆也可以访问
所以不加权，此pageviews每请求一次就会增加一次 在多线程环境下会不准确 但是没关系 问题不大
:return:
"""
response = Response()
try:
_id = request.args["id"]
challenge = fetchone_dict(SelectMap.challenge_by_id, [_id, ], ChallengeTemplate)
if challenge:
ret = execute_sql(UpdateMap.update_challenge_pageviews, [challenge.id, ])
if ret == 0:
raise InvalidArgumentException("数据不存在")
response.data = {
"id": challenge.id, "picture": challenge.picture,
"content": challenge.content, "start_time": challenge.start_time,
"end_time": challenge.end_time, "create_time": challenge.create_time,
"pageviews": challenge.pageviews + 1
}
return jsonify(response.dict_data)
raise InvalidArgumentException("数据不存在")
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def delete(self):
upper = UpperTemplate()
response = post(UpperValid, parse, upper)
if response:
return jsonify(response)
try:
user_id = getattr(request, "user")["id"]
response = Response()
ret = execute_sql(DeleteMap.upper_by_id, (upper.blog_id, user_id))
if ret == 0:
raise InvalidArgumentException("取消失败")
execute_sql(UpdateMap.blog_upper_dev, (upper.blog_id, ))
response.data = {"msg": "ok"}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def post(self):
"""
添加挑战的接口 此接口的content是一个重点 是有格式要求的 要求就是 其中的html不能包含script标签
也不能含有任何的注释
添加挑战的另一个约定就是 start_time和end_time是有格式要求的 必须是2019-11-03这种格式 无法配置
:return:
"""
challenge = ChallengeTemplate()
response = post(ChallengeValid, parse, challenge)
if response is None:
response = Response()
try:
number = generate_number(20)
ret = execute_sql(InsertMap.challenge, [
challenge.picture, challenge.content, challenge.start_time,
challenge.end_time, datetime.datetime.now(), challenge.pageviews,
number
])
if ret == 0:
raise InvalidArgumentException("数据写入失败")
challenge = fetchone_dict(SelectMap.challenge_by_number, [number, ], ChallengeTemplate)
response.data = {
"id": challenge.id, "picture": challenge.picture,
"content": challenge.content, "start_time": challenge.start_time,
"end_time": challenge.end_time, "create_time": challenge.create_time,
"pageviews": challenge.pageviews
}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def post(self):
"""
添加帖子的接口 帖子的添加需要携带 user_id content title picture 等键值对
图片需要先上传 content需要不存在注释以及script标签
:return:
"""
response = Response()
blog = BlogTemplate()
resp = post(BlogValid, parser, blog)
if resp is not None:
return resp
try:
from datetime import datetime
now = datetime.now()
row_id = execute_sql(
InsertMap.blog,
(blog.user_id, blog.content, blog.title, blog.picture, now))
if row_id == 0:
raise InvalidArgumentException("插入记录失败")
response.data = {
"id": row_id,
"user_id": blog.user_id,
"content": blog.content,
"title": blog.title,
"picture": blog.picture,
"create_time": now
}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def delete(self):
response = Response()
try:
_id = request.json["id"]
comment_count = execute_sql(DeleteMap.comment_by_id, (_id, ))
if comment_count == 0:
raise InvalidArgumentException("删除失败!")
reply_count = execute_sql(DeleteMap.reply_by_comment_id, (_id, ))
response.data = {
"comment_count": comment_count,
"reply_count": reply_count,
"msg": "ok"
}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def put(self):
"""
修改课程的接口
:return:
"""
response = Response()
try:
_id = request.json["id"]
course = CourseTemplate()
response = post(CourseValid, parse, course)
if response:
return jsonify(response.dict_data)
ret = execute_sql(UpdateMap.update_course_by_id, [
course.name, course.type, course.level, course.burning,
course.id
])
if ret == 0:
raise UserDoesNotExistException("课程不存在")
response.data = {
"id": course.id, "name": course.name, "type": course.type,
"level": course.level, "burning": course.burning
}
except KeyError as e:
init_key_error_handler(response, e, "缺少入参:")
except UserDoesNotExistException as e:
init_key_error_handler(response, e)
return jsonify(response.dict_data)
def delete(self):
response = Response()
try:
_id = request.json["id"]
ret = execute_sql(DeleteMap.challenge_by_id, (_id, ))
if ret == 0:
raise UserDoesNotExistException("挑战不存在")
except Exception as e:
init_key_error_handler(response, e, "信息")
return jsonify(response.dict_data)
def post(self):
insert_upper = UpperTemplate()
response = post(UpperValid, parse, insert_upper)
if response:
return jsonify(response.dict_data)
try:
user_id = getattr(request, "user")["id"]
old_upper = fetchone_dict(SelectMap.upper_without_delete_flag,
(insert_upper.blog_id, user_id),
UpperTemplate)
if old_upper is not None:
response = self._rollback(old_upper)
else:
response = self._real_insert(insert_upper)
execute_sql(UpdateMap.update_blog_upper_by_id,
(insert_upper.blog_id, ))
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def delete(self):
"""
删除课程接口 但是并不会删除action
:return:
"""
response = Response()
try:
_id = request.json["id"]
course_num = execute_sql(DeleteMap.course_by_id, [_id, ])
response.data = {"msg": "ok", "count": course_num}
except KeyError as e:
init_key_error_handler(response, e)
return jsonify(response.dict_data)
def post(self):
valid = CourseValid(parse.parse_args())
err_map = valid.valid_data()
response = Response()
if err_map:
response.data = err_map
response.errno = len(err_map)
response.code = FORMAT_ERROR
return jsonify(response.dict_data)
try:
data = valid.clean_data
execute_sql(InsertMap.course, [
data["type"], data["name"], datetime.datetime.now(),
data["level"], data["burning"]
])
ret = execute_query_sql(SelectMap.course_by_create, [data["name"], ], lambda c: c.fetchone())
response.data = {
"id": ret[0], "type": ret[1], "name": ret[2], "create_time": ret[3],
"level": ret[4], "burning": ret[5]
}
except Exception as e:
init_key_error_handler(response, e, "插入数据失败:")
return jsonify(response.dict_data)
def _rollback(self, upper):
response = Response()
user_id = getattr(request, "user")["id"]
ret = execute_sql(UpdateMap.update_upper_by_user_and_blog,
(upper.blog_id, user_id))
if ret == 0:
response.code = FORMAT_ERROR
response.errno = 1
response.data = {"msg": "点赞失败"}
return response
response.data = {
"user_id": upper.user_id,
"blog_id": upper.blog_id,
"create_time": upper.create_time
}
return response
def delete(self):
response = Response()
try:
_id = request.json["id"]
b = fetchone_dict(SelectMap.blog_by_id, (_id, ), GeneralObject)
permission = getattr(request, "user")["permission"]
if getattr(
request,
"user")["id"] != b.user_id and permission & ADMIN != ADMIN:
raise InvalidArgumentException("权限不足")
ret = execute_sql(DeleteMap.blog_by_id, (_id, ))
if ret == 0:
raise InvalidArgumentException("删除失败")
response.data = {"msg": "ok"}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
def _real_insert(self, upper):
response = Response()
now = datetime.datetime.now()
user_id = getattr(request, "user")["id"]
ret = execute_sql(InsertMap.upper, (upper.blog_id, user_id, now), True)
if ret == 0:
response.code = FORMAT_ERROR
response.errno = 1
response.data = {"msg": "点赞失败"}
return response
response.data = {
"id": ret,
"user_id": upper.user_id,
"blog_id": upper.blog_id,
"create_time": now
}
return response
def post(self):
"""
此方法用来添加一个对应课程的动作 但是此方法不会校验sequence(动作编号)，所以请谨慎调用
:return:
"""
template = ActionTemplate()
ret = post(ActionValid, parse, template)
response = Response()
if ret:
return jsonify(ret.dict_data)
try:
count = execute_sql(InsertMap.action, [
template.id, template.content, template.picture, template.sequence
])
response.data = {
"course_id": template.id, "content": template.content,
"picture": template.picture, "sequence": template.sequence,
"count": count
}
except Exception as e:
init_key_error_handler(response, e)
return jsonify(response.dict_data)
def put(self):
"""
修改挑战的接口 和post接口相比 需要多传一个id的key 如果id不存在 则无法执行
:return:
"""
challenge = ChallengeTemplate()
response = post(ChallengeValid, parse, template=challenge)
if response is None:
response = Response()
try:
_id = challenge.id
ret = execute_sql(UpdateMap.update_challenge_by_id, [
challenge.picture, challenge.content, challenge.start_time,
challenge.end_time, _id
])
if ret == 0:
raise InvalidArgumentException("数据不存在")
except InvalidArgumentException as e:
response.errno = 1
response.code = 403
response.data = {"msg": str(e)}
except Exception as e:
init_key_error_handler(response, e, "信息:")
return jsonify(response.dict_data)
