WIP: download zip file

Author: oriolpiera

som_switching/demo/som_switching_demo_data.xml

@@ -30,4 +30,4 @@
             <field name="ref2">R1-003</field>
         </record>
     </data>
-</openerp>
+</openerp>

som_switching/tests/tests_wizard_reexport_log.py

@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-
 
-from destral import testing
 from destral.transaction import Transaction
 from giscedata_switching.tests.common_tests import TestSwitchingImport
 

som_switching/wizard/giscedata_switching_log_reexport_wizard.py

@@ -18,6 +18,7 @@ def reexport_files(self, cursor, uid, ids, context=None):
             context = {}
         if not isinstance(ids, list):
             ids = [ids]
+        wizard = self.browse(cursor, uid, ids[0], context=context)
         log_ids = context.get('active_ids', False)
 
         if not log_ids:
@@ -32,6 +33,7 @@ def reexport_files(self, cursor, uid, ids, context=None):
         })
         failed_files = []
         log_vals = log_obj.read(cursor, uid, log_ids, ['status', 'pas', 'proces', 'request_code'])
+        generated_files = []
         for log in log_vals:
             if log['status'] == 'correcte':
                 continue
@@ -43,7 +45,9 @@ def reexport_files(self, cursor, uid, ids, context=None):
             try:
                 step_id = step_obj.search(cursor, uid,
                                           [('name', '=', log['pas']), ('proces_id', '=', proces_id[0])])
-                sw_obj.exportar_xml(cursor, uid, sw_id[0], step_id[0], pas_id[0], context=context)
+                xml = sw_obj.exportar_xml(
+                    cursor, uid, sw_id[0], step_id[0], pas_id[0], context=context)
+                generated_files.append(xml)
             except Exception, e:
                 e_string = str(e)
                 if not e_string:
@@ -54,12 +58,23 @@ def reexport_files(self, cursor, uid, ids, context=None):
             info += _(u"S'han produït els següents errors en els arxius exportats:\n")
             for failed_f in failed_files:
                 info += _(u"\t- Fitxer {} -> Error: {}\n".format(failed_f[0], failed_f[1]))
-
-        self.write(cursor, uid, ids, {'state': 'end', 'msg': info}, context=context)
+        fileHandle = StringIO.StringIO()
+        zf = zipfile.ZipFile(fileHandle, mode='w', compression=comp)
+        nom_zip = 'casos_exportats.zip'
+        for file in generated_files:
+            zf.writestr(file[0], file[1])
+        zf.close()
+        wizard.write({
+            'report': base64.encodestring(fileHandle.getvalue()),
+            'filename': nom_zip,
+        })
+        wizard.write({'state': 'end', 'msg': info}, context=context)
 
     _columns = {
         'state': fields.selection([('init', 'Init'), ('end', 'End')], 'State'),
-        'msg': fields.text('Missatge')
+        'msg': fields.text('Missatge'),
+        'report': fields.binary('Fichero a descargar'),
+        'filename': fields.char('Nombre fichero exportado', size=256),
     }
 
     _defaults = {

som_switching/wizard/giscedata_switching_log_reexport_wizard_view.xml

@@ -8,12 +8,14 @@
             <field name="arch" type="xml">
                 <form string="Reexportar fitxers dels logs">
                     <field name="state" invisible="1"/>
+                    <field name="filename" invisible="1"/>
                     <field name="msg" nolabel="1" readonly="1" height="200" width="600"/>
                     <group col="2" colspan="8" attrs="{'invisible': [('state', '!=', 'init')]}">
                         <button string="Cancel·lar" special="cancel" icon="gtk-close" colspan="1"/>
                         <button string="Reexportar fitxers" name="reexport_files" type="object" icon="gtk-go-forward" colspan="1" />
                     </group>
                     <group col="2" colspan="8" attrs="{'invisible': [('state', '=', 'init')]}">
+                        <field name="report" width="500" filename="filename_report"/>
                         <button string="Tancar" special="cancel" icon="gtk-close" colspan="1"/>
                     </group>
                 </form>