Changeset 9217 for main/waeup.kofa/trunk/src/waeup/kofa/browser/pages.py

Timestamp:

21 Sep 2012, 11:21:05 (12 years ago)

Author:

uli

Message:

Merge changes from uli-async-update back into trunk.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/browser/pages.py (modified) (10 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/browser/pages.py

@@ -18 +18 @@
 """ Viewing components for Kofa objects.
 """
-import copy
 import csv
 import grok
@@ -24 +23 @@
 import re
 import sys
-import time
-import re
 from urllib import urlencode
 from zope import schema
@@ -35 +32 @@
     getUtilitiesFor,
     )
-#from zope.component.interfaces import Invalid
 from zope.event import notify
-from zope.securitypolicy.interfaces import (
-    IPrincipalRoleManager, IPrincipalRoleMap)
+from zope.securitypolicy.interfaces import IPrincipalRoleManager
 from zope.session.interfaces import ISession
 from zope.password.interfaces import IPasswordManager
-from waeup.kofa.browser import (
+from waeup.kofa.browser.layout import (
     KofaPage, KofaForm, KofaEditFormPage, KofaAddFormPage,
     KofaDisplayFormPage, NullValidator)
@@ -48 +43 @@
     IDepartment, IDepartmentAdd, ICourse, ICourseAdd, ICertificate,
     ICertificateAdd, ICertificateCourse, ICertificateCourseAdd,
-    ICaptchaManager)
+    ICaptchaManager, IChangePassword)
 from waeup.kofa.browser.layout import jsaction, action, UtilityView
-from waeup.kofa.browser.resources import warning, datepicker, tabs, datatable
+from waeup.kofa.browser.resources import (
+    warning, tabs, datatable)
 from waeup.kofa.interfaces import MessageFactory as _
 from waeup.kofa.interfaces import(
@@ -56 +52 @@
     IKofaXMLImporter, IKofaXMLExporter, IBatchProcessor,
     ILocalRolesAssignable, DuplicationError, IConfigurationContainer,
-    ISessionConfiguration, ISessionConfigurationAdd,
-    IPasswordValidator, IContactForm, IKofaUtils, ICSVExporter,
-    IChangePassword)
+    ISessionConfiguration, ISessionConfigurationAdd, IJobManager,
+    IPasswordValidator, IContactForm, IKofaUtils, ICSVExporter,)
 from waeup.kofa.permissions import (
     get_users_with_local_roles, get_all_roles, get_all_users)
+
 from waeup.kofa.students.catalog import search as searchstudents
 from waeup.kofa.university.catalog import search
 from waeup.kofa.university.vocabularies import course_levels
 from waeup.kofa.authentication import LocalRoleSetEvent
-#from waeup.kofa.widgets.restwidget import ReSTDisplayWidget
 from waeup.kofa.widgets.htmlwidget import HTMLDisplayWidget
-from waeup.kofa.authentication import get_principal_role_manager
 from waeup.kofa.utils.helpers import get_user_account
 from waeup.kofa.mandates.mandate import PasswordMandate
@@ -797 +791 @@
     def delFiles(self, **data):
         form = self.request.form
-        logger = self.context.logger
         if form.has_key('val_id'):
             child_id = form['val_id']
@@ -1196 +1189 @@
         return False
 
-    @property
-    def nextstep(self):
-        return self.url(self.context, '@@import4')
-
     def update(self, headerfield=None, back2=None, cancel=None, proceed=None):
         datatable.need()
@@ -1390 +1379 @@
     label = _('Download portal data as CSV file')
     pnav = 0
-    export_button = _(u'Download')
+    export_button = _(u'Create CSV file')
+    _running_exports = None
 
     def getExporters(self):
@@ -1398 +1388 @@
         return sorted(title_name_tuples)
 
-    def update(self, export=None, exporter=None):
-        if None in (export, exporter):
-            return
-        self.redirect(
-            self.url(self.context, 'export.csv') + '?exporter=%s' % exporter)
+    def job_finished(self, status):
+        return status == 'completed'
+
+    def getRunningExports(self):
+        """Returns running exports as list of tuples.
+
+        Only exports triggered by the current user (identified by
+        principal.id) are returned.
+
+        Each tuple has the form (<STATUS>, <STATUS_TITLE>, <EXPORTER_NAME>).
+
+        ``STATUS``:
+           the status as machine readable string (something like
+           ``'completed'``)
+
+        ``STATUS_TITLE``:
+           status of export as translated string.
+
+        ``EXPORTER_NAME``:
+           string representing the exporter title used when triggering
+           the export job.
+        """
+        if self._running_exports is None:
+            self._running_exports = self._getRunningExports()
+        return self._running_exports
+
+    def _getRunningExports(self):
+        result = self.context.get_export_jobs_status(self.user_id)
+        return result
+
+    def update(self, export=None, start_export=None, exporter=None,
+               discard=None, download=None):
+        self.user_id = self.request.principal.id
+        if discard:
+            myjobs = self.context.get_running_export_jobs(self.user_id)
+            for entry in myjobs:
+                self.context.delete_export_entry(entry)
+                self.flash(_('Discarded export result'))
+            return
+        if download:
+            myjobs = self.context.get_running_export_jobs(self.user_id)
+            if not len(myjobs):
+                self.flash(_('This export was already discarded.'))
+                return
+            job_id = myjobs[0][0]
+            self.redirect(
+                self.url(self.context, 'export.csv') + '?job_id=%s' % job_id)
+            return
+        if None in (start_export, exporter):
+            return
+        job_id = self.context.start_export_job(
+            exporter, self.request.principal.id)
+        self.redirect(self.url(self.context, 'export'))
         return
 
@@ -1410 +1448 @@
     grok.require('waeup.manageDataCenter')
 
-    def render(self, exporter=None):
-        if exporter is None:
-            return
-        ob_class = self.__implemented__.__name__.replace('waeup.kofa.','')
-        self.context.logger.info(
-            '%s - exported: %s' % (ob_class, exporter))
-        exporter = getUtility(ICSVExporter, name=exporter)
-        csv_data = exporter.export_all(grok.getSite())
-        #csv_data.seek(0)
+    def render(self, exporter=None, job_id=None):
+
+        manager = getUtility(IJobManager)
+        job = manager.get(job_id)
+        if job is None:
+            return
+        if hasattr(job.result, 'traceback'):
+            # XXX: Some error happened. Do something more approriate here...
+            return
+        path = job.result
+        if not os.path.exists(path):
+            # XXX: Do something more appropriate here...
+            return
+        result = open(path, 'rb').read()
+        acronym = grok.getSite()['configuration'].acronym.replace(' ','')
+        filename = "%s_%s" % (acronym, os.path.basename(path))
         self.response.setHeader(
             'Content-Type', 'text/csv; charset=UTF-8')
-        acronym = grok.getSite()['configuration'].acronym.replace(' ','')
-        filename = "%s%s.csv" % (acronym, exporter.title.title().replace(' ',''))
         self.response.setHeader(
-            'Content-Disposition:', 'attachment; filename="%s' % filename)
-        return csv_data
+            'Content-Disposition', 'attachment; filename="%s' % filename)
+        # remove job and running_exports entry from context
+        self.context.delete_export_entry(
+            self.context.entry_from_job_id(job_id))
+        return result
 
 class ExportXMLPage(grok.View):