old-scribeengine/scribeengine/controllers/media.py

# -*- 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')