70 lines
3.0 KiB
Python
70 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=80 tabstop=4 softtabstop=4
|
|
|
|
###############################################################################
|
|
# ScribeEngine - Open Source Blog Software #
|
|
# --------------------------------------------------------------------------- #
|
|
# Copyright (c) 2010 Raoul Snyman #
|
|
# --------------------------------------------------------------------------- #
|
|
# This program is free software; you can redistribute it and/or modify it #
|
|
# under the terms of the GNU General Public License as published by the Free #
|
|
# Software Foundation; version 2 of the License. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT #
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or #
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for #
|
|
# more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License along #
|
|
# with this program; if not, write to the Free Software Foundation, Inc., 59 #
|
|
# Temple Place, Suite 330, Boston, MA 02111-1307 USA #
|
|
###############################################################################
|
|
|
|
import os
|
|
import logging
|
|
from pprint import pprint
|
|
|
|
from scribeengine.lib.base import *
|
|
from scribeengine.lib import utils
|
|
from scribeengine.model import MediaType, File
|
|
from scribeengine.model.meta import Session
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
class MediaController(BaseController):
|
|
|
|
def _get_directories(self, parent=None, tree={}):
|
|
if not parent:
|
|
parent = config[u'paths.media']
|
|
parent = os.path.abspath(parent)
|
|
dir_start = os.path.split(parent)[1]
|
|
tree[dir_start] = {}
|
|
for root, dirs, files in os.walk(parent):
|
|
root_parts = root.split(os.sep)
|
|
root_start = u''
|
|
for root_part in root_parts:
|
|
if root_part == dir_start:
|
|
root_start = root_part
|
|
break
|
|
|
|
for dirpath in dirs:
|
|
root_parts =
|
|
full_dirpath = os.path.join(root, dirpath)
|
|
tree[full_dirpath] = {}
|
|
|
|
for dirpath in os.listdir(parent):
|
|
full_dirpath = os.path.abspath(dirpath)
|
|
print 'path:', full_dirpath
|
|
if os.path.isdir(full_dirpath):
|
|
tree[full_dirpath] = {
|
|
u'path': dirpath,
|
|
u'children': self._get_directories(full_dirpath)
|
|
}
|
|
return tree
|
|
|
|
def index(self):
|
|
c.directories = self._get_directories()
|
|
pprint(c.directories)
|
|
return c.directories
|
|
#return render(u'/media/index.html')
|