yumoqing/ahserver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/yumoqing/ahserver

Browse Source
This commit is contained in:
yumoqing 2020-12-19 14:00:27 +08:00
commit 072cad2343
5 changed files with 53 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ahserver/baseProcessor.py
View File

@ -11,12 +11,9 @@ from appPublic.jsonConfig import getConfig
from appPublic.dictObject import DictObject from appPublic.dictObject import DictObject
from appPublic.folderUtils import listFile from appPublic.folderUtils import listFile
from .utils import unicode_escape
from .serverenv import ServerEnv from .serverenv import ServerEnv
def unicode_escape(s):
x = [ch if ord(ch) < 256 else ch.encode('unicode_escape').decode('utf-8') for ch in s]
return ''.join(x)
class ObjectCache: class ObjectCache:
def __init__(self): def __init__(self):
self.cache = {} self.cache = {}

4
ahserver/filedownload.py
View File

@ -18,9 +18,11 @@ def path_decode(dpath):
rc4 = RC4() rc4 = RC4()
return rc4.decode(dpath,crypto_aim) return rc4.decode(dpath,crypto_aim)
async def file_download(request, filepath): async def file_download(request, filepath, content_type=None):
filename = os.path.basename(filepath) filename = os.path.basename(filepath)
r = web.FileResponse(filepath) r = web.FileResponse(filepath)
ct = content_type
if ct is None:
ct, encoding = mimetypes.guess_type(filepath) ct, encoding = mimetypes.guess_type(filepath)
if ct is not None: if ct is not None:
r.content_type = ct r.content_type = ct

33
ahserver/processorResource.py
View File

@ -1,5 +1,7 @@
import os import os
import re import re
import codecs
from traceback import print_exc
import asyncio import asyncio
@ -40,6 +42,7 @@ from .filestorage import FileStorage
from .restful import DBCrud from .restful import DBCrud
from .dbadmin import DBAdmin from .dbadmin import DBAdmin
from .filedownload import file_download, path_decode from .filedownload import file_download, path_decode
from .utils import unicode_escape
def getHeaderLang(request): def getHeaderLang(request):
al = request.headers.get('Accept-Language') al = request.headers.get('Accept-Language')
@ -227,6 +230,11 @@ class ProcessorResource(StaticResource,Url2File):
if processor: if processor:
return await processor.handle(request) return await processor.handle(request)
filepath = self.url2file(str(request.url))
print('filepath=',filepath,str(request.url))
if filepath and self.isHtml(filepath):
return await self.html_handle(request, filepath)
print(f'path={path} handler by StaticResource..') print(f'path={path} handler by StaticResource..')
if self.isFolder(path): if self.isFolder(path):
config = getConfig() config = getConfig()
@ -234,6 +242,31 @@ class ProcessorResource(StaticResource,Url2File):
raise HTTPNotFound raise HTTPNotFound
return await super()._handle(request) return await super()._handle(request)
async def html_handle(self,request,filepath):
with codecs.open(filepath,'r', 'utf-8') as f:
b = f.read()
b = unicode_escape(b)
headers = {
'Content-Type': 'text/html; utf-8',
'Accept-Ranges': 'bytes',
'Content-Length': str(len(b))
}
resp = Response(text=b,headers=headers)
return resp
def isHtml(self,fn):
try:
with codecs.open(fn,'r','utf-8') as f:
b = f.read()
if b.startswith('<html>'):
return True
if b.startswith('<!doctype html>'):
return True
except Exception as e:
print_exc()
print(e)
return False
def url2processor(self, request, url): def url2processor(self, request, url):
config = getConfig() config = getConfig()
url = self.entireUrl(request, url) url = self.entireUrl(request, url)

15
ahserver/url2file.py
View File

@ -51,14 +51,17 @@ class Url2File:
if url[-1] == '/': if url[-1] == '/':
url = url[:-1] url = url[:-1]
if self.isFolder(url): paths = url.split('/')[3:]
return self.defaultIndex(url) f = os.path.join(self.path,*paths)
if url.startswith(self.starts):
f = self.path + url[len(self.starts):]
real_path = os.path.abspath(f) real_path = os.path.abspath(f)
if os.path.isdir(real_path):
for idx in self.indexes:
p = os.path.join(real_path,idx)
if os.path.isfile(p):
return p
if os.path.isfile(real_path): if os.path.isfile(real_path):
return f return real_path
if not self.inherit: if not self.inherit:
return None return None

4
ahserver/utils.py Normal file
View File

@ -0,0 +1,4 @@
def unicode_escape(s):
x = [ch if ord(ch) < 256 else ch.encode('unicode_escape').decode('utf-8') for ch in s]
return ''.join(x)