增加403,404页面; 支持前端history路由

master
yutent 2023-09-07 11:39:56 +08:00
parent d499991831
commit ed5dcf1245
2 changed files with 42 additions and 5 deletions

View File

@ -2,7 +2,9 @@
import os import os
from gi.repository import Gio, WebKit2 from gi.repository import Gio, WebKit2
from ._mimetypes import get_mimetype from ._mimetypes import get_mimetype
from ._version import version
__dir__ = os.path.dirname(os.path.realpath(__file__))
class Protocal: class Protocal:
root = '' root = ''
@ -22,29 +24,61 @@ class Protocal:
return os.path.join(self.root, filepath) return os.path.join(self.root, filepath)
def _get_error_page(self, tips = '404 not found!'):
data = f"""
<!doctype html>
<html lang="zh-CN">
<head>
<meta charset="utf-8">
<title>Page not found</title>
<style>body {{text-align: center;-webkit-user-select:none}} cite {{font-size:12px}}</style>
</head>
<body>
<h1>Oops!</h1>
<h2>{tips}</h2>
<hr>
<cite>WebEngine v{version}</cite>
</body>
</html>
"""
return data
def handle_response(self, req): def handle_response(self, req):
schema = req.get_scheme() schema = req.get_scheme()
pathname = req.get_path()[1:] pathname = req.get_path()[1:]
ext = pathname.split('.')[-1] ext = pathname.split('.')[-1]
mimetype = get_mimetype(ext)
if pathname == ext:
pathname = 'index.html'
# print('----------------------------------------') # print('----------------------------------------')
# print(req.get_uri(),schema, pathname, ext, get_mimetype(ext)) # print(req.get_uri(),schema, pathname, ext, mimetype)
# print('----------------------------------------') # print('----------------------------------------')
if schema == self.protocal: if schema == self.protocal:
data = open(self.abspath(pathname)).read() filepath = self.abspath(pathname)
if os.path.isfile(filepath):
try:
with open(filepath) as f:
data = f.read()
except Exception:
data = self._get_error_page('403 Forbidden!')
else:
data = self._get_error_page()
data = Gio.MemoryInputStream.new_from_data(data.encode()) data = Gio.MemoryInputStream.new_from_data(data.encode())
# ------- 更多功能的reponse ---------------- # ------- 更多功能的reponse ----------------
# res = WebKit2.URISchemeResponse.new(data, -1) # res = WebKit2.URISchemeResponse.new(data, -1)
# res.set_content_type(get_mimetype(ext)) # res.set_content_type(mimetype)
# res.set_http_headers('text/html') # res.set_http_headers('text/html')
# res.set_status(200) # res.set_status(200)
# req.finish_with_response(res) # req.finish_with_response(res)
# ---------------------------------------- # ----------------------------------------
# 简单的response # 简单的response
req.finish(data, -1, get_mimetype(ext)) req.finish(data, -1, mimetype)
def create_protocal(root): def create_protocal(root):

View File

@ -131,7 +131,10 @@ class WebEngine(WebKit2.WebView):
if self.root is None: if self.root is None:
raise EnvironmentError('web root dir not set!') raise EnvironmentError('web root dir not set!')
else: else:
self.load_uri(f"app://{url}") if url.startswith('/'):
self.load_uri(f"app://{url}")
else:
raise ValueError('url must starts with "/"')