--- ZSQLExtend/ZSQLExtend.py 2007/02/22 13:23:20 1.105 +++ ZSQLExtend/ZSQLExtend.py 2012/09/03 13:06:11 1.144 @@ -4,13 +4,15 @@ from Globals import DTMLFile,package_hom import urllib import re import string -import sys #from pyPgSQL import libpq -from AccessControl import getSecurityManager +import psycopg2 +from AccessControl import getSecurityManager,Unauthorized from Products.PageTemplates.ZopePageTemplate import ZopePageTemplate from Products.PageTemplates.PageTemplateFile import PageTemplateFile from Products.ZSQLMethods.SQL import SQLConnectionIDs +from Shared.DC.ZRDB.Results import Results + from xml.sax.saxutils import escape from types import * import Shared.DC.ZRDB.DA @@ -19,12 +21,12 @@ import os.path import os import copy import unicodedata - -import logging +import tempfile +import sys #ersetzt logging def logger(txt,method,txt2): - """logging""" + """logging""" logging.info(txt+ txt2) @@ -57,6 +59,36 @@ def analyseIntSearch(word): else: return "BETWEEN "+splitted[0]+" AND "+splitted[1] +def unicodify(str): + """decode str (utf-8 or latin-1 representation) into unicode object""" + if not str: + return u"" + if type(str) is StringType: + try: + return str.decode('utf-8') + except: + return str.decode('latin-1') + else: + return str + +def utf8ify(str): + """encode unicode object or string into byte string in utf-8 representation""" + if not str: + return "" + if type(str) is StringType: + return str + else: + return str.encode('utf-8') + + +def setPsycopg2UseUnicode(): + """force Psycopg2DA to return unicode objects""" + try: + import psycopg2 + import psycopg2.extensions + psycopg2.extensions.register_type(psycopg2.extensions.UNICODE) + except: + logging.error("Unable to force psycopg2 to use unicode") def sql_quote(v): @@ -67,9 +99,12 @@ def sql_quote(v): v=string.join(string.split(v,dkey),quote_dict[dkey]) return "'%s'" % v -def showSQLConnectionIDs(self): - return SQLConnectionIDs(self) +#def showSQLConnectionIDs(self): +# return SQLConnectionIDs(self) +class Options: + """options class""" + class ZSQLIndex(SimpleItem): """index""" meta_type="ZSQLIndex" @@ -86,35 +121,88 @@ class ZSQLIndex(SimpleItem): return self.index class ZSQLExtendFolder(Folder,Persistent, Implicit): - """Folder""" + """Klasse die Methoden fuer die Abfrage einer SQL-Datenbank zur Verfuegung stellt. + + """ meta_type="ZSQLExtendFolder" + #primaryKeys={} + #primaryKey= + + def getSQLConnectionIDs(self): + logging.debug(SQLConnectionIDs(self)) + return SQLConnectionIDs(self) + + def getPrimaryKey(self,table=None): + """returns primary key for the database""" + if table is None: + return getattr(self,'primaryKey','oid') + + + + pks= getattr(self, "primaryKeys",{}) + + k = pks.get(table,None) + + if k is None: + logging.debug("getPrimaryKey: no primary key for table %s stored use standard key %s"%(table,getattr(self,'primaryKey','oid'))) + logging.debug(pks) + return getattr(self,'primaryKey','oid') + + return k + + + def getPrimaryKeysAsString(self): + """get all keys""" + + if not hasattr(self, 'primaryKeys'): + return "" + + return ";".join(["%s:%s"%val for val in self.primaryKeys.items()]) + + def ZSQLQuote(self,str): """quote str for sql""" return sql_quote(str) - + def unicodify(self, s): + """return unicode object for string (utf-8 or latin1) or unicode object s""" + return unicodify(s) + + def utf8ify(self, s): + """return utf-8 encoded string object for string or unicode object s""" + return utf8ify(s) + + def normalizeField(self,table,fieldname, newFieldName=None,mode="alter", RESPONSE=None): - """normalize a field""" + """normalize a field, d.h. entfernt alle diakritischen Zeichen und ersetzt diese + durch den Grundbuchstaben in einer Spalte einer Tabelle + @param table: Tabellename + @param fieldname: Name der Spalte + @param newFieldName: (optional) default ist fieldname+"_normal" + @param mode: (optional) default ist "alter". Mode "alter" aendert ein bestehendes Feld newFieldName, mode "create" erzeugt diese zuerst. + """ import unicodedata if not newFieldName: newFieldName=fieldname+"_normal" - + + #normalisierungs routine def normal(str): if str: return unicodedata.normalize('NFKD', str.decode('utf-8')).encode('ASCII', 'ignore') else: return "" + if mode=="create": # create the field qstr="""alter table %s add %s %s""" self.ZSQLSimpleSearch(qstr%(table,newFieldName,'text')) - qstr="select oid,%s from %s"%(fieldname,table) + qstr="select %s from %s"%(fieldname,table) for result in self.ZSQLSimpleSearch(qstr): - qstr="update %s set %s = %s where oid = %s" + qstr="update %s set %s = %s where "+self.getPrimaryKey()+" = %s" - self.ZSQLSimpleSearch(qstr%(table,newFieldName,self.ZSQLQuote(normal(getattr(result,fieldname))),result.oid)) + self.ZSQLSimpleSearch(qstr%(table,newFieldName,self.ZSQLQuote(normal(getattr(result,fieldname))),getattr(result,self.getPrimaryKey))) def importAccessModell(self,configFileName,RESPONSE=None): """import tables from access @@ -302,9 +390,8 @@ class ZSQLExtendFolder(Folder,Persistent logger("update xml",logging.INFO,queryStr) self.ZSQLSimpleSearch(queryStr) ret+="ud: %s \n"%field - else: - + else: fields=",".join(dataSet.keys()) values=",".join([""" %s """%self.ZSQLQuote(dataSet[x]) for x in dataSet.keys()]) @@ -313,246 +400,121 @@ class ZSQLExtendFolder(Folder,Persistent self.ZSQLSimpleSearch(queryStr) logger("update xml",logging.INFO,queryStr) - - - return ret - def importXMLFile(self,table,data=None,identify=None,filename=None,RESPONSE=None): - #TODO: finish importXMLFile - ''' - Import XML file into the table - @param table: name of the table the xml shall be imported into - @param containerTagName: XML-Tag which describes a dataset - @param file: xmlfile handle - @param identify: (optional) field res. tag which identifies a entry uniquely for updating purposes. - @param RESPONSE: (optional) - ''' - from xml.dom.pulldom import parseString - - doc=parseString(file.read()) - while 1: - node=doc.getEvent() - - if node is None: - break; - else: - if node[1].nodeName==containerTagName: - doc.expandNode(node[1]) - cols=node[1].getElementsByTagName('COL') - dataSet=[] - for col in cols: - data=col.getElementsByTagName('DATA') - dataSet.append(getTextFromNode(data[0])) - update=False - if identify: - - nr=fieldNames.index(identify) - field=dataSet[nr] - - searchStr="""select %s from %s where %s = '%s'"""%(identify,table,identify,field) - logger("import xml",logging.INFO,searchStr) - search=self.ZSQLSimpleSearch(searchStr) - if search: - update=True - - if update: - tmp=[] - for fieldName in fieldNames: - tmp.append("""%s = %s"""%(fieldName,self.ZSQLQuote(dataSet[fieldNames.index(fieldName)]))) - setStr=",".join(tmp) - nr=fieldNames.index(identify) - field=dataSet[nr] - - queryStr="""UPDATE %s SET %s WHERE %s = '%s' """%(table,setStr,identify,field) - logger("update xml",logging.INFO,queryStr) - self.ZSQLSimpleSearch(queryStr) - ret+="ud: %s \n"%field - else: - - - fields=",".join(fieldNames) - values=",".join([""" %s """%self.ZSQLQuote(x) for x in dataSet]) - - - queryStr="""INSERT INTO %s (%s) VALUES (%s)"""%(table,fields,values) - self.ZSQLSimpleSearch(queryStr) - logger("update xml",logging.INFO,queryStr) - ret+="ad: %s \n"%field - - elif node[1].nodeName=="METADATA": - fieldNames=[] - doc.expandNode(node[1]) - - names=node[1].getElementsByTagName('FIELD') - - for name in names: - fieldNames.append(name.getAttribute('NAME')) - - logger("update xml: fieldnames",logging.INFO,repr(fieldNames)) - qstr="""select attname from pg_attribute, pg_class where attrelid = pg_class.oid and relname = '%s' """ - columns=[x.attname for x in self.ZSQLSimpleSearch(qstr%table)] - - for fieldName in fieldNames: - logger("update xml: fieldname",logging.INFO,repr(fieldName)) - if fieldName not in columns: - qstr="""alter table %s add %s %s""" - self.ZSQLSimpleSearch(qstr%(table,fieldName,'text')) - logger("update xml: fieldname add",logging.INFO,qstr%(table,fieldName,'text')) - #fn=node[1].getAttribute("xml:id") - #nf=file("xtf/"+fn+".xtf",'w') - #nf.write(""""""+node[1].toxml()+"") - #print "wrote: %s"%fn - - - def importXMLFileFMP(self,table,data=None,filename=None,update_fields=None,id_field=None,sync_mode=False,RESPONSE=None): - ''' + def importXMLFileFMP(self,table,dsn=None,uploadfile=None,update_fields=None,id_field=None,sync_mode=False, + lc_names=True,keep_fields=False,ascii_db=False,replace=False,backup=False, + debug=False,log_to_response=False, + redirect_url=None,RESPONSE=None): + """ Import FileMaker XML file (FMPXMLRESULT format) into the table. - @param table: name of the table the xml shall be imported into - @param data: xml data as bz2 string - @param filename: xmlfile filename + @param dsn: database connection string + @param table: name of the table the xml shall be imported into (may be comma-separated list) + @param uploadfile: xmlfile file @param update_fields: (optional) list of fields to update; default is to create all fields @param id_field: (optional) field which uniquely identifies an entry for updating purposes. @param sync_mode: (optional) really synchronise, i.e. delete entries not in XML file + @param lc_names: (optional) lower case and clean up field names from XML + @param keep_fields: (optional) don't add fields to SQL database + @param ascii_db: (optional) assume ascii encoding in db + @param replace: (optional) delete and re-insert data + @param backup: (optional) create backup of old table (breaks indices) @param RESPONSE: (optional) - ''' - from xml.dom.pulldom import parseString,parse - import transaction - - ret = "" - - if data: - data=bz2.decompress(base64.decodestring(data)) - logger("fmpxml",logging.INFO,"received file") - doc=parseString(data) - logger("fmpxml",logging.INFO,"parsed file") - - elif filename: - fh=file(filename) - logger("fmpxml",logging.INFO,"reading file") - doc=parse(fh) - logger("fmpxml",logging.INFO,"parsed file") - - dbIDs = {} - rowcnt = 0 - - if id_field is not None: - # prepare a list of ids for sync mode - qstr="select %s from %s"%(id_field,table) - for id in self.ZSQLSimpleSearch(qstr): - # value 0: not updated - dbIDs[id[0]] = 0; - rowcnt += 1 - - logger("fmpxml",logging.INFO,"%d entries in DB to sync"%rowcnt) - - fieldNames = [] - rowcnt = 0 - id_val = '' - - while 1: - node=doc.getEvent() + @param redirect_url: (optional) url for redirecting after the upload is done + """ - if node is None: - break; - - # METADATA tag defines number and names of fields in FMPXMLRESULT - if node[1].nodeName == 'METADATA': - doc.expandNode(node[1]) - - names=node[1].getElementsByTagName('FIELD') - - for name in names: - fn = name.getAttribute('NAME') - fieldNames.append(fn) - - if update_fields is None: - # update all fields - update_fields = fieldNames - - logger("fmpxml fieldnames:",logging.INFO,repr(fieldNames)) - # get list of fields in db table - qstr="""select attname from pg_attribute, pg_class where attrelid = pg_class.oid and relname = '%s'""" - columns=[x.attname for x in self.ZSQLSimpleSearch(qstr%table)] - - # adjust db table to fields in XML and fieldlist - for fieldName in fieldNames: - logger("fmpxml fieldname:",logging.INFO,repr(fieldName)) - if (fieldName not in columns) and (fieldName in update_fields): - qstr="""alter table %s add %s %s""" - self.ZSQLSimpleSearch(qstr%(table,fieldName,'text')) - logger("fmpxml add field:",logging.INFO,qstr%(table,fieldName,'text')) - - # ROW tags (in RESULTSET tag) hold data - elif node[1].nodeName == 'ROW': - rowcnt += 1 - - doc.expandNode(node[1]) - cols=node[1].getElementsByTagName('COL') - dataSet={} - i = 0 - # populate with data - for col in cols: - data=col.getElementsByTagName('DATA') - dataSet[fieldNames[i]] = getTextFromNode(data[0]) - i += 1 - - update=False - - # synchronize by id_field - if id_field: - id_val=dataSet[id_field] - if id_val in dbIDs: - dbIDs[id_val] += 1 - update=True - - if update: - # update existing row (by id_field) - setvals=[] - for fieldName in update_fields: - setvals.append("%s = %s"%(fieldName,self.ZSQLQuote(dataSet[fieldName]))) - setStr=string.join(setvals, ',') - id_val=dataSet[id_field] - qstr="""UPDATE %s SET %s WHERE %s = '%s' """%(table,setStr,id_field,id_val) - #logger("fmpxml update:",logging.INFO,queryStr) - self.ZSQLSimpleSearch(qstr) - ret+="up: %s \n"%id_val + tfilehd,filename=tempfile.mkstemp() + tfile=os.fdopen(tfilehd,'w') + logging.info("importXMLFileFMP: importing %s"%uploadfile) + for c in uploadfile.read(): + tfile.write(c) + tfile.close() + + from importFMPXML import importFMPXML + + if not dsn: + dsn=self.getConnectionObj().connection_string + + logging.debug("dsn: %s"%dsn) + logging.debug("table: %s"%table) + logging.debug("update_fields: %s"%update_fields) + logging.debug("id_field: %s"%id_field) + logging.debug("sync_mode: %s"%sync_mode) + logging.debug("lc_names: %s"%lc_names) + logging.debug("keep_fields: %s"%keep_fields) + logging.debug("ascii_db: %s"%ascii_db) + logging.debug("replace: %s"%replace) + logging.debug("backup: %s"%backup) + logging.debug("debug: %s"%debug) + logging.debug("log_to_response: %s"%log_to_response) + logging.debug("RESPONSE: %s"%repr(RESPONSE)) + + tablelist=table.split(',') + logging.debug("tablelist: %s"%tablelist) + #table=tables + + for t in tablelist : + logging.debug("table: %s"%table) + options=Options() + options.dsn=dsn + options.table=t + options.filename=filename + options.update_fields=update_fields + options.id_field=id_field + options.sync_mode=sync_mode + options.lc_names=lc_names + options.replace_table=replace + options.keep_fields=keep_fields + options.ascii_db=ascii_db + options.replace_table=replace + options.backup_table=backup + options.debug=debug + logging.debug(options) + if RESPONSE and log_to_response: + # set up logging to response as plain text + logging.debug("Setting up logging to RESPONSE") + RESPONSE.setHeader("Content-Type","text/plain; charset=utf-8") + RESPONSE.write("Import FMPXML file...\n\n") + RESPONSE.flush() + loghandler = logging.StreamHandler(RESPONSE) + if debug: + loghandler.setLevel(logging.DEBUG) else: - # create new row - fields=string.join(update_fields, ',') - values=string.join([" %s "%self.ZSQLQuote(dataSet[x]) for x in update_fields], ',') - qstr="""INSERT INTO %s (%s) VALUES (%s)"""%(table,fields,values) - self.ZSQLSimpleSearch(qstr) - #logger("fmpxml: insert",logging.INFO,queryStr) - ret+="ad: %s \n"%dataSet.get(id_field, rowcnt) - - #logger("fmpxml row:",logging.INFO,"%d (%s)"%(rowcnt,id_val)) - if (rowcnt % 10) == 0: - logger("fmpxml row:",logging.INFO,"%d (%s)"%(rowcnt,id_val)) - transaction.commit() - - transaction.commit() - if sync_mode: - # delete unmatched entries in db - for id in dbIDs.keys(): - # find all not-updated fields - if dbIDs[id] == 0: - logger("fmpxml delete:",logging.INFO,id) - qstr = "DELETE FROM %s WHERE %s = '%s'" - self.ZSQLSimpleSearch(qstr%(table,id_field,id)) - - elif dbIDs[id] > 1: - logger("fmpxml sync:",logging.INFO,"id used more than once?"+id) - - transaction.commit() + loghandler.setLevel(logging.INFO) + logger = logging.getLogger('db.import.fmpxml') + logger.addHandler(loghandler) + options.use_logger_instance = logger + + try: + err = None + importFMPXML(options) + logging.info("importXMLFileFMP: done") + except Exception, err: + logging.error("Error importing: %s"%err) + + if RESPONSE and log_to_response: + loghandler.flush() + if err is not None: + RESPONSE.write("\n\nERROR while importing: %s"%err) + else: + RESPONSE.write("\n\n DONE!") + + elif RESPONSE and redirect_url: + RESPONSE.redirect(redirect_url) + + os.remove(filename) + - return ret - def generateIndex(self,field,index_name,table,RESPONSE=None): - """erzeuge index aus feld""" + """erzeuge ein Index Objekt einem Feld (experimental) + @param field: Feldname zu dem ein Index erzeugt werden soll + @param index_name: Name des Index + @param table: Tabellen name""" + + index={} - founds=self.ZSQLSimpleSearch("""SELECT %s,oid FROM %s LIMIT 2000"""%(field,table)) + founds=self.ZSQLSimpleSearch("""SELECT %s FROM %s """%(field,table)) for found in founds: tmp=getattr(found,field,None) @@ -560,9 +522,9 @@ class ZSQLExtendFolder(Folder,Persistent strings=tmp.split(" ") for string in strings: if index.get(string): - index[string].append(found.oid) + index[string].append(gettr(found,self.getPrimaryKey())) else: - index[string]=[found.oid] + index[string]=[gettr(found,self.getPrimaryKey())] RESPONSE.write(string+"\n") if not hasattr(self,index_name): @@ -572,26 +534,44 @@ class ZSQLExtendFolder(Folder,Persistent self._getOb(index_name).setIndex(index) def getIndex(self,index_name): - """getIndex""" + """getIndex from index_name + return an indexObject with index_name + """ + founds=self.ZopeFind(self,obj_ids=[index_name]) return founds[0][1].getIndex() - def testneu(self): - """test""" - relStatement="""period like '%s%%'""" - statement="select * from cdli_cat" - wherePart="museum_no like 'VAT%'" - classes=['Uruk III','Uruk IV'] - return self.searchRel(relStatement,statement,wherePart,classes) - def URLquote(self,txt): - """urlquote""" + """urlquote + @param txt: text der urlgequoted werden soll. + """ return urllib.quote(txt) + + def createIdSet(self, resultset, idField=None): + """returns a (frozen)set of IDs from a SQL-resultset (using idField) or a list (if idField=None)""" + logging.debug("createidset for idfield %s"%idField) + if idField is None: + return frozenset(resultset) + else: + idlist = [r[idField] for r in resultset] + return frozenset(idlist) + + def opIdSet(self, a, b, op): + """operate on sets a and b""" + logging.debug("opidset with op %s"%op) + if (op == 'intersect'): + return a.intersection(b) + elif (op == 'union'): + return a.union(b) + elif (op == 'diff'): + return a.difference(b) + + def searchRel(self,relStatement,statement,wherePart,classes): - """suche relative haufigkeiten""" + """suche relative haufigkeiten (experimental)""" ret={} allRecords=len(self.ZSQLSimpleSearch(statement + " where "+wherePart)) @@ -601,7 +581,7 @@ class ZSQLExtendFolder(Folder,Persistent return (ret,allRecords) def content_html(self): - """template fuer content""" + """template fuer content_html Aufruf, notwendig fuer Kompatibiliaet bei gemeinsamem Einsatz mich ECHO-Produkt""" try: obj=getattr(self,"ZSQLBibliography_template") @@ -614,21 +594,21 @@ class ZSQLExtendFolder(Folder,Persistent def getWeight(self): - """getLabe""" + """getWeight, gewicht notwendig fuer Kompatibiliaet bei gemeinsamem Einsatz mich ECHO-Produkt""" try: return self.weight except: return "" def getLabel(self): - """getLabe""" + """getLabel notwendig fuer Kompatibiliaet bei gemeinsamem Einsatz mich ECHO-Produkt""" try: return self.label except: return "" def getDescription(self): - """getLabe""" + """getDEscription: notwendig fuer Kompatibiliaet bei gemeinsamem Einsatz mich ECHO-Produkt""" try: return self.description except: @@ -644,19 +624,42 @@ class ZSQLExtendFolder(Folder,Persistent return pt() - def changeZSQLExtend(self,label,description,weight=0,REQUEST=None,connection_id=None): - """change it""" + def changeZSQLExtend(self,label,description,primaryKey,primaryKeys="",weight=0,connection_id=None,autocommit=None,REQUEST=None,): + """change the Konfiguration""" self.connection_id=connection_id self.weight=weight self.label=label self.description=description + self.autocommit = (autocommit == "on") + self.primaryKey=primaryKey + + if primaryKeys.lstrip().rstrip()!="": + + for vals in primaryKeys.split(";"): + splitted=vals.split(":") + if len(splitted)<2: + if REQUEST is not None: + return """Wrong Syntax of keystring: %s
%s is not of the form TABLE:key."""%(primaryKeys,vals) + + self.primaryKeys[splitted[0]]=splitted[1] + + + + if REQUEST is not None: return self.manage_main(self, REQUEST) def formatAscii(self,str,url=None): - """ersetze ascii umbrueche durch
""" - #url=None + """ersetze ascii umbrueche durch
+ @param str: string der Formatiert werden soll. + @param url: (optional) default ist "None", sonderfall erzeugt einen Link aus String mit unterliegender url + """ + #logging.debug("formatascii str=%s url=%s"%(repr(str),repr(url))) + + if not str: + return "" + str=str.rstrip().lstrip() if url and str: @@ -670,7 +673,9 @@ class ZSQLExtendFolder(Folder,Persistent retStr+="""%s
"""%(strUrl,word) str=retStr if str: - return re.sub(r"[\n]","
",str) + retStr = re.sub(r"[\n]","
",str) + #logging.debug("formatascii out=%s"%(repr(retStr))) + return retStr else: return "" @@ -712,15 +717,81 @@ class ZSQLExtendFolder(Folder,Persistent """oinly for demo""" return os.path.splitext(path)[0]+".jpg" - def ZSQLisEmpty(self,field): - """Teste ob Treffer leer""" + def ZSQLisEmpty(self,str): + """Teste ob String leer bzw. none ist. + """ #print "field",field - if not field: + if not str: return 1 - if field.strip()=="": + if str.strip()=="": return 1 return 0 + def ZSQLMultiSearch(self,_table,_searchField,_value,_idField,_additionalStatement="",_select=None,_subselectAddition="",_storename=None): + """ + Durchsucht in einer Tabelle "table" die Spalte "searchfield" nach dem allen Vorkommnissen + von Worten in value und gibt alle Werte mit gleichem id field zurueck, d.h. es wird die "und" suche ueber mehrere Eintrsege in einer + Tabelle mit gleichem idField werd realisiert, + z.B. fuer simplesearch ueber mehrere Felder + @param _table: Tabelle, die durchsucht werden soll. + @param _searchField: Feld, das durchsucht wird + @param _value: String der gesucht werden soll, gesucht wird nach allen Worten des Strings, die durch " "-getrennt sind. + @param _idField: Feld mit id fuer die identifikation gleicher Eintraege + @param _additionalStatement: (optional) Zusaetzliches SQL Statement, dass zwischen dem ersten "select from" und dem ersten "where" eingegefuegt wird. + @param _subselectAddition: (optiona) Zusaetliche SQL Statement die hinter das select statement der subselects eingefuegt werde. + @param _select: (optional) Alternativer Wert fuer den ersten SELECT Aufruf. + @param _storename: (optional) Name fuer die Zwischenspeicherung von Werten in der Session + """ + if _storename: + """store""" + else: + _storename="foundCount" + + queries=[] + #baue jede einzelne abfrage + splitted=_value.split(" ") + if not _select: + _select=_idField + + query="select %s from %s %s where lower(%s) like '%%%s%%'"%(_select,_table,_additionalStatement,_searchField,splitted[0].lower()) + + if len(splitted)>1: # mehr als ein Wort + query+=" and %s in"%_idField # dann einschraenken + for v in splitted[1:]: + queries.append("select %s from %s %s where lower(%s) like '%%%s%%'"%(_idField,_table,_subselectAddition,_searchField,v.lower())) + + + intersect=" intersect ".join(queries) # nun baue sie zusammen + query+="(%s)"%intersect + + + logging.info("ZSQLSimple: %s"%query) + retT=self.ZSQLSimpleSearch(query) + logging.info("ZSQLSimple: %s"%retT) + + #das Ergebis enthaelt unter u.U. eine id mehrfach, dieses wir jetzt vereinheitlicht. + + retFinalT={} + for x in retT: + split=_idField.split(".") + if len(split)>1: + f=split[1] + else: + f=_idField + + retFinalT[getattr(x,f)]=x + + ret=list(retFinalT.values()) + + + #aus Kompatibilaetsgruenen mit ZSQLSearch / ZSQLInlineSeach noch einzelne Felder in der SESSION belegen. + if not self.REQUEST.SESSION.has_key(_storename): + self.REQUEST.SESSION[_storename]={} + + self.REQUEST.SESSION[_storename]['searchFieldsOnly']={} + self.REQUEST.SESSION[_storename]['qs']=query + return ret + def ZSQLsearchOptions(self,fieldname=""): """return HTML Fragment with search options""" @@ -733,7 +804,12 @@ class ZSQLExtendFolder(Folder,Persistent return ret def ZSQLSelectionFromCRList(self,fieldname,listField,boxType="checkbox",checked=None): - """generate select options from a cr seperated list""" + """generate selection HTML Fragemnt from a cr seperated list + @param fieldname: Wert fuer das "name"-Attribute der erzeugten input-Tags + @param listField: "cr" (\n) getrennte Liste der Werte + @param boxType: (optional) default ist "checkbox", moegliche Werte "checkbox" und "radio" + @param checked: "cr" getrennt Liste von Werten aus listField, die als ausgewahlt markiert werden sollen. + """ fields=listField.split("\n") ret="" for field in fields: @@ -744,7 +820,14 @@ class ZSQLExtendFolder(Folder,Persistent return ret def ZSQLSelectionFromSearchList(self,fieldname,results,fieldnameResult,boxType="checkbox",checked=None): - """generate select options from a cr seperated list""" + """generate select options from research-results Objekt + generate selection HTML Fragemnt from a cr seperated list + @param fieldname: Wert fuer das "name"-Attribute der erzeugten input-Tags + @param results: result Object einer SQL-suche + @param fieldNameResult: Feldname des Resultobjekts, das angezeigt werden soll. + @param boxType: (optional) default ist "checkbox", moegliche Werte "checkbox" und "radio" + @param checked: "cr" getrennt Liste von Werten aus results.fieldNameResult, die als ausgewahlt markiert werden sollen. + """ ret="" if not results: return "" @@ -763,11 +846,11 @@ class ZSQLExtendFolder(Folder,Persistent """generate select oprions form a cr seperated list""" fields=listField.split("\n") if size: - ret=""" """%(fieldname,multiple,size) else: - ret=""" """%(fieldname,multiple) if start: if start==' ': @@ -784,7 +867,13 @@ class ZSQLExtendFolder(Folder,Persistent ret+="""""" return ret - def ZSQLOptionsFromSearchList(self,fieldname,results,fieldName,valueName=None,start=None, multiple='',startValue=None,additionalSelect="",size=None,linelen=None,selected=None): + def ZSQLOptionsFromSearchList(self,fieldname, + results,fieldName, + valueName=None,start=None, + multiple='',startValue=None, + additionalSelect="",size=None, + linelen=None,selected=None, + clear=False): """generate select options form a search list es wird """ return ret @@ -858,7 +950,7 @@ class ZSQLExtendFolder(Folder,Persistent #print "INLINE:",argv for a in argTmp.keys(): - aFiltered=re.sub(r"^-","_",a) # beginning of a command should always be "_" + aFiltered=re.sub(r"^-","_",a) # beginning of a command should always be "_" qs.append(aFiltered+"="+urllib.quote(str(argTmp[a]))) #return [] ret = self.parseQueryString(string.join(qs,","),"_",storename=storename) @@ -892,15 +984,13 @@ class ZSQLExtendFolder(Folder,Persistent def ZSQLInlineSearch(self,storename=None,args=None,**argv): """inlinesearch""" - + #logging.debug("ZSQLInlineSearch args=%s argv=%s"%(args,argv)) qs=[] if storename: """store""" else: storename="foundCount" - - if args: argTmp=args else: @@ -914,13 +1004,19 @@ class ZSQLExtendFolder(Folder,Persistent if type(argTmp[a]) is ListType: # ein parameter zweimal value="" #TODO find a better solution, currently only the last non empty entry is used. - for x in argTmp[a]: - if x: - value=x + #for x in argTmp[a]: + # if x: + # value=x + # version: join with spaces (makes sense with checkbox and -op=all) + value = " ".join(argTmp[a]) else: + try: value=str(argTmp[a]) + except: + value=utf8ify(argTmp[a]) + qs.append(aFiltered+"="+urllib.quote(value)) - + #logging.debug("InlineSearch:"+string.join(qs,",")) #return [] @@ -945,9 +1041,60 @@ class ZSQLExtendFolder(Folder,Persistent except: logger("ZSQLResetConnection",logging.ERROR, '%s %s'%sys.exc_info()[:2]) + def ZSQLSimpleSearch(self,query=None,max_rows=1000000): - """simple search""" + """new simple search""" + logging.debug("new ZSQLSimpleSearch %s"%query) + # get Connection instance + con = self.getConnectionObj() + # call to get db object + dbc = con() + if getattr(self, 'autocommit', False): + # force transaction isolation level (for psycopg2 0=autocommit) + logging.debug(" old tilevel=%s"%dbc.tilevel) + dbc.tilevel = 0 + # modified code from ZPsycopgDA.db without _register: + c = dbc.getcursor() + desc = () + r = [] + try: + try: + c.execute(query) + + except psycopg2.OperationalError: + #logging.exception("Operational error on connection, closing it.") + try: + # Only close our connection + dbc.putconn(True) + except: + #logging.debug("Something went wrong when we tried to close the pool", exc_info=True) + pass + + if c.description is not None: + if max_rows: + r = c.fetchmany(max_rows) + else: + r = c.fetchall() + desc = c.description + + dbc.failures = 0 + except StandardError, err: + raise err + + res = (dbc.convert_description(desc), r) + + else: + logging.debug(" no autocommit") + # just use DA's query method + res = dbc.query(query, max_rows=max_rows) + + # return result set as Result object with Brains + return Results(res) + + def oldZSQLSimpleSearch(self,query=None,max_rows=1000000): + """simple search""" + logging.error("ZSQLSimpleSearch X %s"%query) #print query if not query: query=self.query @@ -956,25 +1103,33 @@ class ZSQLExtendFolder(Folder,Persistent if (hasattr(self,"_v_searchSQL") and (self._v_searchSQL == None)) or (not hasattr(self,"_v_searchSQL")): self._v_searchSQL=Shared.DC.ZRDB.DA.DA("_v_searchSQL","_v_searchSQL",self.getConnectionObj().getId(),"var","") + #self._v_searchSQL=self.getConnectionObj()() self._v_searchSQL.max_rows_=max_rows + #self._v_searchSQL.set_client_encoding('UNICODE') try: - return self._v_searchSQL.__call__(var=query) + logging.error("I am here") + t=self._v_searchSQL.__call__(var=query) + #t=self._v_searchSQL.query(query) + logging.error("I am here %s"%t) + return t except : - + logger("ZSQLSimpleSearch ERROR1",logging.ERROR, '%s %s'%sys.exc_info()[:2]) if sys.exc_info()[0]=="Database Error": try: self.getConnectionObj().manage_open_connection() except: - logger("ZSQLSimpleSearch",logging.ERROR, '%s %s'%sys.exc_info()[:2]) + logger("ZSQLSimpleSearch ERROR2",logging.ERROR, '%s %s'%sys.exc_info()[:2]) else: try: self._v_searchSQL.max_rows_=max_rows - + #self._v_searchSQL.set_client_encoding('UNICODE') + return self._v_searchSQL.__call__(var=query) + #return self._v_searchSQL.query(query) except : - + logger("ZSQLSimpleSearch ERROR2",logging.ERROR, '%s %s'%sys.exc_info()[:2]) if sys.exc_info()[0]=="Database Error": try: self.getConnectionObj().manage_open_connection() @@ -1012,18 +1167,19 @@ class ZSQLExtendFolder(Folder,Persistent - def ZSQLAdd(self,format=None,RESPONSE=None,args=None,**argv): + def ZSQLAdd(self,format=None,RESPONSE=None,args=None,_useRequest=True,**argv): """Neuer Eintrag""" - if args: + if args: argTmp=args else: argTmp=argv qs_temp=[] - for a in self.REQUEST.form.keys(): - qs_temp.append(a+"="+urllib.quote(str(self.REQUEST.form[a]))) + if _useRequest: + for a in self.REQUEST.form.keys(): + qs_temp.append(a+"="+urllib.quote(str(self.REQUEST.form[a]))) qs=string.join(qs_temp,",") @@ -1038,7 +1194,10 @@ class ZSQLExtendFolder(Folder,Persistent addList={} for q in qs.split(","): + if len(q.split("="))<2: + continue name=re.sub("r'+'"," ",q.split("=")[0].lower()) + value=q.split("=")[1] value=re.sub(r'\+'," ",value) value=urllib.unquote(value) @@ -1107,16 +1266,19 @@ class ZSQLExtendFolder(Folder,Persistent table=urllib.unquote(value) elif name=="-identify": identify=urllib.unquote(value) - identify=identify.split("=")[0]+"="+sql_quote(identify.split("=")[1]) + # old code did identify with lower() which doesn't work for oids + #identify="lower("+identify.split("=")[0]+")="+sql_quote(identify.split("=")[1].lower()) + (k,v) = identify.split("=") + identify="%s=%s"%(k,sql_quote(v)) elif name=="-format": format=urllib.unquote(value) #elif (not (name[0]=="-" or name[0]=="_")) and (not len(value)==0): elif (not (name[0]=="-" or name[0]=="_")): - if value=="": - changeList.append("\""+name+"\"=null") - else: - changeList.append("\""+name+"\"="+sql_quote(urllib.unquote(value))) + if value=="": + changeList.append("\""+name+"\"=null") + else: + changeList.append("\""+name+"\"="+sql_quote(urllib.unquote(value))) changeString=string.join(changeList,",") @@ -1132,7 +1294,7 @@ class ZSQLExtendFolder(Folder,Persistent return True - def ZSQLFindIndexed(self,qs="",select="oid,*",storename=None,indexedFields=['data_line'],restrictField='id_text',**argv): + def ZSQLFindIndexed(self,tableList=[],qs="",select="*",storename=None,indexedFields=['data_line'],restrictField='id_text',**argv): """find2""" for index in self.ZopeFind(self,obj_ids=indexedFields): @@ -1145,7 +1307,7 @@ class ZSQLExtendFolder(Folder,Persistent search2 = self.ZSQLFind(tableExt=tableList[0],qs=qs,select=select,storename=storename,restrictConnect=(tableList[0]+"."+restrictField,search1),**argv) return search2 - def ZSQLFind2(self,qs="",select="oid,*",storename=None,tableList=['cdli_translit','cdli_cat'],restrictField='id_text',**argv): + def ZSQLFind2(self,qs="",select="*",storename=None,tableList=['cdli_translit','cdli_cat'],restrictField='id_text',**argv): """find2""" search1= self.ZSQLFind(qs=qs,select=select,storename=storename,tableExt=tableList[1],restrictField=restrictField,NoQuery='yes',NoLimit='yes',**argv) @@ -1155,11 +1317,11 @@ class ZSQLExtendFolder(Folder,Persistent - def ZSQLFind(self,qs="",select="oid,*",storename="foundCount",tableExt=None,NoQuery=None,NoLimit=None,restrictField=None,restrictConnect=None,filter=None,**argv): + def ZSQLFind(self,qs="",select="*",storename="foundCount",tableExt=None,NoQuery=None,NoLimit=None,restrictField=None,restrictConnect=None,filter=None,**argv): """search in database""" def delEmpty(list): - """"loesche leere elemente aus der liste""" + """loesche leere elemente aus der liste""" ret=[] for x in list: splitted=x.split("=") @@ -1218,9 +1380,11 @@ class ZSQLExtendFolder(Folder,Persistent #print "calling Query with",repr(NoQuery) + ret=self.parseQueryString(qs,"-",select=select,storemax="yes",storename=storename,tableExt=tableExt,NoQuery=NoQuery,NoLimit=NoLimit,restrictField=restrictField,restrictConnect=restrictConnect,filter=filter) #print self.REQUEST.SESSION["foundCount"] + return ret @@ -1257,28 +1421,25 @@ class ZSQLExtendFolder(Folder,Persistent """suche mit alten parametern bis auf die in argv getauschten""" if args: argv=args - + #get the old queries qs=self.REQUEST.SESSION[storename]['qs'] querys=qs.split(",") #which arguments are in the old query string - queryList={} for query in querys: arg=query.split("=")[0] if arg[0]=="_": arg="-"+arg[1:] # sicherstellen, dass an Anfang stets "_" - queryList[arg]=query.split("=")[1] - - + try: + queryList[arg]=urllib.unquote_plus(query.split("=")[1]) + except: + queryList[arg]='' + argList=[] arg="" - - - #gehe durch die zu aendernden Argumente for argTmp in argv.keys(): - arg=argTmp[0:]# sicherstellen, dass der string auh kopiert wird if arg[0]=="_": arg="-"+arg[1:] # sicherstellen, dass an Anfang stets "_" @@ -1290,10 +1451,11 @@ class ZSQLExtendFolder(Folder,Persistent str="ZSQLSearch?"+urllib.urlencode(queryList) return str + def parseQueryString(self,qs,iCT,storemax="no",select=None,nostore=None,storename="foundCount",tableExt=None,NoQuery=None,NoLimit=None,restrictField=None,restrictConnect=None,filter=None): """analysieren den QueryString""" - + logging.debug("parseQueryString qs=%s"%qs) #setzte generische werte @@ -1315,9 +1477,10 @@ class ZSQLExtendFolder(Folder,Persistent searchFieldsOnly={} queryTemplate=[] outerjoin="" + debug=None if not select: - select="oid,*" + select="*" #check for op splitted=qs.split(",") @@ -1329,14 +1492,41 @@ class ZSQLExtendFolder(Folder,Persistent select=restrictField - #erster durchgang suche operatoren + params={} for q in splitted: - - name=re.sub("r'+'"," ",q.split("=")[0].lower()) - try: - value=urllib.unquote(q.split("=",1)[1]) - except: - value="" + name=re.sub("r'+'"," ",q.split("=")[0].lower()) + if name=="_debug": + debug=True + + try: + value=urllib.unquote(q.split("=",1)[1]) + except: + value="" + + params[name]=value + + + + #set table + primaryKey="" + if not tableExt: + table=params.get(iCT+'table') + primaryKey=self.getPrimaryKey(table) + logging.debug("table:"+table) + logging.debug("primkey:"+primaryKey); + + + #erster durchgang suche operatoren + for name,value in params.items(): + +# name=re.sub("r'+'"," ",q.split("=")[0].lower()) +# if name=="_debug": +# debug=True +# +# try: +# value=urllib.unquote(q.split("=",1)[1]) +# except: +# value="" #print "Hi",name[0:3],q if name[0:3]==iCT+"op": op=value @@ -1366,18 +1556,18 @@ class ZSQLExtendFolder(Folder,Persistent sortfields[field]=value #zweiter durchgang analysiere felder - for q in qs.split(","): + for name,value in params.items(): - - name=re.sub("r'+'"," ",q.split("=")[0].lower()) - - try: - value=urllib.unquote(q.split("=",1)[1]) - - except: - value="" - +# +# name=re.sub("r'+'"," ",q.split("=")[0].lower()) +# +# try: +# value=urllib.unquote(q.split("=",1)[1]) +# +# except: +# value="" +# punktsplit=name.split(".") #sonderfall feld mit punkten(tabelle.suchFeld.ausgewaehltesFeld,feldinoriginal), d.h. suche in anderer tabelle: #analysiere alle anderen faelle @@ -1440,10 +1630,12 @@ class ZSQLExtendFolder(Folder,Persistent #something is defined by _op_TABELLE.SUCHFELD_IN_DIESER_TABELLE.SELECT_FIELD.IDENTIFIER_IN_TABELLE elif (not name[0]==iCT) and len(punktsplit)==4: + if opfields.has_key(name): op=opfields[name] else: op="ct" + namealt=name name="LOWER("+punktsplit[1]+")" value=value.lower() @@ -1468,7 +1660,7 @@ class ZSQLExtendFolder(Folder,Persistent elif op=="numerical": term=analyseIntSearch(value) - tmp=(name+" "+term) + tmp=(namealt+" "+term) # take namealt without LOWER elif op=="grep": tmp=(name+" ~* "+sql_quote(value)) elif op=="one": @@ -1480,10 +1672,10 @@ class ZSQLExtendFolder(Folder,Persistent op="all" + if value!='': #lehre Werte werde nicht hinzugefuegt + searchTmp="""%s in (select %s from %s where %s)"""%(punktsplit[3],punktsplit[2],punktsplit[0],tmp) - searchTmp="""%s in (select %s from %s where %s)"""%(punktsplit[3],punktsplit[2],punktsplit[0],tmp) - - queryTemplate.append(searchTmp) + queryTemplate.append(searchTmp) elif (not name[0]==iCT) and (not len(value)==0): @@ -1496,7 +1688,18 @@ class ZSQLExtendFolder(Folder,Persistent else: op="ct" namealt=name - name="LOWER("+name+")" + ##DW hack 29-8-12 lower(oid) funktioniert nicht, generell muss oid handling noch korrigiert werden. + + + + + if not name==primaryKey: + name="LOWER("+name+")" #immer lower key nicht definiert fuer keys. + else: + op="eq" # bei keys immer eq benutzen + + logging.debug("NAME: %s"%name) + if op=="ct": tmp=(name+" LIKE "+sql_quote("%"+value+"%")) elif op=="gt": @@ -1507,8 +1710,7 @@ class ZSQLExtendFolder(Folder,Persistent tmp=(name+"="+sql_quote(value)) elif op=="bw": tmp=(name+" LIKE "+sql_quote(value+"%")) - elif op=="ew": - tmp=(name+" LIKE "+sql_quote("%"+value)) + #tmp=(name+" LIKE "+sql_quote("%"+value)) elif op=="all": tmps=[] for word in value.split(" "): @@ -1518,7 +1720,7 @@ class ZSQLExtendFolder(Folder,Persistent elif op=="numerical": term=analyseIntSearch(value) - tmp=(name+" "+term) + tmp=(namealt+" "+term) # take namealt without LOWER elif op=="grep": tmp=(name+" ~* "+sql_quote(value)) elif op=="one": @@ -1530,7 +1732,7 @@ class ZSQLExtendFolder(Folder,Persistent op="all" - if (not tableExt) or (namealt.split('.')[0]==tableExt): + if (value!='') and ((not tableExt) or (namealt.split('.')[0]==tableExt)): #keine leeren werte und keine auschluss if searchFields.has_key(namealt): searchFields[namealt]+=lopfields.get(name,'OR')+" "+tmp searchFieldsOnly[namealt]+=lopfields.get(name,'OR')+" "+value @@ -1616,6 +1818,8 @@ class ZSQLExtendFolder(Folder,Persistent self.REQUEST.SESSION[storename]['searchFieldsOnly']=searchFieldsOnly + if debug: + logging.error("ZSQLSimpleSearch %s"%query) if not NoQuery: return self.ZSQLQuery(query) @@ -1798,7 +2002,11 @@ class ZSQLExtendFolder(Folder,Persistent return "%s"%(self.REQUEST['URL']+"?"+newquerystring,html) - + def pydev_settrace(self): + """do settrace to start debugging""" + import pydevd + pydevd.settrace() + manage_addZSQLExtendFolderForm=DTMLFile('ZSQLExtendFolderAdd', globals()) @@ -1878,12 +2086,30 @@ class ZSQLBibliography(Folder,ZSQLExtend pt.content_type="text/html" return pt() - def changeZSQLBibliography(self,tableName,label,description,connection_id=None,REQUEST=None): + def changeZSQLBibliography(self,tableName,label,description,primaryKey="",primaryKeys="",connection_id=None,REQUEST=None): """change it""" self.connection_id=connection_id self.tableName=tableName self.label=label self.description=description + self.primaryKey=primaryKey + + self.primaryKeys={} + if primaryKeys.lstrip().rstrip()!="": + + for vals in primaryKeys.split(";"): + splitted=vals.split(":") + if len(splitted)<2: + if REQUEST is not None: + return """Wrong Syntax of keystring: %s
%s is not of the form TABLE:key."""%(primaryKeys,vals) + + logging.debug(splitted) + self.primaryKeys[splitted[0]]=splitted[1] + + + + + if REQUEST is not None: return self.manage_main(self, REQUEST) @@ -1917,44 +2143,74 @@ class ZSQLBibliography(Folder,ZSQLExtend except: return None + + def getMetaDataManager(self): + return self.metadata + def findTagsFromMapping(self,referenceType): """gib hash mit label -> generic zurueck""" - self.referencetypes=self.ZopeFind(self.standardMD) + self.referencetypes=self.ZopeFind(self.getMetaDataManager(),search_sub=1) bibdata={} retdata={} - fieldlist=self.standardMD.fieldList + #fieldlist=self.standardMD.fieldList for referenceTypeF in self.referencetypes: #print referenceType,referenceTypeF[1].title - if referenceTypeF[1].title == referenceType: + if referenceTypeF[1].title.lower() == referenceType.lower(): bibdata[referenceTypeF[1].title]=referenceTypeF[1].fields bibdata['data']=referenceTypeF[1] - self.fields=bibdata[referenceType] - for field in fieldlist: + self.fields=bibdata[referenceTypeF[1].title] + + + + for field in referenceTypeF[1].getFieldList(): if referenceTypeF[1].getValue(field)[0]==None: retdata[field]=field else: retdata[field]=referenceTypeF[1].getValue(field)[0] - return retdata,fieldlist + return retdata,referenceTypeF[1].getFieldList() def findLabelsFromMapping(self,referenceType): - """gib hash mit label -> generic zurueck""" - self.referencetypes=self.ZopeFind(self.standardMD) - bibdata={} - retdata={} - fieldlist=self.standardMD.fieldList + """gib hash mit tagname -> lalbe zurueck""" - for referenceTypeF in self.referencetypes: - #print referenceType,referenceTypeF[1].title - if referenceTypeF[1].title == referenceType: - bibdata[referenceTypeF[1].title]=referenceTypeF[1].fields - bibdata['data']=referenceTypeF[1] - self.fields=bibdata[referenceType] - for field in fieldlist: - retdata[field]=referenceTypeF[1].getValue(field)[1] + + + mapping= self.getMetaDataManager().getBibMapping(referenceType) + + + + fields=mapping.getFields() + + + + + fieldlist=fields.keys() + + retdata={} + for field in fieldlist: + retdata[field]=fields[field]['label'] + +# self.referencetypes=self.ZopeFind(self.getMetaDataManager(),search_sub=1) +# +# bibdata={} +# retdata={} +# #fieldlist=self.standardMD.fieldList +# logging.debug("XX") +# for referenceTypeF in self.referencetypes: +# #print referenceType,referenceTypeF[1].title +# logging.debug("%s=%s"%(referenceTypeF[1].title,referenceType)) +# if referenceTypeF[1].title.lower() == referenceType.lower(): +# +# bibdata[referenceTypeF[1].title]=referenceTypeF[1].fields +# bibdata['data']=referenceTypeF[1] +# self.fields=bibdata[referenceTypeF[1].title] +# for field in referenceTypeF[1].getFieldList(): +# retdata[field]=referenceTypeF[1].getValue(field)[1] + return retdata,fieldlist + def createRDFTag(self,tag,content,namespace="cdli"): """create RDF""" @@ -2050,14 +2306,14 @@ class ZSQLBibliography(Folder,ZSQLExtend ret=""" """ - for found in self.ZSQLSimpleSearch("select oid from %s limit ALL"%self.tableName): + for found in self.ZSQLSimpleSearch("select % from %s limit ALL"%(self.getPrimaryKey,self.tableName)): base_url = self.absolute_url() if fix_host: #print "replacing ", http_host, " by ", fix_host base_url = string.replace(base_url, http_host, fix_host, 1) - link=base_url+"/"+"record.html?oid=%i"%found.oid - metalink=base_url+"/"+"getMetaDataXML?oid=%i"%found.oid + link=base_url+"/"+"record.html?oid=%i"%getattr(found,self.getPrimaryKey()) + metalink=base_url+"/"+"getMetaDataXML?oid=%i"%getattr(found,self.getPrimaryKey()) ret+="""\n"""%(link,metalink) @@ -2111,4 +2367,3 @@ def manage_addZSQLBibliography(self, id, - \ No newline at end of file