def test_list_dirs(self):
for i in range(0, 5):
fsutil.create_dir(self.temp_path('a/b/c/d-{}'.format(i)))
fsutil.create_file(self.temp_path('a/b/c/f-{}'.format(i)),
content='{}'.format(i))
dirpaths = fsutil.list_dirs(self.temp_path('a/b/c'))
dirnames = [fsutil.split_path(dirpath)[-1] for dirpath in dirpaths]
self.assertEqual(len(dirpaths), 5)
self.assertEqual(dirnames, ['d-0', 'd-1', 'd-2', 'd-3', 'd-4'])
def test_list_dirs(self):
for i in range(0, 5):
fsutil.create_dir(self.temp_path("a/b/c/d-{}".format(i)))
fsutil.create_file(
self.temp_path("a/b/c/f-{}".format(i)), content="{}".format(i)
)
dirpaths = fsutil.list_dirs(self.temp_path("a/b/c"))
dirnames = [fsutil.split_path(dirpath)[-1] for dirpath in dirpaths]
self.assertEqual(len(dirpaths), 5)
self.assertEqual(dirnames, ["d-0", "d-1", "d-2", "d-3", "d-4"])
def list_dir(request):
path = request.GET.get('path', '')
with_dirs = request.GET.get('dirs', 1)
with_files = request.GET.get('files', 0)
node = Node.parse(relative=path)
result = []
if not node.is_exist or node.is_file:
raise Http404
if with_dirs:
try:
folders = fsutil.list_dirs(node.absolute)
except FileNotFoundError:
raise Http404
for i, folder in enumerate(folders):
relative_path = folder.replace(base_files_dir, '')
id_path = '/'.join(fsutil.split_path(relative_path))
folders[i] = {
'id':
hashlib.sha1(folder.encode('utf-8')).hexdigest(),
'load':
reverse('bloom:file_manager:list_dir',
params={'path': id_path}),
'class':
'folder',
'children':
len(fsutil.list_dirs(folder)) > 0,
'text':
fsutil.get_filename(folder),
'url':
static(relative_path),
'list':
reverse('bloom:file_manager:list_dir',
params={
'path': id_path,
'files': 1
}),
}
result.extend(folders)
if with_files:
try:
files = fsutil.list_files(node.absolute)
except FileNotFoundError:
raise Http404
for i, file in enumerate(files):
files[i] = {
'id': fsutil.get_file_hash(file, 'sha1'),
'class': 'file ex-' + fsutil.get_file_extension(file),
'text': fsutil.get_filename(file),
'url': static(file.replace(static_dir, '')),
}
result.extend(files)
if not path and not with_files:
result = {
'id': 'root',
'text': 'Root Folder',
'children': result,
'state': {
'opened': True,
'selected': True
},
'list': reverse('bloom:file_manager:list_dir', params={'files':
1}),
}
return JsonResponse(result, safe=False)
def test_split_path(self):
s = "/root/a/b/c/Document.txt"
self.assertEqual(fsutil.split_path(s), ["root", "a", "b", "c", "Document.txt"])
def test_split_path(self):
s = '/root/a/b/c/Document.txt'
self.assertEqual(fsutil.split_path(s),
['root', 'a', 'b', 'c', 'Document.txt'])