source: trunk/Piece.py @ 580

# Copyright 2006 by the LeMill Team (see AUTHORS)
#
# This file is part of LeMill.
#
# LeMill 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; either version 2 of the License, or
# (at your option) any later version.
#
# LeMill 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 LeMill; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

from SharedMetadata import *
from Products.Archetypes.public import *
from Globals import InitializeClass
from Products.CMFCore.utils import getToolByName
from AccessControl import ClassSecurityInfo, Unauthorized
from config import PROJECTNAME, VIEW
from FieldsWidgets import WYSIWYMField, WYSIWYMWidget
from Resources import Resource

schema = BaseSchema + description + tags + author_schema + original_author_schema + license_schema + coverImage + parentVersion + contentFile + sourceFile + lemill_metadata_mods  

schema = schema + Schema((
    ImageField('image',
               required=False,
               accessor='getImage',
               mutator='setImage',
               sizes={'small':(160,120),'medium':(320,240),'large':(500,500)},
               widget = ImageWidget(visible={'view':'invisible','edit':'invisible'}),
               ),
    StringField(
        'language',
        schemata="metadata",
        accessor="Language",
        default_method ='defaultLanguage',
        vocabulary='getLanguagelist',
        default = '',
        widget=SelectionWidget(
            label='Language',
            label_msgid="label_language",
            description="Select the language of this resource",
            description_msgid="help_language",
            i18n_domain="lemill",
            visible = {'edit': 'invisible', 'view': 'invisible'},
            ),
        ),
    ))

schema.moveField('rights', pos='bottom')

class Piece(Resource):
    """Piece"""
    schema = schema
    meta_type = "Piece"
    archetype_name = "Piece"
    security = ClassSecurityInfo()
    security.declareObjectPublic()

    # Use the file as cover image, if it's an image
    def setFile(self,value,**kwargs):
        file=self.getField('file')
        # Call normal mutator for this field
        file.set(self,value,**kwargs)
        if file.getContentType(self).startswith('image/'):
            # Call mutator for the coverImage field as well
            cover=self.getField('coverImage')
            cover.set(self,value,**kwargs)
            # set 'hasCoverImage'-flag
            has_cover=self.getField('hasCoverImage')
            has_cover.set(self,True)
            # Populate the image field
            image=self.getField('image')
            image.set(self,value,**kwargs)

    def pretty_title_or_id(self):
        """Return modified title."""
        return '[P] ' + self.Title()

    def isImage(self):
        """Returns whether the piece is an image."""
        file=self.getField('file')
        return file.getContentType(self).startswith('image/')

    def download(self, REQUEST, RESPONSE):
        """ download a file """
        from Products.Archetypes.utils import contentDispositionHeader
        field = self.getField('file')
        org_filename = field.getFilename(self)
        filename = self.Title()
        pu = getToolByName(self, 'plone_utils')
        filename = pu.normalizeString(filename)
        extension = ''
        if org_filename:        # extract .doc .pdf or something
            extension = org_filename[org_filename.rfind('.'):]
            if extension == -1: extension = ''
        else:                   # try to guess extension 
            ct = field.getContentType(self)
            mr = getToolByName(self, 'mimetypes_registry')
            mt = mr.lookup(ct)
            if mt:              # mt is something like (<mimetype text/plain>,) so we'll take first one
                extension = mt[0].extensions[0]       # and take first one from here too
                extension = '.'+extension
        if extension:
            filename += extension
        header_value = contentDispositionHeader('attachment', self.getCharset(), filename=filename)
        RESPONSE.setHeader("Content-disposition", header_value)
        file = field.get(self)
        return file.index_html(REQUEST, RESPONSE)

registerType(Piece, PROJECTNAME)