--- ZSQLExtend/ZSQLExtend.py	2007/07/24 09:22:14	1.117
+++ ZSQLExtend/ZSQLExtend.py	2007/11/05 18:45:35	1.120
@@ -77,6 +77,15 @@ def utf8ify(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):
@@ -349,7 +358,8 @@ class ZSQLExtendFolder(Folder,Persistent
         
     
     def importXMLFileFMP(self,table,dsn=None,uploadfile=None,update_fields=None,id_field=None,sync_mode=False,
-                         lc_names=True,keep_fields=False,replace=False,ascii_db=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.
@@ -363,6 +373,7 @@ class ZSQLExtendFolder(Folder,Persistent
         @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)
         @param redirect_url: (optional) url for redirecting after the upload is done
         '''
@@ -391,11 +402,31 @@ class ZSQLExtendFolder(Folder,Persistent
         options.keep_fields=keep_fields
         options.ascii_db=ascii_db
         options.replace_table=replace
+        options.backup_table=backup
+        options.debug=debug
+        
+        if RESPONSE and log_to_response:
+            # set up logging to response as plain text
+            RESPONSE.setHeader("Content-Type","text/plain; charset=utf-8")
+            RESPONSE.write("Import FMPXML file...\n\n")
+            loghandler = logging.StreamHandler(RESPONSE)
+            if debug:
+                loghandler.setLevel(logging.DEBUG)
+            else:
+                loghandler.setLevel(logging.INFO)
+            logger = logging.getLogger('db.import.fmpxml')
+            logger.addHandler(loghandler)
+            options.use_logger_instance = logger
 
         importFMPXML(options)
         
         os.remove(filename)
         
+        if RESPONSE and log_to_response:
+            loghandler.flush()
+            RESPONSE.write("\n\n DONE!")
+            return
+        
         if RESPONSE and redirect_url:
             RESPONSE.redirect(redirect_url)
 
@@ -443,6 +474,25 @@ class ZSQLExtendFolder(Folder,Persistent
         """
         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)"""
+        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"""
+        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 (experimental)"""
         ret={}
@@ -497,13 +547,13 @@ class ZSQLExtendFolder(Folder,Persistent
         return pt()
 
 
-    def changeZSQLExtend(self,label,description,weight=0,REQUEST=None,connection_id=None):
+    def changeZSQLExtend(self,label,description,weight=0,connection_id=None,REQUEST=None,):
         """change the Konfiguration"""
         self.connection_id=connection_id
         self.weight=weight
         self.label=label
         self.description=description
-        
+                    
         if REQUEST is not None:
             return self.manage_main(self, REQUEST)