This repository has been archived on 2024-11-20. You can view files and clone it, but cannot push or open issues or pull requests.
old-scribeengine/scribeengine/controllers/media.py

107 lines
4.2 KiB
Python
Raw Permalink Normal View History

# -*- 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 #
###############################################################################
2010-04-12 11:21:00 +00:00
import os
import logging
2010-04-21 13:50:16 +00:00
from pprint import pprint, pformat
2010-04-12 11:21:00 +00:00
2010-05-28 06:51:45 +00:00
from pylons.decorators import jsonify
from scribeengine.lib.base import *
from scribeengine.lib import utils
from scribeengine.model import MediaType, File
from scribeengine.model.meta import Session
2010-04-12 11:21:00 +00:00
try:
import json
except ImportError:
import simplejson as json
2010-04-12 11:21:00 +00:00
log = logging.getLogger(__name__)
class MediaController(BaseController):
2010-04-23 13:07:12 +00:00
def __before__(self):
BaseController.__before__(self)
2010-11-11 05:17:00 +00:00
self._add_javascript(u'tinymce/tiny_mce_popup.js')
2010-06-03 13:32:07 +00:00
self._add_javascript(u'ScribeEngine.Media.js')
2010-04-23 13:07:12 +00:00
def _get_directories(self, parent=None, tree=[]):
2010-04-21 13:50:16 +00:00
old_root = parent
dirname = os.path.split(parent)[1]
2010-06-03 13:32:07 +00:00
node = {u'data': dirname, u'state': u'open'}
2010-04-21 13:32:12 +00:00
for root, dirs, files in os.walk(parent):
2010-04-21 13:50:16 +00:00
if root != old_root:
break
2010-05-28 06:51:45 +00:00
if dirs:
node[u'children'] = []
2010-04-21 13:32:12 +00:00
for dirpath in dirs:
full_dirpath = os.path.join(root, dirpath)
self._get_directories(full_dirpath, node[u'children'])
tree.append(node)
def _get_files(self, dirpath):
for root, dirs, files in os.walk(os.path.abspath(dirpath)):
filelist = []
for filename in files:
if os.path.splitext(filename)[1] in [u'.png', u'.jpg', u'.gif']:
filelist.append({u'name': filename, u'type': u'/images/file-image.png'})
else:
filelist.append({u'name': filename, u'type': u'/images/file-unknown.png'})
return filelist
break
2010-04-12 11:21:00 +00:00
def index(self):
2010-05-28 06:51:45 +00:00
self._add_javascript(u'jstree/jquery.jstree.js')
directories = []
2010-04-23 13:07:12 +00:00
path = os.path.join(config[u'paths.media'], u'user%s' % c.current_user.id)
if not os.path.exists(path):
os.makedirs(path)
self._get_directories(path, directories)
c.directories = json.dumps(directories)
c.files = self._get_files(path)
c.page_title = u'Media Browser'
return render(u'/media/index.mako')
2010-06-03 13:32:07 +00:00
@jsonify
def get_files(self):
path = request.GET.get(u'path', u'').split(u',')
dirpath = os.path.join(config[u'paths.media'], *path)
2010-11-11 06:16:08 +00:00
log.debug("Path: %s", path)
return {u'results': self._get_files(dirpath)}
2010-06-03 13:32:07 +00:00
2010-05-28 06:51:45 +00:00
@jsonify
def create_directory(self):
"""
Create a directory, called via AJAX.
"""
dirname = request.GET.get(u'directory')
parent = request.GET.get(u'parent')
if dirname and parent:
parent = os.path.abspath(parent)
2010-05-28 06:51:45 +00:00
log.debug(parent)
#os.makedirs(os.path.join(parent, dirname))
return u'1'
else:
return u'0'