source: QX/RPlain/Rplain.py @ 957

__doc__ = """Rplain product module"""
__version__ = '0.1' 

import os
from App.special_dtml import HTMLFile
from App.Dialogs import MessageDialog
from Persistence import Persistent

import OFS.SimpleItem
import Acquisition
import AccessControl.Role

import rpy2.robjects
from rpy2.robjects import r
rpy2.robjects.__allow_access_to_unprotected_subobjects__ = 1
import rpy2.rlike.container as rlc
rlc.__allow_access_to_unprotected_subobjects__ = 1

from zope.proxy import removeAllProxies 
from zope.traversing.browser import absoluteURL
from plot import cachedImages, registerImage
from rstat import gdd
from http_import import http_import


def manage_addRplain(self, id='R', REQUEST=None):
    self._setObject(id, Rplain(id))
    if REQUEST is not None:
        return self.manage_main(self, REQUEST)



class Rplain( OFS.SimpleItem.Item, 
              Persistent, 
              Acquisition.Implicit, 
              AccessControl.Role.RoleManager ):

    meta_type = 'Rplain'
    manage_main = HTMLFile('RplainEdit', globals())
    manage_options = ( 
      {'label': 'Edit',       'action': 'manage_main'},
    ) 
    __ac_permissions__=( 
      ('View management screens', ('manage_tabs','manage_main')),
      ('Change Rplains'     ,     ('manage_edit',)             ),
    )
    def manage_edit(self, REQUEST=None):
        if REQUEST is not None:
            return MessageDialog(
              title = 'Edited',
              message = "Properties for %s changed." % self.id,
              action = './manage_main',
            )


    ############################################################################
    # Constructor
    ############################################################################
    def __init__(self, id): 
        self.id = id

    ############################################################################
    # r
    ############################################################################
    def r(self):
      return rpy2.robjects.r

    ############################################################################
    # robjects
    ############################################################################
    def robjects(self):
      return rpy2.robjects

    ############################################################################
    # rlc
    ############################################################################
    def rlc(self):
      return rpy2.rlike.container



    ############################################################################
    # cybertools: __call__
    ############################################################################

    index_html=None

    def __call__(self, REQUEST=None, **kw):
        key = self.REQUEST.form.get('image', 'not_found')
        if '.' in key:  # remove extension possibly added to make Flash happy
          key = key.split('.', 1)[0]
        path = cachedImages[key].path
        #--return {'key':key,'keys':cachedImages.keys(),'path':path}
        self.setHeaders(path)
        f = open(path, 'rb')
        data = f.read()
        f.close()
        return data

    def setHeaders(self, name=None):
        response = self.REQUEST.response
        # TODO: get content type from name (extension)
        response.setHeader('Content-Type', 'image/jpeg')
        response.setHeader('Expires', 'Sat, 1 Jan 2000 00:00:00 GMT');
        response.setHeader('Pragma', 'no-cache');

    ############################################################################
    # cybertools: rplot
    # @deprecated
    ############################################################################
    def rplot(self, r , c, **kw):
        request = r
        context = c
        fn, robj = gdd(**kw)
        key = registerImage(fn)
        return '%s/R?image=%s.jpg' % (absoluteURL(context, request), key)

    ############################################################################
    # cybertools: rfile
    ############################################################################
    def rfile(self, r , c, **kw):
        request = r
        context = c
        fn = os.tempnam(None, 'rfile')+'.prn'
        data = http_import(kw['datasource'])
        f = open(fn,'wb')
        f.write(data)
        f.close()
        return fn

    ############################################################################
    # registerImage
    ############################################################################
    def registerImage(self, r , c, **kw):
        request = r
        context = c
        fn = os.tempnam(None, 'rplot')
        key = registerImage(fn)
        return fn,'%s/R?image=%s.jpg' % (absoluteURL(context, request), key)