[7196] | 1 | ## $Id: batching.py 13159 2015-07-10 11:33:03Z henrik $ |
---|
| 2 | ## |
---|
| 3 | ## Copyright (C) 2011 Uli Fouquet & Henrik Bettermann |
---|
| 4 | ## This program is free software; you can redistribute it and/or modify |
---|
| 5 | ## it under the terms of the GNU General Public License as published by |
---|
| 6 | ## the Free Software Foundation; either version 2 of the License, or |
---|
| 7 | ## (at your option) any later version. |
---|
| 8 | ## |
---|
| 9 | ## This program is distributed in the hope that it will be useful, |
---|
| 10 | ## but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 11 | ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 12 | ## GNU General Public License for more details. |
---|
| 13 | ## |
---|
| 14 | ## You should have received a copy of the GNU General Public License |
---|
| 15 | ## along with this program; if not, write to the Free Software |
---|
| 16 | ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
---|
| 17 | ## |
---|
[7819] | 18 | """Kofa components for batch processing. |
---|
[4806] | 19 | |
---|
| 20 | Batch processors eat CSV files to add, update or remove large numbers |
---|
| 21 | of certain kinds of objects at once. |
---|
| 22 | """ |
---|
| 23 | import grok |
---|
[8380] | 24 | import datetime |
---|
[4821] | 25 | import os |
---|
[9217] | 26 | import shutil |
---|
[4900] | 27 | import tempfile |
---|
[4821] | 28 | import time |
---|
[10027] | 29 | import unicodecsv |
---|
[9816] | 30 | import zc.async.interfaces |
---|
[7859] | 31 | from cStringIO import StringIO |
---|
[9217] | 32 | from persistent.list import PersistentList |
---|
| 33 | from zope.component import createObject, getUtility |
---|
| 34 | from zope.component.hooks import setSite |
---|
[9726] | 35 | from zope.interface import Interface, implementer |
---|
[4806] | 36 | from zope.schema import getFields |
---|
[11849] | 37 | from zope.schema.interfaces import ConstraintNotSatisfied |
---|
[8332] | 38 | from zope.event import notify |
---|
[9217] | 39 | from waeup.kofa.async import AsyncJob |
---|
[7811] | 40 | from waeup.kofa.interfaces import ( |
---|
[9217] | 41 | IBatchProcessor, FatalCSVError, IObjectConverter, IJobManager, |
---|
| 42 | ICSVExporter, IGNORE_MARKER, DuplicationError, JOB_STATUS_MAP, |
---|
[9726] | 43 | IExportJobContainer, IExportJob, IExportContainerFinder) |
---|
[4806] | 44 | |
---|
| 45 | class BatchProcessor(grok.GlobalUtility): |
---|
| 46 | """A processor to add, update, or remove data. |
---|
| 47 | |
---|
| 48 | This is a non-active baseclass. |
---|
| 49 | """ |
---|
[8220] | 50 | grok.implements(IBatchProcessor) |
---|
[4806] | 51 | grok.context(Interface) |
---|
| 52 | grok.baseclass() |
---|
| 53 | |
---|
| 54 | # Name used in pages and forms... |
---|
[7933] | 55 | name = u'Non-registered base processor' |
---|
[6259] | 56 | |
---|
[4806] | 57 | # Internal name... |
---|
[12869] | 58 | util_name = '' |
---|
[6259] | 59 | |
---|
[4806] | 60 | # Items for this processor need an interface with zope.schema fields. |
---|
[5009] | 61 | iface = Interface |
---|
[6259] | 62 | |
---|
[4806] | 63 | # The name must be the same as the util_name attribute in order to |
---|
| 64 | # register this utility correctly. |
---|
| 65 | grok.name(util_name) |
---|
| 66 | |
---|
| 67 | # Headers needed to locate items... |
---|
[12869] | 68 | location_fields = [] |
---|
[6259] | 69 | |
---|
[4806] | 70 | # A factory with this name must be registered... |
---|
[12869] | 71 | factory_name = '' |
---|
[4806] | 72 | |
---|
| 73 | @property |
---|
| 74 | def required_fields(self): |
---|
[4829] | 75 | """Required fields that have no default. |
---|
| 76 | |
---|
| 77 | A list of names of field, whose value cannot be set if not |
---|
| 78 | given during creation. Therefore these fields must exist in |
---|
| 79 | input. |
---|
| 80 | |
---|
| 81 | Fields with a default != missing_value do not belong to this |
---|
| 82 | category. |
---|
| 83 | """ |
---|
[4806] | 84 | result = [] |
---|
| 85 | for key, field in getFields(self.iface).items(): |
---|
| 86 | if key in self.location_fields: |
---|
| 87 | continue |
---|
[4829] | 88 | if field.default is not field.missing_value: |
---|
| 89 | continue |
---|
[4806] | 90 | if field.required: |
---|
| 91 | result.append(key) |
---|
| 92 | return result |
---|
[6259] | 93 | |
---|
[4806] | 94 | @property |
---|
| 95 | def req(self): |
---|
| 96 | result = dict( |
---|
| 97 | create = self.location_fields + self.required_fields, |
---|
| 98 | update = self.location_fields, |
---|
| 99 | remove = self.location_fields, |
---|
| 100 | ) |
---|
| 101 | return result |
---|
| 102 | |
---|
| 103 | @property |
---|
| 104 | def available_fields(self): |
---|
| 105 | return sorted(list(set( |
---|
| 106 | self.location_fields + getFields(self.iface).keys()))) |
---|
[6259] | 107 | |
---|
[4806] | 108 | def getHeaders(self, mode='create'): |
---|
| 109 | return self.available_fields |
---|
| 110 | |
---|
| 111 | def checkHeaders(self, headerfields, mode='create'): |
---|
| 112 | req = self.req[mode] |
---|
| 113 | # Check for required fields... |
---|
| 114 | for field in req: |
---|
| 115 | if not field in headerfields: |
---|
| 116 | raise FatalCSVError( |
---|
| 117 | "Need at least columns %s for import!" % |
---|
| 118 | ', '.join(["'%s'" % x for x in req])) |
---|
[6828] | 119 | # Check for double fields. Cannot happen because this error is |
---|
| 120 | # already catched in views |
---|
[4806] | 121 | not_ignored_fields = [x for x in headerfields |
---|
| 122 | if not x.startswith('--')] |
---|
| 123 | if len(set(not_ignored_fields)) < len(not_ignored_fields): |
---|
| 124 | raise FatalCSVError( |
---|
| 125 | "Double headers: each column name may only appear once.") |
---|
| 126 | return True |
---|
| 127 | |
---|
| 128 | def applyMapping(self, row, mapping): |
---|
[4811] | 129 | """Apply mapping to a row of CSV data. |
---|
| 130 | """ |
---|
[4806] | 131 | result = dict() |
---|
| 132 | for key, replacement in mapping.items(): |
---|
[6824] | 133 | if replacement == u'--IGNORE--': |
---|
| 134 | # Skip ignored columns in failed and finished data files. |
---|
| 135 | continue |
---|
[4806] | 136 | result[replacement] = row[key] |
---|
| 137 | return result |
---|
[6259] | 138 | |
---|
[4832] | 139 | def getMapping(self, path, headerfields, mode): |
---|
[6824] | 140 | """Get a mapping from CSV file headerfields to actually used fieldnames. |
---|
| 141 | |
---|
[4811] | 142 | """ |
---|
[4832] | 143 | result = dict() |
---|
[10027] | 144 | reader = unicodecsv.reader(open(path, 'rb')) |
---|
[4806] | 145 | raw_header = reader.next() |
---|
[4832] | 146 | for num, field in enumerate(headerfields): |
---|
| 147 | if field not in self.location_fields and mode == 'remove': |
---|
[6824] | 148 | # Skip non-location fields when removing. |
---|
| 149 | continue |
---|
| 150 | if field == u'--IGNORE--': |
---|
| 151 | # Skip ignored columns in failed and finished data files. |
---|
| 152 | continue |
---|
[4832] | 153 | result[raw_header[num]] = field |
---|
| 154 | return result |
---|
[4806] | 155 | |
---|
[6273] | 156 | def stringFromErrs(self, errors, inv_errors): |
---|
| 157 | result = [] |
---|
| 158 | for err in errors: |
---|
| 159 | fieldname, message = err |
---|
| 160 | result.append("%s: %s" % (fieldname, message)) |
---|
| 161 | for err in inv_errors: |
---|
| 162 | result.append("invariant: %s" % err) |
---|
| 163 | return '; '.join(result) |
---|
| 164 | |
---|
[4806] | 165 | def callFactory(self, *args, **kw): |
---|
| 166 | return createObject(self.factory_name) |
---|
| 167 | |
---|
| 168 | def parentsExist(self, row, site): |
---|
[4811] | 169 | """Tell whether the parent object for data in ``row`` exists. |
---|
| 170 | """ |
---|
[4806] | 171 | raise NotImplementedError('method not implemented') |
---|
| 172 | |
---|
| 173 | def entryExists(self, row, site): |
---|
[4811] | 174 | """Tell whether there already exists an entry for ``row`` data. |
---|
| 175 | """ |
---|
[4806] | 176 | raise NotImplementedError('method not implemented') |
---|
| 177 | |
---|
| 178 | def getParent(self, row, site): |
---|
[4811] | 179 | """Get the parent object for the entry in ``row``. |
---|
| 180 | """ |
---|
[4806] | 181 | raise NotImplementedError('method not implemented') |
---|
[6259] | 182 | |
---|
[5009] | 183 | def getEntry(self, row, site): |
---|
[12513] | 184 | """Get the object for the entry in ``row``. |
---|
[5009] | 185 | """ |
---|
| 186 | raise NotImplementedError('method not implemented') |
---|
[6259] | 187 | |
---|
[4806] | 188 | def addEntry(self, obj, row, site): |
---|
[4811] | 189 | """Add the entry given given by ``row`` data. |
---|
| 190 | """ |
---|
[4806] | 191 | raise NotImplementedError('method not implemented') |
---|
| 192 | |
---|
| 193 | def delEntry(self, row, site): |
---|
[4811] | 194 | """Delete entry given by ``row`` data. |
---|
| 195 | """ |
---|
[6259] | 196 | raise NotImplementedError('method not implemented') |
---|
[4806] | 197 | |
---|
[7950] | 198 | def checkUpdateRequirements(self, obj, row, site): |
---|
| 199 | """Checks requirements the object must fulfill when being updated. |
---|
[7938] | 200 | |
---|
| 201 | This method is not used in case of deleting or adding objects. |
---|
| 202 | |
---|
[7950] | 203 | Returns error messages as strings in case of requirement |
---|
[7938] | 204 | problems. |
---|
[7937] | 205 | """ |
---|
[7938] | 206 | return None |
---|
[7937] | 207 | |
---|
[9706] | 208 | def updateEntry(self, obj, row, site, filename): |
---|
[4984] | 209 | """Update obj to the values given in row. |
---|
[8220] | 210 | |
---|
| 211 | Returns a string describing the fields changed. |
---|
[4984] | 212 | """ |
---|
[8220] | 213 | changed = [] |
---|
[4829] | 214 | for key, value in row.items(): |
---|
[8220] | 215 | # Skip fields to be ignored. |
---|
| 216 | if value == IGNORE_MARKER: |
---|
| 217 | continue |
---|
[8304] | 218 | # Skip fields not declared in interface and which are |
---|
| 219 | # not yet attributes of existing objects. We can thus not |
---|
| 220 | # add non-existing attributes here. |
---|
[8220] | 221 | if not hasattr(obj, key): |
---|
| 222 | continue |
---|
[13159] | 223 | # DefaultObjectConverter.fromStringDict fails for |
---|
| 224 | # list-of-choices fields because we are using a different |
---|
| 225 | # widget for this combination. Thus the ListFieldConverter |
---|
| 226 | # returns a useless dictionary which causes getWidgetsData to |
---|
| 227 | # skip the field. The value in row remains unchanged. |
---|
| 228 | # We have to evaluate the string and replace the value here. |
---|
[9265] | 229 | try: |
---|
[13159] | 230 | evalvalue = eval(value) |
---|
| 231 | if isinstance(evalvalue, list): |
---|
| 232 | value = evalvalue |
---|
| 233 | except: |
---|
| 234 | pass |
---|
| 235 | try: |
---|
[9265] | 236 | setattr(obj, key, value) |
---|
| 237 | except AttributeError: |
---|
| 238 | # Computed attributes can't be set. |
---|
| 239 | continue |
---|
[8222] | 240 | log_value = getattr(value, 'code', value) |
---|
| 241 | changed.append('%s=%s' % (key, log_value)) |
---|
[8332] | 242 | |
---|
[8333] | 243 | # If any catalog is involved it must be updated. |
---|
| 244 | # |
---|
| 245 | # XXX: The event is also triggered when creating objects as |
---|
| 246 | # updateEntry is called also when creating entries resulting |
---|
| 247 | # in objectAdded and additional objectModified events. |
---|
| 248 | if len(changed): |
---|
| 249 | notify(grok.ObjectModifiedEvent(obj)) |
---|
[8332] | 250 | |
---|
[8220] | 251 | return ', '.join(changed) |
---|
[4821] | 252 | |
---|
[4832] | 253 | def createLogfile(self, path, fail_path, num, warnings, mode, user, |
---|
[4885] | 254 | timedelta, logger=None): |
---|
| 255 | """Write to log file. |
---|
[4821] | 256 | """ |
---|
[4885] | 257 | if logger is None: |
---|
| 258 | return |
---|
[9739] | 259 | logger.info( |
---|
| 260 | "processed: %s, %s mode, %s lines (%s successful/ %s failed), " |
---|
| 261 | "%0.3f s (%0.4f s/item)" % ( |
---|
| 262 | path, mode, num, num - warnings, warnings, |
---|
| 263 | timedelta, timedelta/(num or 1))) |
---|
[4821] | 264 | return |
---|
[4877] | 265 | |
---|
| 266 | def writeFailedRow(self, writer, row, warnings): |
---|
| 267 | """Write a row with error messages to error CSV. |
---|
| 268 | |
---|
| 269 | If warnings is a list of strings, they will be concatenated. |
---|
| 270 | """ |
---|
| 271 | error_col = warnings |
---|
| 272 | if isinstance(warnings, list): |
---|
| 273 | error_col = ' / '.join(warnings) |
---|
| 274 | row['--ERRORS--'] = error_col |
---|
| 275 | writer.writerow(row) |
---|
| 276 | return |
---|
[6259] | 277 | |
---|
[8220] | 278 | def checkConversion(self, row, mode='ignore', ignore_empty=True): |
---|
[6847] | 279 | """Validates all values in row. |
---|
| 280 | """ |
---|
| 281 | converter = IObjectConverter(self.iface) |
---|
| 282 | errs, inv_errs, conv_dict = converter.fromStringDict( |
---|
[8220] | 283 | row, self.factory_name, mode=mode) |
---|
[6847] | 284 | return errs, inv_errs, conv_dict |
---|
| 285 | |
---|
[12810] | 286 | |
---|
| 287 | def emptyRow(self, row): |
---|
| 288 | """Detect empty rows. |
---|
| 289 | """ |
---|
| 290 | for value in row.values(): |
---|
[12811] | 291 | if value.strip() and not value in (None, IGNORE_MARKER): |
---|
[12810] | 292 | return False |
---|
| 293 | return True |
---|
| 294 | |
---|
[4885] | 295 | def doImport(self, path, headerfields, mode='create', user='Unknown', |
---|
[8220] | 296 | logger=None, ignore_empty=True): |
---|
[12869] | 297 | """In contrast to most other methods, `doImport` is not supposed to |
---|
[12867] | 298 | be customized, neither in custom packages nor in derived batch |
---|
| 299 | processor classes. Therefore, this is the only place where we |
---|
| 300 | do import data. |
---|
| 301 | |
---|
| 302 | Before this method starts creating or updating persistent data, it |
---|
| 303 | prepares two more files in a temporary folder of the filesystem: (1) |
---|
| 304 | a file for pending data with file extension ``.pending`` and (2) |
---|
| 305 | a file for successfully processed data with file extension |
---|
| 306 | ``.finished``. Then the method starts iterating over all rows of |
---|
| 307 | the CSV file. Each row is treated as follows: |
---|
| 308 | |
---|
| 309 | 1. An empty row is skipped. |
---|
| 310 | |
---|
[12997] | 311 | 2. Empty strings or lists (``[]``) in the row are replaced by |
---|
| 312 | ignore markers. |
---|
[12867] | 313 | |
---|
[12868] | 314 | 3. The `BatchProcessor.checkConversion` method validates and converts |
---|
| 315 | all values in the row. Conversion means the transformation of strings |
---|
| 316 | into Python objects. For instance, number expressions have to be |
---|
| 317 | transformed into integers, dates into datetime objects, phone number |
---|
| 318 | expressions into phone number objects, etc. The converter returns a |
---|
| 319 | dictionary with converted values or, if the validation of one of the |
---|
| 320 | elements fails, an appropriate warning message. If the conversion |
---|
| 321 | fails a pending record is created and stored in the pending data file |
---|
| 322 | together with a warning message the converter has raised. |
---|
[12867] | 323 | |
---|
[12868] | 324 | 4. In **create mode** only: |
---|
[12867] | 325 | |
---|
[12868] | 326 | The parent object must be found and a child |
---|
| 327 | object with same object id must not exist. Otherwise the row |
---|
| 328 | is skipped, a corresponding warning message is raised and a |
---|
| 329 | record is stored in the pending data file. |
---|
[12867] | 330 | |
---|
[12869] | 331 | Now `doImport` tries to add the new object with the data |
---|
[12868] | 332 | from the conversion dictionary. In some cases this |
---|
[12869] | 333 | may fail and a `DuplicationError` is raised. For example, a new |
---|
[12868] | 334 | payment ticket is created but the same payment for same session |
---|
| 335 | has already been made. In this case the object id is unique, no |
---|
| 336 | other object with same id exists, but making the 'same' payment |
---|
| 337 | twice does not make sense. The import is skipped and a |
---|
| 338 | record is stored in the pending data file. |
---|
[12867] | 339 | |
---|
[12868] | 340 | 5. In **update mode** only: |
---|
| 341 | |
---|
| 342 | If the object can't be found, the row is skipped, |
---|
| 343 | a ``no such entry`` warning message is raised and a record is |
---|
| 344 | stored in the pending data file. |
---|
| 345 | |
---|
| 346 | The `BatchProcessor.checkUpdateRequirements` method checks additional |
---|
| 347 | requirements the object must fulfill before being updated. These |
---|
| 348 | requirements are not imposed by the data type but the context |
---|
| 349 | of the object. For example, post-graduate students have a different |
---|
| 350 | registration workflow. With this method we do forbid certain workflow |
---|
| 351 | transitions or states. |
---|
| 352 | |
---|
[12869] | 353 | Finally, `doImport` updates the existing object with the data |
---|
[12868] | 354 | from the conversion dictionary. |
---|
| 355 | |
---|
| 356 | 6. In **remove mode** only: |
---|
| 357 | |
---|
| 358 | If the object can't be found, the row is skipped, |
---|
| 359 | a ``no such entry`` warning message is raised and a record is |
---|
| 360 | stored in the pending data file. |
---|
| 361 | |
---|
[12869] | 362 | Finally, `doImport` removes the existing object. |
---|
[12868] | 363 | |
---|
[4811] | 364 | """ |
---|
[4832] | 365 | time_start = time.time() |
---|
[4806] | 366 | self.checkHeaders(headerfields, mode) |
---|
[4832] | 367 | mapping = self.getMapping(path, headerfields, mode) |
---|
[10027] | 368 | reader = unicodecsv.DictReader(open(path, 'rb')) |
---|
[4889] | 369 | |
---|
[4900] | 370 | temp_dir = tempfile.mkdtemp() |
---|
[6259] | 371 | |
---|
[6273] | 372 | base = os.path.basename(path) |
---|
| 373 | (base, ext) = os.path.splitext(base) |
---|
[4900] | 374 | failed_path = os.path.join(temp_dir, "%s.pending%s" % (base, ext)) |
---|
[6831] | 375 | failed_headers = mapping.values() |
---|
[4877] | 376 | failed_headers.append('--ERRORS--') |
---|
[10027] | 377 | failed_writer = unicodecsv.DictWriter(open(failed_path, 'wb'), |
---|
| 378 | failed_headers) |
---|
[8573] | 379 | os.chmod(failed_path, 0664) |
---|
[6831] | 380 | failed_writer.writerow(dict([(x,x) for x in failed_headers])) |
---|
[4891] | 381 | |
---|
[4900] | 382 | finished_path = os.path.join(temp_dir, "%s.finished%s" % (base, ext)) |
---|
[6831] | 383 | finished_headers = mapping.values() |
---|
[10027] | 384 | finished_writer = unicodecsv.DictWriter(open(finished_path, 'wb'), |
---|
| 385 | finished_headers) |
---|
[8905] | 386 | os.chmod(finished_path, 0664) |
---|
[4891] | 387 | finished_writer.writerow(dict([(x,x) for x in finished_headers])) |
---|
[6259] | 388 | |
---|
[4806] | 389 | num =0 |
---|
[4878] | 390 | num_warns = 0 |
---|
[4806] | 391 | site = grok.getSite() |
---|
[7859] | 392 | |
---|
[4806] | 393 | for raw_row in reader: |
---|
| 394 | num += 1 |
---|
[12810] | 395 | # Skip row if empty |
---|
| 396 | if self.emptyRow(raw_row): |
---|
| 397 | continue |
---|
[4806] | 398 | string_row = self.applyMapping(raw_row, mapping) |
---|
[12981] | 399 | if ignore_empty: |
---|
| 400 | # Replace empty strings and empty lists with ignore-markers |
---|
[8222] | 401 | for key, val in string_row.items(): |
---|
[12981] | 402 | if val == '' or val == '[]': |
---|
[8222] | 403 | string_row[key] = IGNORE_MARKER |
---|
| 404 | row = dict(string_row.items()) # create deep copy |
---|
[6847] | 405 | errs, inv_errs, conv_dict = self.checkConversion(string_row, mode) |
---|
[6273] | 406 | if errs or inv_errs: |
---|
[4878] | 407 | num_warns += 1 |
---|
[6273] | 408 | conv_warnings = self.stringFromErrs(errs, inv_errs) |
---|
| 409 | self.writeFailedRow( |
---|
[6824] | 410 | failed_writer, string_row, conv_warnings) |
---|
[4821] | 411 | continue |
---|
[6273] | 412 | row.update(conv_dict) |
---|
[6259] | 413 | |
---|
[4806] | 414 | if mode == 'create': |
---|
| 415 | if not self.parentsExist(row, site): |
---|
[4878] | 416 | num_warns += 1 |
---|
[4877] | 417 | self.writeFailedRow( |
---|
[6824] | 418 | failed_writer, string_row, |
---|
[12868] | 419 | "Not all parents do exist yet.") |
---|
[4806] | 420 | continue |
---|
| 421 | if self.entryExists(row, site): |
---|
[4878] | 422 | num_warns += 1 |
---|
[4877] | 423 | self.writeFailedRow( |
---|
[6824] | 424 | failed_writer, string_row, |
---|
[12868] | 425 | "This object already exists.") |
---|
[4806] | 426 | continue |
---|
| 427 | obj = self.callFactory() |
---|
[7273] | 428 | # Override all values in row, also |
---|
| 429 | # student_ids and applicant_ids which have been |
---|
| 430 | # generated in the respective __init__ methods before. |
---|
[9706] | 431 | self.updateEntry(obj, row, site, base) |
---|
[6243] | 432 | try: |
---|
| 433 | self.addEntry(obj, row, site) |
---|
[6273] | 434 | except KeyError, error: |
---|
[6219] | 435 | num_warns += 1 |
---|
| 436 | self.writeFailedRow( |
---|
[12868] | 437 | failed_writer, string_row, error.message) |
---|
[8540] | 438 | continue |
---|
[8509] | 439 | except DuplicationError, error: |
---|
| 440 | num_warns += 1 |
---|
| 441 | self.writeFailedRow( |
---|
[12868] | 442 | failed_writer, string_row, error.msg) |
---|
[6219] | 443 | continue |
---|
[4806] | 444 | elif mode == 'remove': |
---|
| 445 | if not self.entryExists(row, site): |
---|
[4878] | 446 | num_warns += 1 |
---|
[4877] | 447 | self.writeFailedRow( |
---|
[6824] | 448 | failed_writer, string_row, |
---|
[9219] | 449 | "Cannot remove: no such entry") |
---|
[4806] | 450 | continue |
---|
| 451 | self.delEntry(row, site) |
---|
| 452 | elif mode == 'update': |
---|
| 453 | obj = self.getEntry(row, site) |
---|
| 454 | if obj is None: |
---|
[4878] | 455 | num_warns += 1 |
---|
[4877] | 456 | self.writeFailedRow( |
---|
[6824] | 457 | failed_writer, string_row, |
---|
[9219] | 458 | "Cannot update: no such entry") |
---|
[4806] | 459 | continue |
---|
[7950] | 460 | update_errors = self.checkUpdateRequirements(obj, row, site) |
---|
[7938] | 461 | if update_errors is not None: |
---|
[7937] | 462 | num_warns += 1 |
---|
| 463 | self.writeFailedRow( |
---|
| 464 | failed_writer, string_row, update_errors) |
---|
| 465 | continue |
---|
[11849] | 466 | try: |
---|
| 467 | self.updateEntry(obj, row, site, base) |
---|
| 468 | except ConstraintNotSatisfied, err: |
---|
| 469 | num_warns += 1 |
---|
| 470 | self.writeFailedRow( |
---|
| 471 | failed_writer, string_row, |
---|
| 472 | "ConstraintNotSatisfied: %s" % err) |
---|
| 473 | continue |
---|
[4891] | 474 | finished_writer.writerow(string_row) |
---|
[4821] | 475 | |
---|
[4832] | 476 | time_end = time.time() |
---|
| 477 | timedelta = time_end - time_start |
---|
[6259] | 478 | |
---|
[4878] | 479 | self.createLogfile(path, failed_path, num, num_warns, mode, user, |
---|
[4885] | 480 | timedelta, logger=logger) |
---|
[4894] | 481 | failed_path = os.path.abspath(failed_path) |
---|
[4878] | 482 | if num_warns == 0: |
---|
[4821] | 483 | del failed_writer |
---|
| 484 | os.unlink(failed_path) |
---|
[4894] | 485 | failed_path = None |
---|
| 486 | return (num, num_warns, |
---|
| 487 | os.path.abspath(finished_path), failed_path) |
---|
[7859] | 488 | |
---|
[9032] | 489 | def get_csv_skeleton(self): |
---|
| 490 | """Export CSV file only with a header of available fields. |
---|
| 491 | |
---|
| 492 | A raw string with CSV data should be returned. |
---|
| 493 | """ |
---|
| 494 | outfile = StringIO() |
---|
[10027] | 495 | writer = unicodecsv.DictWriter(outfile, self.available_fields) |
---|
[9734] | 496 | writer.writerow( |
---|
| 497 | dict(zip(self.available_fields, self.available_fields))) # header |
---|
[9032] | 498 | outfile.seek(0) |
---|
| 499 | return outfile.read() |
---|
| 500 | |
---|
[7859] | 501 | class ExporterBase(object): |
---|
| 502 | """A base for exporters. |
---|
| 503 | """ |
---|
| 504 | grok.implements(ICSVExporter) |
---|
| 505 | |
---|
| 506 | #: Fieldnames considered by this exporter |
---|
| 507 | fields = ('code', 'title', 'title_prefix') |
---|
| 508 | |
---|
[7907] | 509 | #: The title under which this exporter will be displayed |
---|
| 510 | #: (if registered as a utility) |
---|
| 511 | title = 'Override this title' |
---|
| 512 | |
---|
[7859] | 513 | def mangle_value(self, value, name, context=None): |
---|
[12857] | 514 | """Hook for mangling values in derived classes. |
---|
[7859] | 515 | """ |
---|
| 516 | if isinstance(value, bool): |
---|
| 517 | value = value and '1' or '0' |
---|
| 518 | elif isinstance(value, unicode): |
---|
| 519 | # CSV writers like byte streams better than unicode |
---|
| 520 | value = value.encode('utf-8') |
---|
[8380] | 521 | elif isinstance(value, datetime.datetime): |
---|
[11737] | 522 | #value = str(value) |
---|
| 523 | value = str('%s#' % value) # changed 2014-07-06, see ticket #941 |
---|
[8380] | 524 | elif isinstance(value, datetime.date): |
---|
| 525 | # Order is important here: check for date after datetime as |
---|
| 526 | # datetimes are also dates. |
---|
| 527 | # |
---|
| 528 | # Append hash '#' to dates to circumvent unwanted excel automatic |
---|
| 529 | value = str('%s#' % value) |
---|
[7859] | 530 | elif value is None: |
---|
| 531 | # None is not really representable in CSV files |
---|
| 532 | value = '' |
---|
| 533 | return value |
---|
| 534 | |
---|
| 535 | def get_csv_writer(self, filepath=None): |
---|
| 536 | """Get a CSV dict writer instance open for writing. |
---|
| 537 | |
---|
| 538 | Returns a tuple (<writer>, <outfile>) where ``<writer>`` is a |
---|
| 539 | :class:`csv.DictWriter` instance and outfile is the real file |
---|
| 540 | which is written to. The latter is important when writing to |
---|
| 541 | StringIO and can normally be ignored otherwise. |
---|
| 542 | |
---|
| 543 | The returned file will already be filled with the header row. |
---|
| 544 | |
---|
| 545 | Please note that if you give a filepath, the returned outfile |
---|
| 546 | is open for writing only and you might have to close it before |
---|
| 547 | reopening it for reading. |
---|
| 548 | """ |
---|
| 549 | if filepath is None: |
---|
| 550 | outfile = StringIO() |
---|
| 551 | else: |
---|
| 552 | outfile = open(filepath, 'wb') |
---|
[10027] | 553 | writer = unicodecsv.DictWriter(outfile, self.fields) |
---|
[7859] | 554 | writer.writerow(dict(zip(self.fields, self.fields))) # header |
---|
| 555 | return writer, outfile |
---|
| 556 | |
---|
| 557 | def write_item(self, obj, writer): |
---|
| 558 | """Write a row extracted from `obj` into CSV file using `writer`. |
---|
| 559 | """ |
---|
| 560 | row = {} |
---|
| 561 | for name in self.fields: |
---|
| 562 | value = getattr(obj, name, None) |
---|
| 563 | value = self.mangle_value(value, name, obj) |
---|
| 564 | row[name] = value |
---|
| 565 | writer.writerow(row) |
---|
| 566 | return |
---|
| 567 | |
---|
| 568 | def close_outfile(self, filepath, outfile): |
---|
| 569 | """Close outfile. |
---|
| 570 | If filepath is None, the contents of outfile is returned. |
---|
| 571 | """ |
---|
| 572 | outfile.seek(0) |
---|
| 573 | if filepath is None: |
---|
| 574 | return outfile.read() |
---|
| 575 | outfile.close() |
---|
| 576 | return |
---|
| 577 | |
---|
[9797] | 578 | def get_filtered(self, site, **kw): |
---|
| 579 | """Get datasets to export filtered by keyword arguments. |
---|
| 580 | Returns an iterable. |
---|
| 581 | """ |
---|
| 582 | raise NotImplementedError |
---|
| 583 | |
---|
[12516] | 584 | def get_selected(self, site, selected): |
---|
| 585 | """Get datasets to export for selected items |
---|
| 586 | specified by a list of identifiers. |
---|
| 587 | Returns an iterable. |
---|
| 588 | """ |
---|
| 589 | raise NotImplementedError |
---|
| 590 | |
---|
[7859] | 591 | def export(self, iterable, filepath=None): |
---|
| 592 | """Export `iterable` as CSV file. |
---|
| 593 | If `filepath` is ``None``, a raw string with CSV data should |
---|
| 594 | be returned. |
---|
| 595 | """ |
---|
| 596 | raise NotImplementedError |
---|
| 597 | |
---|
| 598 | def export_all(self, site, filepath=None): |
---|
| 599 | """Export all appropriate objects in `site` into `filepath` as |
---|
| 600 | CSV data. |
---|
| 601 | If `filepath` is ``None``, a raw string with CSV data should |
---|
| 602 | be returned. |
---|
| 603 | """ |
---|
| 604 | raise NotImplementedError |
---|
[9217] | 605 | |
---|
[9797] | 606 | def export_filtered(self, site, filepath=None, **kw): |
---|
[12861] | 607 | """Export items denoted by `kw`. |
---|
[9797] | 608 | If `filepath` is ``None``, a raw string with CSV data should |
---|
| 609 | be returned. |
---|
| 610 | """ |
---|
| 611 | data = self.get_filtered(site, **kw) |
---|
| 612 | return self.export(data, filepath=filepath) |
---|
| 613 | |
---|
[12516] | 614 | def export_selected(self, site, filepath=None, **kw): |
---|
| 615 | """Export those items specified by a list of identifiers |
---|
| 616 | called `selected`. |
---|
| 617 | If `filepath` is ``None``, a raw string with CSV data should |
---|
| 618 | be returned. |
---|
| 619 | """ |
---|
| 620 | selected = kw.get('selected', []) |
---|
| 621 | data = self.get_selected(site, selected) |
---|
| 622 | return self.export(data, filepath=filepath) |
---|
| 623 | |
---|
[9797] | 624 | def export_job(site, exporter_name, **kw): |
---|
[9217] | 625 | """Export all entries delivered by exporter and store it in a temp file. |
---|
| 626 | |
---|
| 627 | `site` gives the site to search. It will be passed to the exporter |
---|
| 628 | and also be set as 'current site' as the function is used in |
---|
| 629 | asynchronous jobs which run in their own threads and have no site |
---|
| 630 | set initially. Therefore `site` must also be a valid value for use |
---|
| 631 | with `zope.component.hooks.setSite()`. |
---|
| 632 | |
---|
| 633 | `exporter_name` is the utility name under which the desired |
---|
| 634 | exporter was registered with the ZCA. |
---|
| 635 | |
---|
| 636 | The resulting CSV file will be stored in a new temporary directory |
---|
| 637 | (using :func:`tempfile.mkdtemp`). It will be named after the |
---|
| 638 | exporter used with `.csv` filename extension. |
---|
| 639 | |
---|
| 640 | Returns the path to the created CSV file. |
---|
| 641 | |
---|
| 642 | .. note:: It is the callers responsibility to clean up the used |
---|
| 643 | file and its parent directory. |
---|
| 644 | """ |
---|
| 645 | setSite(site) |
---|
| 646 | exporter = getUtility(ICSVExporter, name=exporter_name) |
---|
| 647 | output_dir = tempfile.mkdtemp() |
---|
| 648 | filename = '%s.csv' % exporter_name |
---|
| 649 | output_path = os.path.join(output_dir, filename) |
---|
[9797] | 650 | if kw == {}: |
---|
| 651 | exporter.export_all(site, filepath=output_path) |
---|
[12516] | 652 | elif kw.has_key('selected'): |
---|
| 653 | exporter.export_selected(site, filepath=output_path, **kw) |
---|
[9797] | 654 | else: |
---|
| 655 | exporter.export_filtered(site, filepath=output_path, **kw) |
---|
[9217] | 656 | return output_path |
---|
| 657 | |
---|
| 658 | class AsyncExportJob(AsyncJob): |
---|
| 659 | """An IJob that exports data to CSV files. |
---|
| 660 | |
---|
| 661 | `AsyncExportJob` instances are regular `AsyncJob` instances with a |
---|
| 662 | different constructor API. Instead of a callable to execute, you |
---|
| 663 | must pass a `site` and some `exporter_name` to trigger an export. |
---|
| 664 | |
---|
| 665 | The real work is done when an instance of this class is put into a |
---|
| 666 | queue. See :mod:`waeup.kofa.async` to learn more about |
---|
| 667 | asynchronous jobs. |
---|
| 668 | |
---|
| 669 | The `exporter_name` must be the name under which an ICSVExporter |
---|
| 670 | utility was registered with the ZCA. |
---|
| 671 | |
---|
| 672 | The `site` must be a valid site or ``None``. |
---|
| 673 | |
---|
| 674 | The result of an `AsyncExportJob` is the path to generated CSV |
---|
| 675 | file. The file will reside in a temporary directory that should be |
---|
| 676 | removed after being used. |
---|
| 677 | """ |
---|
| 678 | grok.implements(IExportJob) |
---|
| 679 | |
---|
[9718] | 680 | def __init__(self, site, exporter_name, *args, **kwargs): |
---|
[9217] | 681 | super(AsyncExportJob, self).__init__( |
---|
[9718] | 682 | export_job, site, exporter_name, *args, **kwargs) |
---|
[9217] | 683 | |
---|
[9816] | 684 | @property |
---|
| 685 | def finished(self): |
---|
| 686 | """A job is marked `finished` if it is completed. |
---|
| 687 | |
---|
| 688 | Please note: a finished report job does not neccessarily |
---|
| 689 | provide an IReport result. See meth:`failed`. |
---|
| 690 | """ |
---|
| 691 | return self.status == zc.async.interfaces.COMPLETED |
---|
| 692 | |
---|
| 693 | @property |
---|
| 694 | def failed(self): |
---|
| 695 | """A report job is marked failed iff it is finished and the |
---|
| 696 | result is None. |
---|
| 697 | |
---|
| 698 | While a job is unfinished, the `failed` status is ``None``. |
---|
| 699 | |
---|
| 700 | Failed jobs normally provide a `traceback` to examine reasons. |
---|
| 701 | """ |
---|
| 702 | if not self.finished: |
---|
| 703 | return None |
---|
| 704 | if getattr(self, 'result', None) is None: |
---|
| 705 | return True |
---|
| 706 | return False |
---|
| 707 | |
---|
[9217] | 708 | class ExportJobContainer(object): |
---|
| 709 | """A mix-in that provides functionality for asynchronous export jobs. |
---|
| 710 | """ |
---|
| 711 | grok.implements(IExportJobContainer) |
---|
| 712 | running_exports = PersistentList() |
---|
| 713 | |
---|
[9718] | 714 | def start_export_job(self, exporter_name, user_id, *args, **kwargs): |
---|
[9217] | 715 | """Start asynchronous export job. |
---|
| 716 | |
---|
| 717 | `exporter_name` is the name of an exporter utility to be used. |
---|
| 718 | |
---|
| 719 | `user_id` is the ID of the user that triggers the export. |
---|
| 720 | |
---|
| 721 | The job_id is stored along with exporter name and user id in a |
---|
| 722 | persistent list. |
---|
| 723 | |
---|
[9718] | 724 | The method supports additional positional and keyword |
---|
| 725 | arguments, which are passed as-is to the respective |
---|
| 726 | :class:`AsyncExportJob`. |
---|
| 727 | |
---|
[9217] | 728 | Returns the job ID of the job started. |
---|
| 729 | """ |
---|
| 730 | site = grok.getSite() |
---|
| 731 | manager = getUtility(IJobManager) |
---|
[9718] | 732 | job = AsyncExportJob(site, exporter_name, *args, **kwargs) |
---|
[9217] | 733 | job_id = manager.put(job) |
---|
| 734 | # Make sure that the persisted list is stored in ZODB |
---|
| 735 | self.running_exports = PersistentList(self.running_exports) |
---|
| 736 | self.running_exports.append((job_id, exporter_name, user_id)) |
---|
| 737 | return job_id |
---|
| 738 | |
---|
| 739 | def get_running_export_jobs(self, user_id=None): |
---|
| 740 | """Get export jobs for user with `user_id` as list of tuples. |
---|
| 741 | |
---|
| 742 | Each tuples holds ``<job_id>, <exporter_name>, <user_id>`` in |
---|
| 743 | that order. The ``<exporter_name>`` is the utility name of the |
---|
| 744 | used exporter. |
---|
| 745 | |
---|
| 746 | If `user_id` is ``None``, all running jobs are returned. |
---|
| 747 | """ |
---|
| 748 | entries = [] |
---|
| 749 | to_delete = [] |
---|
| 750 | manager = getUtility(IJobManager) |
---|
| 751 | for entry in self.running_exports: |
---|
| 752 | if user_id is not None and entry[2] != user_id: |
---|
| 753 | continue |
---|
| 754 | if manager.get(entry[0]) is None: |
---|
| 755 | to_delete.append(entry) |
---|
| 756 | continue |
---|
| 757 | entries.append(entry) |
---|
| 758 | if to_delete: |
---|
| 759 | self.running_exports = PersistentList( |
---|
| 760 | [x for x in self.running_exports if x not in to_delete]) |
---|
| 761 | return entries |
---|
| 762 | |
---|
| 763 | def get_export_jobs_status(self, user_id=None): |
---|
| 764 | """Get running/completed export jobs for `user_id` as list of tuples. |
---|
| 765 | |
---|
| 766 | Each tuple holds ``<raw status>, <status translated>, |
---|
| 767 | <exporter title>`` in that order, where ``<status |
---|
| 768 | translated>`` and ``<exporter title>`` are translated strings |
---|
| 769 | representing the status of the job and the human readable |
---|
| 770 | title of the exporter used. |
---|
| 771 | """ |
---|
| 772 | entries = self.get_running_export_jobs(user_id) |
---|
| 773 | result = [] |
---|
| 774 | manager = getUtility(IJobManager) |
---|
| 775 | for entry in entries: |
---|
| 776 | job = manager.get(entry[0]) |
---|
| 777 | if job is None: |
---|
| 778 | continue |
---|
| 779 | status, status_translated = JOB_STATUS_MAP[job.status] |
---|
| 780 | exporter_name = getUtility(ICSVExporter, name=entry[1]).title |
---|
| 781 | result.append((status, status_translated, exporter_name)) |
---|
| 782 | return result |
---|
| 783 | |
---|
| 784 | def delete_export_entry(self, entry): |
---|
| 785 | """Delete the export denoted by `entry`. |
---|
| 786 | |
---|
| 787 | Removes given entry from the local `running_exports` list and also |
---|
| 788 | removes the regarding job via the local job manager. |
---|
| 789 | |
---|
| 790 | `entry` must be a tuple ``(<job id>, <exporter name>, <user |
---|
| 791 | id>)`` as created by :meth:`start_export_job` or returned by |
---|
| 792 | :meth:`get_running_export_jobs`. |
---|
| 793 | """ |
---|
| 794 | manager = getUtility(IJobManager) |
---|
| 795 | job = manager.get(entry[0]) |
---|
| 796 | if job is not None: |
---|
| 797 | # remove created export file |
---|
| 798 | if isinstance(job.result, basestring): |
---|
| 799 | if os.path.exists(os.path.dirname(job.result)): |
---|
| 800 | shutil.rmtree(os.path.dirname(job.result)) |
---|
| 801 | manager.remove(entry[0], self) |
---|
| 802 | new_entries = [x for x in self.running_exports |
---|
| 803 | if x != entry] |
---|
| 804 | self.running_exports = PersistentList(new_entries) |
---|
| 805 | return |
---|
| 806 | |
---|
| 807 | def entry_from_job_id(self, job_id): |
---|
| 808 | """Get entry tuple for `job_id`. |
---|
| 809 | |
---|
| 810 | Returns ``None`` if no such entry can be found. |
---|
| 811 | """ |
---|
| 812 | for entry in self.running_exports: |
---|
| 813 | if entry[0] == job_id: |
---|
| 814 | return entry |
---|
| 815 | return None |
---|
[9726] | 816 | |
---|
| 817 | class VirtualExportJobContainer(ExportJobContainer): |
---|
| 818 | """A virtual export job container. |
---|
| 819 | |
---|
| 820 | Virtual ExportJobContainers can be used as a mixin just like real |
---|
| 821 | ExportJobContainer. |
---|
| 822 | |
---|
| 823 | They retrieve and store data in the site-wide ExportJobContainer. |
---|
| 824 | |
---|
| 825 | Functionality is currently entirely as for regular |
---|
| 826 | ExportJobContainers, except that data is stored elsewhere. |
---|
| 827 | |
---|
| 828 | VirtualExportJobContainers need a registered |
---|
| 829 | IExportContainerFinder utility to find a suitable container for |
---|
| 830 | storing data. |
---|
| 831 | """ |
---|
| 832 | grok.implements(IExportJobContainer) |
---|
| 833 | |
---|
| 834 | @property |
---|
| 835 | def _site_container(self): |
---|
| 836 | return getUtility(IExportContainerFinder)() |
---|
| 837 | |
---|
| 838 | # The following is a simple trick. While ExportJobContainers store |
---|
| 839 | # only one attribute in ZODB, it is sufficient to replace this |
---|
| 840 | # attribute `running_exports` with a suitable manager to make the |
---|
| 841 | # whole virtual container work like the original but with the data |
---|
| 842 | # stored in the site-wide exports container. This way, virtual |
---|
| 843 | # export containers provide the whole functionality of a regular |
---|
| 844 | # exports container but store no data at all with themselves. |
---|
| 845 | @property |
---|
| 846 | def running_exports(self): |
---|
| 847 | """Exports stored in the site-wide exports container. |
---|
| 848 | """ |
---|
| 849 | return self._site_container.running_exports |
---|
| 850 | |
---|
| 851 | @running_exports.setter |
---|
| 852 | def running_exports(self, value): |
---|
| 853 | self._site_container.running_exports = value |
---|
| 854 | |
---|
| 855 | @running_exports.deleter |
---|
| 856 | def running_exports(self): |
---|
| 857 | del self._site_container.running_exports |
---|
| 858 | |
---|
[9823] | 859 | @property |
---|
| 860 | def logger(self): |
---|
| 861 | return self._site_container.logger |
---|
[9726] | 862 | |
---|
| 863 | @implementer(IExportContainerFinder) |
---|
| 864 | class ExportContainerFinder(grok.GlobalUtility): |
---|
| 865 | """Finder for local (site-wide) export container. |
---|
| 866 | """ |
---|
| 867 | |
---|
| 868 | def __call__(self): |
---|
| 869 | """Get the local export container- |
---|
| 870 | |
---|
| 871 | If no site can be determined or the site provides no export |
---|
| 872 | container, None is returned. |
---|
| 873 | """ |
---|
| 874 | site = grok.getSite() |
---|
| 875 | if site is None: |
---|
| 876 | return None |
---|
| 877 | return site.get('datacenter', None) |
---|