def create_initial_revisions(self, app, model_class, comment, verbosity=2, **kwargs):
     """Creates the set of initial revisions for the given model."""
     # Import the relevant admin module.
     try:
         import_module("%s.admin" % app.__name__.rsplit(".", 1)[0])
     except ImportError:
         pass
     # Check all models for empty revisions.
     if revision.is_registered(model_class):
         created_count = 0
         # HACK: This join can't be done in the database, due to incompatibilities
         # between unicode object_ids and integer pks on strict backends like postgres.
         for obj in model_class._default_manager.iterator():
             if Version.objects.get_for_object(obj).count() == 0:
                 try:
                     self.version_save(obj, comment)
                 except:
                     print "ERROR: Could not save initial version for %s %s." % (model_class.__name__, obj.pk)
                     raise
                 created_count += 1
         # Print out a message, if feeling verbose.
         if created_count > 0 and verbosity >= 2:
             print u"Created %s initial revisions for model %s." % (created_count, model_class._meta.verbose_name)
     else:
         if verbosity >= 2:
             print u"Model %s is not registered."  % (model_class._meta.verbose_name)
 def create_initial_revisions(self, app, model_class, verbosity=2, **kwargs):
     """
     all stolen from management/__init__.py :)
     """
     # Import the relevant admin module.
     try:
         import_module("%s.admin" % app.__name__.rsplit(".", 1)[0])
     except ImportError:
         pass
     # Check all models for empty revisions.
     if revision.is_registered(model_class):
         content_type = ContentType.objects.get_for_model(model_class)
         # Get the id for all models that have not got at least one revision.
         # HACK: This join can't be done in the database, for potential incompatibilities
         # between unicode object_ids and integer pks on strict backends like postgres.
         versioned_ids = frozenset(Version.objects.filter(content_type=content_type).values_list("object_id", flat=True).distinct().iterator())
         all_ids = frozenset(unicode(id) for id in model_class._default_manager.values_list("pk", flat=True).iterator())
         unversioned_ids = all_ids - versioned_ids
         # Create the initial revision for all unversioned models.
         created_count = 0
         for unversioned_obj in model_class._default_manager.filter(pk__in=unversioned_ids).iterator():
             self.version_save(unversioned_obj)
             created_count += 1
         # Print out a message, if feeling verbose.
         if created_count > 0 and verbosity >= 2:
             self.stdout.write(u"Created %s initial revisions for model %s.\n" % (created_count, model_class._meta.verbose_name))
     else:
         if verbosity >= 2:
             self.stdout.write(u"Model %s is not registered.\n"  % (model_class._meta.verbose_name))
Example #3
0
 def create_initial_revisions(self,
                              app,
                              model_class,
                              comment,
                              verbosity=2,
                              **kwargs):
     """Creates the set of initial revisions for the given model."""
     # Import the relevant admin module.
     try:
         import_module("%s.admin" % app.__name__.rsplit(".", 1)[0])
     except ImportError:
         pass
     # Check all models for empty revisions.
     if revision.is_registered(model_class):
         created_count = 0
         # HACK: This join can't be done in the database, due to incompatibilities
         # between unicode object_ids and integer pks on strict backends like postgres.
         for obj in model_class._default_manager.iterator():
             if Version.objects.get_for_object(obj).count() == 0:
                 try:
                     self.version_save(obj, comment)
                 except:
                     print "ERROR: Could not save initial version for %s %s." % (
                         model_class.__name__, obj.pk)
                     raise
                 created_count += 1
         # Print out a message, if feeling verbose.
         if created_count > 0 and verbosity >= 2:
             print u"Created %s initial revisions for model %s." % (
                 created_count, model_class._meta.verbose_name)
     else:
         if verbosity >= 2:
             print u"Model %s is not registered." % (
                 model_class._meta.verbose_name)
Example #4
0
def create_initial_revisions(app, verbosity=2, **kwargs):
    """
    Post-syncdb hook to create an initial revision for all registered models.
    """
    # Import the relevant admin module.
    try:
        import_module("%s.admin" % app.__name__.rsplit(".", 1)[0])
    except ImportError:
        pass
    # Check all models for empty revisions.
    for model_class in models.get_models(app):
        if revision.is_registered(model_class):
            content_type = ContentType.objects.get_for_model(model_class)
            # Get the id for all models that have not got at least one revision.
            # HACK: This join can't be done in the database, for potential incompatibilities
            # between unicode object_ids and integer pks on strict backends like postgres.
            versioned_ids = frozenset(
                Version.objects.filter(content_type=content_type).values_list(
                    "object_id", flat=True).distinct().iterator())
            all_ids = frozenset(
                unicode(id) for id in model_class._default_manager.values_list(
                    "pk", flat=True).iterator())
            unversioned_ids = all_ids - versioned_ids
            # Create the initial revision for all unversioned models.
            created_count = 0
            for unversioned_obj in model_class._default_manager.filter(
                    pk__in=unversioned_ids).iterator():
                version_save(unversioned_obj)
                created_count += 1
            # Print out a message, if feeling verbose.
            if created_count > 0 and verbosity >= 2:
                print u"Created %s initial revisions for model %s." % (
                    created_count, model_class._meta.verbose_name)
