ahserver_g/ahserver/url2file.py
2020-12-18 13:46:35 +08:00

114 lines
2.5 KiB
Python

import os
class Url2File:
def __init__(self,path:str,prefix: str,
indexes: list, inherit: bool=False):
self.path = path
self.starts = prefix
self.indexes = indexes
self.inherit = inherit
def realurl(self,url:str) -> str :
items = url.split('/')
items = [ i for i in items if i != '.' ]
while '..' in items:
for i,v in enumerate(items):
if v=='..' and i > 0:
del items[i]
del items[i-1]
break
return '/'.join(items)
def isFolder(self,url: str) ->bool:
if url.startswith(self.starts):
rp = self.path + url[len(self.starts):]
real_path = os.path.abspath(rp)
if os.path.isdir(real_path):
return True
return False
def isFile(self,url:str) ->bool:
if url.startswith(self.starts):
rp = self.path + url[len(self.starts):]
real_path = os.path.abspath(rp)
if os.path.isfile(real_path):
return True
return False
def defaultIndex(self,url: str) -> str:
for p in self.indexes:
rp = url + '/' + p
r = self.url2file(rp)
if r is not None:
return r
return None
def url2file(self,url: str):
if url[-1] == '/':
url = url[:-1]
print('url2file.py:self.starts=',self.starts)
if url.startswith(self.starts):
f = self.path + url[len(self.starts):]
real_path = os.path.abspath(f)
if os.path.isdir(real_path):
for f in self.indexes:
p = os.path.join(real_path,'f)
if os.path.isfile(p):
return p
if os.path.isfile(real_path):
return real_path
if not self.inherit:
print('url2file.py:real_path=',real_path)
return None
items = url.split('/')
if len(items) > 2:
del items[-2]
url = '/'.join(items)
return self.url2file(url)
return None
def relatedurl(self,url: str, name: str) -> str:
if url[-1] == '/':
url = url[:-1]
if not self.isFolder(url):
items = url.split('/')
del items[-1]
url = '/'.join(items)
url = url + '/' + name
return self.realurl(url)
def relatedurl2file(self,url: str, name: str):
url = self.relatedurl(url,name)
return self.url2file(url)
class TmplUrl2File(Url2File):
def __init__(self,paths,indexes, subffixes=['.tmpl'],inherit=False):
self.paths = paths
self.u2fs = [ Url2File(p,prefix,indexes,inherit=inherit) \
for p,prefix in paths ]
self.subffixes = subffixes
def url2file(self,url):
for u2f in self.u2fs:
fp = u2f.url2file(url)
if fp:
return fp
return None
def list_tmpl(self):
ret = []
for rp,_ in self.paths:
p = os.path.abspath(rp)
[ ret.append(i) for i in listFile(p,suffixs=self.subffixes,rescursive=True) ]
return sorted(ret)