ahserver_g/ahserver/url2file.py

111 lines
2.4 KiB
Python
Raw Normal View History

2019-07-10 17:34:45 +08:00
import os
class Url2File:
2019-12-03 11:30:15 +08:00
def __init__(self,path:str,prefix: str,
indexes: list, inherit: bool=False):
self.path = path
self.starts = prefix
2019-07-10 17:34:45 +08:00
self.indexes = indexes
self.inherit = inherit
2019-12-03 11:30:15 +08:00
def realurl(self,url:str) -> str :
2019-07-10 17:34:45 +08:00
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)
2019-12-03 11:30:15 +08:00
def isFolder(self,url: str) ->bool:
if url.startswith(self.starts):
rp = self.path + url[len(self.starts):]
2019-07-10 17:34:45 +08:00
real_path = os.path.abspath(rp)
if os.path.isdir(real_path):
return True
2019-12-05 06:04:38 +08:00
return False
2019-12-03 11:30:15 +08:00
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
2019-12-05 06:04:38 +08:00
return False
2019-07-10 17:34:45 +08:00
2019-12-03 11:30:15 +08:00
def defaultIndex(self,url: str) -> str:
2019-07-10 17:34:45 +08:00
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]
2020-12-18 14:25:46 +08:00
paths = url.split('/')[3:]
f = os.path.join(self.path,*paths)
real_path = os.path.abspath(f)
2020-12-18 13:46:35 +08:00
if os.path.isdir(real_path):
2020-12-18 14:25:46 +08:00
for idx in self.indexes:
p = os.path.join(real_path,idx)
2020-12-18 13:46:35 +08:00
if os.path.isfile(p):
return p
if os.path.isfile(real_path):
return real_path
2019-12-03 11:30:15 +08:00
2019-07-10 17:34:45 +08:00
if not self.inherit:
return None
items = url.split('/')
if len(items) > 2:
del items[-2]
url = '/'.join(items)
return self.url2file(url)
return None
2019-12-03 11:30:15 +08:00
def relatedurl(self,url: str, name: str) -> str:
2019-07-10 17:34:45 +08:00
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):
2019-12-05 06:04:38 +08:00
self.paths = paths
self.u2fs = [ Url2File(p,prefix,indexes,inherit=inherit) \
for p,prefix in paths ]
2019-07-10 17:34:45 +08:00
self.subffixes = subffixes
2019-12-05 06:04:38 +08:00
def url2file(self,url):
for u2f in self.u2fs:
fp = u2f.url2file(url)
if fp:
return fp
return None
2019-07-10 17:34:45 +08:00
def list_tmpl(self):
ret = []
2019-12-05 06:04:38 +08:00
for rp,_ in self.paths:
2019-07-10 17:34:45 +08:00
p = os.path.abspath(rp)
[ ret.append(i) for i in listFile(p,suffixs=self.subffixes,rescursive=True) ]
return sorted(ret)