Gabor Guzmics
ee289dc39e
upgraded dejaqt to use folders implemented correct folder matching rearranged pagereply to be a resourcereply (and falling back on it)
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
|
|
import logging, os
|
|
try:
|
|
from django.conf import settings
|
|
except:
|
|
logging.error('Django Settings could not be loaded. Maybe Django has not been initialized?')
|
|
settings = None
|
|
|
|
class FolderLibrary(object):
|
|
def __init__(self, folders=None):
|
|
self._folders = {}
|
|
try:
|
|
if settings:
|
|
self.folders.update( getattr(settings, 'DEJAQT_DIRS', {}) )
|
|
except:
|
|
logging.error('DEJAQT_DIRS in django settings threw error.')
|
|
import traceback
|
|
traceback.print_exc()
|
|
if folders:
|
|
# no try here: if this fails, you got yourself a programming error.
|
|
self.folders.update(folders)
|
|
self._keys = []
|
|
self.build_keycache()
|
|
|
|
def get_folders(self):
|
|
return self._folders
|
|
|
|
def set_folders(self, folders):
|
|
self._folders = folders
|
|
self.build_keycache()
|
|
folders = property(get_folders, set_folders)
|
|
|
|
def build_keycache(self):
|
|
self._keys = self._folders.keys()
|
|
self._keys.sort(key=lambda item: (-len(item), item))
|
|
|
|
def add_folder(self, url, folder):
|
|
if not url:
|
|
url = ''
|
|
self._folders[url] = folder
|
|
self.build_keycache()
|
|
|
|
def match(self, url):
|
|
# run down our keycache, first match wins.
|
|
for key in self._keys:
|
|
if url.startswith(key):
|
|
return key
|
|
|
|
def matched_folder(self, url):
|
|
m = self.match(url)
|
|
if m is not None:
|
|
folder = self._folders[m]
|
|
#heading, rest = url[:len(m)], url[len(m):]
|
|
rest = url[len(m):]
|
|
real_folder = os.path.abspath( os.path.join(folder, rest) )
|
|
if real_folder.startswith(os.path.abspath(folder)):
|
|
return real_folder
|
|
else:
|
|
logging.error('%s does not seem to be a subpath of %s' % (real_folder, folder))
|
|
|
|
def print_folders(self):
|
|
print '{'
|
|
for k in self._keys:
|
|
print "'%s': '%s'," % (k, self._folders[k])
|
|
print '}'
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# test this:
|
|
import os
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'scon.dj.settings'
|
|
f = FolderLibrary({'abc/dab/': 'c:/media',
|
|
'abc': 'd:/abc',
|
|
'abc/dab/tmp': '/tmp',
|
|
'uiuiui': 'x:/',
|
|
'abc/vul/no': 'x:/2',
|
|
'abc/vul': 'x:/3',
|
|
'abc/vul/yes': 'x:/1',
|
|
})
|
|
f.add_folder('abc/dub/', 'c:/dubdub')
|
|
f.print_folders()
|
|
|
|
print f.matched_folder('abc/dab/okokok/some.png')
|
|
|