mirror of
https://github.com/Kakune55/ComiPy.git
synced 2025-05-06 18:29:26 +08:00
121 lines
3.4 KiB
Python
121 lines
3.4 KiB
Python
import configparser, gc
|
|
import db, file
|
|
from flask import *
|
|
|
|
|
|
config = configparser.ConfigParser()
|
|
config.read("./conf/app.ini")
|
|
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
def appinit():
|
|
file.init()
|
|
db.init()
|
|
file.auotLoadFile()
|
|
|
|
|
|
@app.route("/", methods=["GET", "POST"])
|
|
def login(): # 登录页面
|
|
if request.method == "GET":
|
|
if request.cookies.get("islogin") is not None:
|
|
return redirect("/overview/1")
|
|
return render_template("login.html")
|
|
elif request.method == "POST":
|
|
if request.form["username"] == config.get("user", "username") and request.form[
|
|
"password"
|
|
] == config.get("user", "password"):
|
|
resp = make_response(redirect("/overview/1"))
|
|
resp.set_cookie("islogin", "True")
|
|
return resp
|
|
else:
|
|
return redirect("/")
|
|
|
|
|
|
|
|
@app.route("/overview/<page>")
|
|
def overview(page): # 概览
|
|
page = int(page)
|
|
if request.cookies.get("islogin") is None:
|
|
return redirect("/")
|
|
metaDataList = db.getMetadata((page - 1) * 20, page * 20)
|
|
if page <= 3:
|
|
lastPageList = range(1,page)
|
|
else:
|
|
lastPageList = range(page-3,page)
|
|
nextPageList = range(page+1,page+4)
|
|
return render_template("overview.html",list=metaDataList,lastPageList=lastPageList,pagenow=page,nextPageList=nextPageList)
|
|
|
|
|
|
# @app.route("/api/info") 暂时弃用
|
|
# def api(): # 接口
|
|
# if request.cookies.get("islogin") is None:
|
|
# return abort(403)
|
|
# func = request.args.get("func")
|
|
# if func == "bookname" and request.args.get("page") is not None:
|
|
# page = int(request.args.get("page"))
|
|
# return db.getMetadata((page - 1) * 20, page * 20)
|
|
|
|
# return abort(400)
|
|
|
|
|
|
@app.route("/api/img/<bookid>/<index>")
|
|
def img(bookid, index): # 图片接口
|
|
if request.cookies.get("islogin") is None:
|
|
return abort(403)
|
|
if db.searchByid(bookid) == "":
|
|
return abort(404)
|
|
# 设置响应类型为图片
|
|
data, filename = file.raedZip(bookid,index)
|
|
if isinstance(data, str):
|
|
abort(404)
|
|
if request.args.get("mini") == "yes":
|
|
data = file.thumbnail(data)
|
|
else:
|
|
data = file.imageToWebP(data)
|
|
response = make_response(data) #读取文件
|
|
del data
|
|
response.headers.set('Content-Type', 'image/Webp')
|
|
response.headers.set('Content-Disposition', 'inline', filename=filename)
|
|
gc.collect()
|
|
return response
|
|
|
|
|
|
@app.route("/book/<bookid>")
|
|
def book(bookid): # 接口
|
|
if request.cookies.get("islogin") is None:
|
|
return abort(403)
|
|
data = db.searchByid(bookid)
|
|
if data == "":
|
|
return abort(404)
|
|
return render_template("view.html",id = bookid , index = range(1,data[0][3]))
|
|
|
|
|
|
@app.route("/view/<bookid>")
|
|
def view(bookid): # 接口
|
|
if request.cookies.get("islogin") is None:
|
|
return abort(403)
|
|
return bookid
|
|
|
|
|
|
@app.route('/upload', methods=["GET", "POST"]) #文件上传
|
|
def upload_file():
|
|
if request.method == "GET":
|
|
return render_template("upload.html")
|
|
uploaded_file = request.files.getlist('files[]') # 获取上传的文件列表
|
|
for fileitem in uploaded_file:
|
|
if fileitem.filename != '':
|
|
fileitem.save(config.get("file", "inputdir") + "/" + fileitem.filename)
|
|
file.auotLoadFile()
|
|
return redirect("/")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
appinit()
|
|
app.run(
|
|
debug=config.get("server", "debug"),
|
|
host=config.get("server", "host"),
|
|
port=config.get("server", "port"),
|
|
)
|