ComiPy/web/page.py
2024-04-15 15:19:56 +08:00

76 lines
2.2 KiB
Python

from flask import *
from flask import Blueprint
import configparser
import db.file , file
page_bp = Blueprint("page_bp", __name__)
config = configparser.ConfigParser()
config.read("./conf/app.ini")
@page_bp.route("/overview/<page>")
def overview(page): # 概览
page = int(page)
if request.cookies.get("islogin") is None:
return redirect("/")
metaDataList = db.file.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,
)
@page_bp.route("/book/<bookid>")
def book(bookid): # 接口
if request.cookies.get("islogin") is None:
return abort(403)
data = db.file.searchByid(bookid)
if data == "":
return abort(404)
return render_template("view.html", id=bookid, index=range(1, data[0][3]))
@page_bp.route("/view/<bookid>")
def view(bookid): # 接口
if request.cookies.get("islogin") is None:
return abort(403)
return bookid
@page_bp.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("/")
@page_bp.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("/")