view ZDBInterfaceFolder.py @ 4:0ade331198de

first version of ZDBInlineSearch
author casties
date Tue, 15 Feb 2011 19:59:07 +0100
parents d70e57193731
children ca30cf0e810d
line wrap: on
line source

'''
Created on 14.2.2011

@author: casties
'''

from OFS.Folder import Folder
from Products.PageTemplates.PageTemplateFile import PageTemplateFile
from AccessControl import getSecurityManager, Unauthorized
import logging
import re

from DBInterface import *


class ZDBInterfaceFolder(DBInterface, Folder):
    """Folder for database queries
    """
    
    meta_type="ZDBInterfaceFolder"
    manage_options=Folder.manage_options+(
        {'label':'Config','action':'manage_editZDBInterfaceFolderForm'},
        )

    # management templates
    manage_editZDBInterfaceFolderForm=PageTemplateFile('zpt/editZDBInterfaceFolder',globals())

    
    def __init__(self, id, title, connection_id=None):
        """init"""
        self.id = id
        self.title = title
        # database connection id
        self.connection_id = connection_id


    def getSQLQuery(self, table=None, argv):
        """returns dict with SQL query string and args"""
        logging.debug("getSQLquery table=%s argv=%s"%(table,argv))
        args = []
        select = "*"
        order = None
        wheres = []
        whereOp = "AND"
        done = {}
        # evaluate argv
        for (key, val) in argv.items():
            if key in done:
                # parameter has been processed
                continue
            
            if isinstance(val, list): 
                # join multiple parameters with spaces (makes sense with checkbox and -op=all)
                val = " ".join(val)

            # beginning of a command should always be "_"
            if key[0] == "-":
                key[0] = "_"
      
            if key == "_select":
                # SELECT expression
                select = sqlName(val, more="*,")
            elif key == "_sort":
                # sort i.e. ORDER BY expression
                order = sqlName(val, more=",")
            elif key == "_lop":
                # logical operation joining WHERE clauses
                whereOp = sqlName(val)
            elif key[:3] == "_op":
                # operation parameters _op_param=option
                param = sqlName(key[4:])
                if param in argv:
                    doOp(op, param)
                    done[param] = True
                else:
                    # no corresponding parameter
                    logging.error("getSQLquery: param=%s for op not found!"%param)
                
        # process remaining parameters (without _op)
        for (key, val) in argv.items():
            if key not in done:
                param = sqlName(key)
                # default is begins-with
                doOp("bw", param)

        # join it all
        query = "SELECT %s FROM %s"%(select, table)
        if wheres:
            query += " WHERE " + (" " + whereOp + " ").join(wheres)
        
        if order:
            query += " ORDER BY " + order

        return {'query' : query, 'args' : args}
                
        def doOp(op, param):
            # handles comparison operations in WHERE clause
            if op == "eq":
                wheres.append(param + " = %s")
                args.append(argv[param])
            elif op == "lt":
                wheres.append(param + " < %s")
                args.append(argv[param])
            elif op == "gt":
                wheres.append(param + " > %s")
                args.append(argv[param])
            elif op == "bw":
                wheres.append(param + " LIKE %s")
                args.append("%" + argv[param])
            elif op == "ew":
                wheres.append(param + " LIKE %s")
                args.append(argv[param] + "%")
            elif op == "ct":
                wheres.append(param + " LIKE %s")
                args.append("%" + argv[param] + "%")
            else:
                logging.error("getSQLquery: unknown op=%s!"%op)
                
            return
                

    def ZDBInlineSearch(self, _table=None, **argv):
        """returns result set from search"""
        query = self.getSQLQuery(_table=_table, argv)
        result = self.executeZSQL(query['query'], query['args'])
        return result
    
    def manage_editZDBInterfaceFolder(self, title=None, connection_id=None,
                     REQUEST=None):
        """Change the object"""
        if title is not None:
            self.title = title
            
        if connection_id is not None:
            self.connection_id = connection_id
                
        #checkPermission=getSecurityManager().checkPermission
        REQUEST.RESPONSE.redirect('manage_main')


manage_addZDBInterfaceFolderForm=PageTemplateFile('zpt/addZDBInterfaceFolder',globals())

def manage_addZDBInterfaceFolder(self, id, title='', label='', description='',
                     createPublic=0,
                     createUserF=0,
                     REQUEST=None):
        """Add a new object with id *id*."""
    
        ob=ZDBInterfaceFolder(str(id),title)
        self._setObject(id, ob)
        
        #checkPermission=getSecurityManager().checkPermission
        REQUEST.RESPONSE.redirect('manage_main')