refactor(移除生产部署文件)

This commit is contained in:
zhengxinonly
2022-01-23 01:09:35 +08:00
parent 6978002393
commit c0d963628b
15 changed files with 20 additions and 177 deletions
+12
View File
@@ -0,0 +1,12 @@
from flask_restful import Api
from .file import FilePhotosResource, FilePhotoResource
from .passport import LoginResource
def register_system_api(api_bp):
sys_api = Api(api_bp)
sys_api.add_resource(FilePhotosResource, '/file/photos')
sys_api.add_resource(FilePhotoResource, '/file/photo/<int:photo_id>')
sys_api.add_resource(LoginResource, '/passport/login')
+77
View File
@@ -0,0 +1,77 @@
import os
from flask import request, jsonify, current_app
from flask_restful import Resource
from sqlalchemy import desc
from common.utils.http import fail_api, success_api, table_api
from common.utils.upload import upload_one, delete_photo_by_id
from extensions import db
from models import PhotoModels
class FilePhotosResource(Resource):
def get(self):
page = request.args.get('page', type=int)
limit = request.args.get('limit', type=int)
photo_paginate = PhotoModels.query.order_by(desc(PhotoModels.create_at)
).paginate(page=page,
per_page=limit,
error_out=False)
data = [
{
'id': item.id,
'name': item.name,
'href': item.href,
'mime': item.mime,
'size': item.size,
'ext': item.ext if hasattr(item, 'ext') else "",
'create_at': str(item.create_at),
} for item in photo_paginate.items
]
return table_api(result={'items': data,
'total': photo_paginate.total, },
code=0)
def post(self):
if 'file' in request.files:
photo = request.files['file']
mime = request.files['file'].content_type
file_url = upload_one(photo=photo, mime=mime)
res = {
"message": "上传成功",
"code": 0,
"success": True,
"data":
{"src": file_url}
}
return jsonify(res)
return fail_api()
def delete(self):
"""图片批量删除"""
# TODO bugs 图片删除失败
ids = request.form.getlist('ids[]')
photo_name = PhotoModels.query.filter(PhotoModels.id.in_(ids)).all()
upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
for p in photo_name:
os.remove(upload_url + '/' + p.name)
photo = PhotoModels.query.filter(PhotoModels.id.in_(ids)).delete(synchronize_session=False)
db.session.commit()
if photo:
return success_api(message="删除成功")
else:
return fail_api(message="删除失败")
class FilePhotoResource(Resource):
"""图片数据"""
def delete(self, photo_id):
res = delete_photo_by_id(photo_id)
if res:
return success_api(message="删除成功")
else:
return fail_api(message="删除失败")
+49
View File
@@ -0,0 +1,49 @@
from flask import render_template, make_response
from flask import session, redirect, url_for
from flask_login import current_user, login_user
from flask_restful import Resource, reqparse
from common.gen_captcha import add_auth_session
from common.utils.http import fail_api, success_api
from common.utils.rights import record_logging
from models import UserModels
class LoginResource(Resource):
def get(self):
if current_user.is_authenticated:
return redirect(url_for('admin.index'))
return make_response(render_template('index/login.html'))
def post(self):
login_req = reqparse.RequestParser(bundle_errors=True)
login_req.add_argument('username', type=str, help='请输入用户名', required=True)
login_req.add_argument('password', type=str, help='请输入密码', required=True)
login_req.add_argument('captcha', type=str, help='请输入验证码', required=True)
req = login_req.parse_args()
s_code = session.get("code", None)
session["code"] = None
if req.captcha != s_code:
return fail_api(message="验证码错误")
user = UserModels.query.filter_by(username=req.username).first()
if user is None:
return fail_api(message="不存在的用户")
if user.enable == 0:
return fail_api(message="用户被暂停使用")
if user.validate_password(req.password):
# 登录
login_user(user)
# 记录登录日志
record_logging()
# 存入权限
add_auth_session()
return success_api(message="登录成功")
record_logging()
return fail_api(message="用户名或密码错误")