Example #5
0
def post_save_model(sender, instance, created, **kwargs):
    """
    Функция для тригера post-save-model
    Сейчас нужна лишь для сохранения openness_first.

    """
    must_register = False
    if revision.is_registered(instance.__class__):
        revision.unregister(instance.__class__)
        must_register = True
    #update task openness hook
    if instance.__class__ == Monitoring:
        for task in Task.objects.filter(organization__monitoring=instance):
            task.update_openness()
    if must_register:
        revision.register(instance.__class__)
Example #6
0
def monitoring_parameter_import(request, monitoring_id):
    """
    Импорт параметров из CSV.

    """
    from reversion import revision
    must_register = False
    if revision.is_registered(Parameter):
        revision.unregister(Parameter)
        must_register = True

    monitoring = get_object_or_404(Monitoring, pk=monitoring_id)
    if not request.user.has_perm('exmo2010.edit_monitoring', monitoring):
        return HttpResponseForbidden(_('Forbidden'))
    if not 'paramfile' in request.FILES:
        return HttpResponseRedirect(reverse('exmo2010:monitoring_list'))
    reader = UnicodeReader(request.FILES['paramfile'])
    errLog = []
    rowOKCount = 0
    rowALLCount = 0
    try:
        for row in reader:
            rowALLCount += 1
            if row[0].startswith('#'):
                errLog.append("row %d. Starts with '#'. Skipped" % reader.line_num)
                continue
            if row[0] == '':
                errLog.append("row %d (csv). Empty code" % reader.line_num)
                continue
            if row[1] == '':
                errLog.append("row %d (csv). Empty name" % reader.line_num)
                continue
            try:
                code = row[0]
                name = row[1]
                parameter = Parameter.objects.get(monitoring=monitoring,
                    code=code, name=name)
            except Parameter.DoesNotExist:
                parameter = Parameter()
            except Exception, e:
                errLog.append("row %d. %s" % (reader.line_num, e))
                continue
            try:
                parameter.monitoring = monitoring
                parameter.code = code
                parameter.name = name
                # Присваиваем пустую строку, а не None.
                parameter.description = row[2] or ''
                parameter.complete = bool(int(row[3]))
                parameter.topical = bool(int(row[4]))
                parameter.accessible = bool(int(row[5]))
                parameter.hypertext = bool(int(row[6]))
                parameter.document = bool(int(row[7]))
                parameter.image = bool(int(row[8]))
                parameter.weight = row[9]
                parameter.keywords = row[10]
                parameter.full_clean()
                parameter.save()
            except ValidationError, e:
                errLog.append("row %d (validation). %s" % (
                    reader.line_num,
                    '; '.join(['%s: %s' % (i[0], ', '.join(i[1])) for i in e.message_dict.items()])))
            except Exception, e:
                errLog.append("row %d. %s" % (reader.line_num, e))
Example #7
0
def monitoring_organization_import(request, monitoring_id):
    """
    Импорт организаций из CSV.

    """
    from reversion import revision
    must_register = False
    if revision.is_registered(Organization):
        revision.unregister(Organization)
        must_register = True

    monitoring = get_object_or_404(Monitoring, pk=monitoring_id)
    if not request.user.has_perm('exmo2010.edit_monitoring', monitoring):
        return HttpResponseForbidden(_('Forbidden'))
    if not 'orgfile' in request.FILES:
        return HttpResponseRedirect(reverse('exmo2010:monitoring_list'))
    reader = UnicodeReader(request.FILES['orgfile'])
    errLog = []
    indexes = {}
    rowOKCount = 0
    rowALLCount = 0

    try:
        for row in reader:
            if rowALLCount == 0 and row[0] and row[0].startswith('#'):
                for key in ['name', 'url', 'email', 'phone', 'comments', 'keywords']:
                    for item in row:
                        if item and key in item.lower():
                            indexes[key] = row.index(item)
                continue

            if not 'name' in indexes:
                errLog.append("header row (csv). Field 'Name' does not exist")
                break

            rowALLCount += 1

            if row[0] and row[0].startswith('#'):
                errLog.append("row %d. Starts with '#'. Skipped" % rowALLCount)
                continue
            if row[indexes['name']] == '':
                errLog.append("row %d (csv). Empty organization name" % rowALLCount)
                continue
            try:
                organization = Organization.objects.get(monitoring=monitoring, name=row[indexes['name']])
            except Organization.DoesNotExist:
                organization = Organization()
                organization.monitoring = monitoring
            except Exception, e:
                errLog.append("row %d. %s" % (rowALLCount, e))
                continue
            try:
                for key in indexes.keys():
                    cell = row[indexes[key]]
                    if key in ['email', 'phone'] and cell:
                        cell = replace_string(cell)
                    setattr(organization, key, cell.strip() if cell else '')
                organization.inv_code = generate_inv_code(6)
                organization.full_clean()
                organization.save()
            except ValidationError, e:
                errLog.append("row %d (validation). %s" % (
                    rowALLCount,
                    '; '.join(['%s: %s' % (i[0], ', '.join(i[1])) for i in e.message_dict.items()])))
            except Exception, e:
                errLog.append("row %d. %s" % (rowALLCount, e))