udata 9.1.2.dev30355__py2.py3-none-any.whl → 9.1.2.dev30382__py2.py3-none-any.whl

udata/commands/db.py

@@ -1,87 +1,85 @@
 import collections
-from itertools import groupby
 import logging
 import os
 import traceback
+from itertools import groupby
 
-from bson import DBRef
 import click
 import mongoengine
+from bson import DBRef
 
-from udata import migrations, models as core_models
+from udata import migrations
+from udata import models as core_models
 from udata.api import oauth2 as oauth2_models
-from udata.commands import cli, green, yellow, cyan, red, magenta, white, echo
+from udata.commands import cli, cyan, echo, green, magenta, red, white, yellow
 from udata.harvest import models as harvest_models
 from udata.mongo import db
 
-
 # Date format used to for display
-DATE_FORMAT = '%Y-%m-%d %H:%M'
+DATE_FORMAT = "%Y-%m-%d %H:%M"
 
 log = logging.getLogger(__name__)
 
 
-@cli.group('db')
+@cli.group("db")
 def grp():
-    '''Database related operations'''
+    """Database related operations"""
     pass
 
 
 def log_status(migration, status):
-    '''Properly display a migration status line'''
+    """Properly display a migration status line"""
     name = os.path.splitext(migration.filename)[0]
-    display = ':'.join((migration.plugin, name)) + ' '
-    log.info('%s [%s]', '{:.<70}'.format(display), status)
+    display = ":".join((migration.plugin, name)) + " "
+    log.info("%s [%s]", "{:.<70}".format(display), status)
 
 
 def status_label(record):
     if record.ok:
         return green(record.last_date.strftime(DATE_FORMAT))
     elif not record.exists():
-        return yellow('Not applied')
+        return yellow("Not applied")
     else:
         return red(record.status)
 
 
 def format_output(output, success=True, traceback=None):
-    echo('  │')
+    echo("  │")
     for level, msg in output:
-        echo('  │ {0}'.format(msg))
-    echo('  │')
+        echo("  │ {0}".format(msg))
+    echo("  │")
     if traceback:
-        for tb in traceback.split('\n'):
-            echo('  │ {0}'.format(tb))
-    echo('  │')
-    echo('  └──[{0}]'.format(green('OK') if success else red('KO')))
-    echo('')
+        for tb in traceback.split("\n"):
+            echo("  │ {0}".format(tb))
+    echo("  │")
+    echo("  └──[{0}]".format(green("OK") if success else red("KO")))
+    echo("")
 
 
 @grp.command()
 def status():
-    '''Display the database migrations status'''
+    """Display the database migrations status"""
     for migration in migrations.list_available():
         log_status(migration, status_label(migration.record))
 
 
 @grp.command()
-@click.option('-r', '--record', is_flag=True,
-              help='Only records the migrations')
-@click.option('-d', '--dry-run', is_flag=True,
-              help='Only print migrations to be applied')
+@click.option("-r", "--record", is_flag=True, help="Only records the migrations")
+@click.option("-d", "--dry-run", is_flag=True, help="Only print migrations to be applied")
 def migrate(record, dry_run=False):
-    '''Perform database migrations'''
+    """Perform database migrations"""
     success = True
     for migration in migrations.list_available():
         if migration.record.ok or not success:
-            log_status(migration, cyan('Skipped'))
+            log_status(migration, cyan("Skipped"))
         else:
-            status = magenta('Recorded') if record else yellow('Apply')
+            status = magenta("Recorded") if record else yellow("Apply")
             log_status(migration, status)
             try:
                 output = migration.execute(recordonly=record, dryrun=dry_run)
             except migrations.RollbackError as re:
                 format_output(re.migrate_exc.output, False)
-                log_status(migration, red('Rollback'))
+                log_status(migration, red("Rollback"))
                 format_output(re.output, not re.exc)
                 success = False
             except migrations.MigrationError as me:
@@ -93,10 +91,10 @@ def migrate(record, dry_run=False):
 
 
 @grp.command()
-@click.argument('plugin_or_specs')
-@click.argument('filename', default=None, required=False, metavar='[FILENAME]')
+@click.argument("plugin_or_specs")
+@click.argument("filename", default=None, required=False, metavar="[FILENAME]")
 def unrecord(plugin_or_specs, filename):
-    '''
+    """
     Remove a database migration record.
 
     \b
@@ -105,155 +103,202 @@ def unrecord(plugin_or_specs, filename):
      - plugin filename.js
      - plugin:filename
      - plugin:fliename.js
-    '''
+    """
     migration = migrations.get(plugin_or_specs, filename)
     removed = migration.unrecord()
     if removed:
-        log.info('Removed migration %s', migration.label)
+        log.info("Removed migration %s", migration.label)
     else:
-        log.error('Migration not found %s', migration.label)
+        log.error("Migration not found %s", migration.label)
 
 
 @grp.command()
-@click.argument('plugin_or_specs')
-@click.argument('filename', default=None, required=False, metavar='[FILENAME]')
+@click.argument("plugin_or_specs")
+@click.argument("filename", default=None, required=False, metavar="[FILENAME]")
 def info(plugin_or_specs, filename):
-    '''
+    """
     Display detailed info about a migration
-    '''
+    """
     migration = migrations.get(plugin_or_specs, filename)
     log_status(migration, status_label(migration.record))
     try:
         echo(migration.module.__doc__)
     except migrations.MigrationError:
-        echo(yellow('Module not found'))
+        echo(yellow("Module not found"))
 
-    for op in migration.record.get('ops', []):
+    for op in migration.record.get("ops", []):
         display_op(op)
 
 
 def display_op(op):
-    timestamp = white(op['date'].strftime(DATE_FORMAT))
-    label = white(op['type'].title()) + ' '
-    echo('{label:.<70} [{date}]'.format(label=label, date=timestamp))
-    format_output(op['output'], success=op['success'], traceback=op.get('traceback'))
+    timestamp = white(op["date"].strftime(DATE_FORMAT))
+    label = white(op["type"].title()) + " "
+    echo("{label:.<70} [{date}]".format(label=label, date=timestamp))
+    format_output(op["output"], success=op["success"], traceback=op.get("traceback"))
+
 
 def check_references(models_to_check):
     # Cannot modify local scope from Python… :-(
-    class Log: errors = []
+    class Log:
+        errors = []
 
     def print_and_save(text: str):
         Log.errors.append(text.strip())
-        print(text) 
+        print(text)
 
     errors = collections.defaultdict(int)
 
     _models = []
     for models in core_models, harvest_models, oauth2_models:
         _models += [
-            elt for _, elt in models.__dict__.items()
+            elt
+            for _, elt in models.__dict__.items()
             if isinstance(elt, type) and issubclass(elt, (db.Document))
         ]
 
     references = []
     for model in set(_models):
-        if model.__name__ == 'Activity':
-            print(f'Skipping Activity model, scheduled for deprecation')
+        if model.__name__ == "Activity":
+            print(f"Skipping Activity model, scheduled for deprecation")
             continue
-        if model.__name__ == 'GeoLevel':
-            print(f'Skipping GeoLevel model, scheduled for deprecation')
+        if model.__name__ == "GeoLevel":
+            print(f"Skipping GeoLevel model, scheduled for deprecation")
             continue
 
         if models_to_check and model.__name__ not in models_to_check:
             continue
 
         # find "root" ReferenceField fields
-        refs = [elt for elt in model._fields.values()
-                if isinstance(elt, mongoengine.fields.ReferenceField)]
-        references += [{
-            'model': model,
-            'repr': f'{model.__name__}.{r.name}',
-            'name': r.name,
-            'destination': r.document_type.__name__,
-            'type': 'direct',
-        } for r in refs]
+        refs = [
+            elt
+            for elt in model._fields.values()
+            if isinstance(elt, mongoengine.fields.ReferenceField)
+        ]
+        references += [
+            {
+                "model": model,
+                "repr": f"{model.__name__}.{r.name}",
+                "name": r.name,
+                "destination": r.document_type.__name__,
+                "type": "direct",
+            }
+            for r in refs
+        ]
 
         # find "root" GenericReferenceField
-        refs = [elt for elt in model._fields.values()
-                if isinstance(elt, mongoengine.fields.GenericReferenceField)]
-        references += [{
-            'model': model,
-            'repr': f'{model.__name__}.{r.name}',
-            'name': r.name,
-            'destination': 'Generic',
-            'type': 'direct',
-        } for r in refs]
+        refs = [
+            elt
+            for elt in model._fields.values()
+            if isinstance(elt, mongoengine.fields.GenericReferenceField)
+        ]
+        references += [
+            {
+                "model": model,
+                "repr": f"{model.__name__}.{r.name}",
+                "name": r.name,
+                "destination": "Generic",
+                "type": "direct",
+            }
+            for r in refs
+        ]
 
         # find ListField with ReferenceField
-        list_fields = [elt for elt in model._fields.values()
-                       if isinstance(elt, mongoengine.fields.ListField)]
-        list_refs = [elt for elt in list_fields
-                     if isinstance(elt.field, mongoengine.fields.ReferenceField)]
-        references += [{
-            'model': model,
-            'repr': f'{model.__name__}.{lr.name}',
-            'name': lr.name,
-            'destination': lr.field.document_type.__name__,
-            'type': 'list',
-        } for lr in list_refs]
+        list_fields = [
+            elt for elt in model._fields.values() if isinstance(elt, mongoengine.fields.ListField)
+        ]
+        list_refs = [
+            elt for elt in list_fields if isinstance(elt.field, mongoengine.fields.ReferenceField)
+        ]
+        references += [
+            {
+                "model": model,
+                "repr": f"{model.__name__}.{lr.name}",
+                "name": lr.name,
+                "destination": lr.field.document_type.__name__,
+                "type": "list",
+            }
+            for lr in list_refs
+        ]
 
         # find ListField w/ EmbeddedDocumentField w/ ReferenceField
-        list_embeds = [(elt, elt.field) for elt in list_fields
-                       if isinstance(elt.field, mongoengine.fields.EmbeddedDocumentField)]
+        list_embeds = [
+            (elt, elt.field)
+            for elt in list_fields
+            if isinstance(elt.field, mongoengine.fields.EmbeddedDocumentField)
+        ]
         for embed, embed_field in list_embeds:
-            embed_refs = [elt for elt in embed_field.document_type_obj._fields.values()
-                          if isinstance(elt, mongoengine.fields.ReferenceField)]
-            references += [{
-                'model': model,
-                'repr': f'{model.__name__}.{embed.name}__{er.name}',
-                'name': f'{embed.name}__{er.name}',
-                'destination': er.document_type.__name__,
-                'type': 'embed_list',
-            } for er in embed_refs]
+            embed_refs = [
+                elt
+                for elt in embed_field.document_type_obj._fields.values()
+                if isinstance(elt, mongoengine.fields.ReferenceField)
+            ]
+            references += [
+                {
+                    "model": model,
+                    "repr": f"{model.__name__}.{embed.name}__{er.name}",
+                    "name": f"{embed.name}__{er.name}",
+                    "destination": er.document_type.__name__,
+                    "type": "embed_list",
+                }
+                for er in embed_refs
+            ]
 
         # find EmbeddedDocumentField w/ ReferenceField
-        embeds = [elt for elt in model._fields.values()
-                  if isinstance(elt, mongoengine.fields.EmbeddedDocumentField)]
+        embeds = [
+            elt
+            for elt in model._fields.values()
+            if isinstance(elt, mongoengine.fields.EmbeddedDocumentField)
+        ]
         for embed_field in embeds:
-            embed_refs = [elt for elt in embed_field.document_type_obj._fields.values()
-                          if isinstance(elt, mongoengine.fields.ReferenceField)]
-            references += [{
-                'model': model,
-                'repr': f'{model.__name__}.{embed_field.name}__{er.name}',
-                'name': f'{embed_field.name}__{er.name}',
-                'destination': er.document_type.__name__,
-                'type': 'embed',
-            } for er in embed_refs]
+            embed_refs = [
+                elt
+                for elt in embed_field.document_type_obj._fields.values()
+                if isinstance(elt, mongoengine.fields.ReferenceField)
+            ]
+            references += [
+                {
+                    "model": model,
+                    "repr": f"{model.__name__}.{embed_field.name}__{er.name}",
+                    "name": f"{embed_field.name}__{er.name}",
+                    "destination": er.document_type.__name__,
+                    "type": "embed",
+                }
+                for er in embed_refs
+            ]
 
         # find EmbeddedDocumentField w/ ListField w/ ReferenceField
         for embed_field in embeds:
-            embed_lists = [elt for elt in embed_field.document_type_obj._fields.values()
-                           if isinstance(elt, mongoengine.fields.ListField)]
-            elists_refs = [elt for elt in embed_lists
-                           if isinstance(elt.field, mongoengine.fields.ReferenceField)]
-            references += [{
-                'model': model,
-                'repr': f'{model.__name__}.{embed_field.name}__{lr.name}',
-                'name': f'{embed_field.name}__{lr.name}',
-                'destination': lr.field.document_type.__name__,
-                'type': 'embed_list_ref',
-            } for lr in elists_refs]
-
-    print('Those references will be inspected:')
+            embed_lists = [
+                elt
+                for elt in embed_field.document_type_obj._fields.values()
+                if isinstance(elt, mongoengine.fields.ListField)
+            ]
+            elists_refs = [
+                elt
+                for elt in embed_lists
+                if isinstance(elt.field, mongoengine.fields.ReferenceField)
+            ]
+            references += [
+                {
+                    "model": model,
+                    "repr": f"{model.__name__}.{embed_field.name}__{lr.name}",
+                    "name": f"{embed_field.name}__{lr.name}",
+                    "destination": lr.field.document_type.__name__,
+                    "type": "embed_list_ref",
+                }
+                for lr in elists_refs
+            ]
+
+    print("Those references will be inspected:")
     for reference in references:
         print(f'- {reference["repr"]}({reference["destination"]}) — {reference["type"]}')
-    print('')
+    print("")
 
     total = 0
     for model, model_references in groupby(references, lambda i: i["model"]):
         model_references = list(model_references)
         count = model.objects.count()
-        print(f'- doing {count} {model.__name__}…')
+        print(f"- doing {count} {model.__name__}…")
         errors[model] = {}
 
         qs = model.objects().no_cache().all()
@@ -265,71 +310,87 @@ def check_references(models_to_check):
                         errors[model][key] = 0
 
                     try:
-                        if reference['type'] == 'direct':
+                        if reference["type"] == "direct":
                             try:
-                                _ = getattr(obj, reference['name'])
+                                _ = getattr(obj, reference["name"])
                             except mongoengine.errors.DoesNotExist:
                                 errors[model][key] += 1
-                                print_and_save(f'\t{model.__name__}#{obj.id} have a broken reference for `{reference["name"]}`')
-                        elif reference['type'] == 'list':
-                            attr_list = getattr(obj, reference['name'], [])
+                                print_and_save(
+                                    f'\t{model.__name__}#{obj.id} have a broken reference for `{reference["name"]}`'
+                                )
+                        elif reference["type"] == "list":
+                            attr_list = getattr(obj, reference["name"], [])
                             for i, sub in enumerate(attr_list):
                                 # If it's still an instance of DBRef it means that it failed to
                                 # dereference the ID.
                                 if isinstance(sub, DBRef):
                                     errors[model][key] += 1
-                                    print_and_save(f'\t{model.__name__}#{obj.id} have a broken reference for {reference["name"]}[{i}]')
-                        elif reference['type'] == 'embed_list':
-                            p1, p2 = reference['name'].split('__')
+                                    print_and_save(
+                                        f'\t{model.__name__}#{obj.id} have a broken reference for {reference["name"]}[{i}]'
+                                    )
+                        elif reference["type"] == "embed_list":
+                            p1, p2 = reference["name"].split("__")
                             attr_list = getattr(obj, p1, [])
                             for i, sub in enumerate(attr_list):
                                 try:
                                     getattr(sub, p2)
                                 except mongoengine.errors.DoesNotExist:
                                     errors[model][key] += 1
-                                    print_and_save(f'\t{model.__name__}#{obj.id} have a broken reference for {p1}[{i}].{p2}')
-                        elif reference['type'] == 'embed':
-                            p1, p2 = reference['name'].split('__')
+                                    print_and_save(
+                                        f"\t{model.__name__}#{obj.id} have a broken reference for {p1}[{i}].{p2}"
+                                    )
+                        elif reference["type"] == "embed":
+                            p1, p2 = reference["name"].split("__")
                             sub = getattr(obj, p1)
-                            if sub is None: continue
+                            if sub is None:
+                                continue
                             try:
                                 getattr(sub, p2)
                             except mongoengine.errors.DoesNotExist:
                                 errors[model][key] += 1
-                                print_and_save(f'\t{model.__name__}#{obj.id} have a broken reference for {p1}.{p2}')
-                        elif reference['type'] == 'embed_list_ref':
-                            p1, p2 = reference['name'].split('__')
+                                print_and_save(
+                                    f"\t{model.__name__}#{obj.id} have a broken reference for {p1}.{p2}"
+                                )
+                        elif reference["type"] == "embed_list_ref":
+                            p1, p2 = reference["name"].split("__")
                             a = getattr(obj, p1)
-                            if a is None: continue
+                            if a is None:
+                                continue
                             sub = getattr(a, p2, [])
                             for i, child in enumerate(sub):
                                 # If it's still an instance of DBRef it means that it failed to
                                 # dereference the ID.
                                 if isinstance(child, DBRef):
                                     errors[model][key] += 1
-                                    print_and_save(f'\t{model.__name__}#{obj.id} have a broken reference for {p1}.{p2}[{i}]')
+                                    print_and_save(
+                                        f"\t{model.__name__}#{obj.id} have a broken reference for {p1}.{p2}[{i}]"
+                                    )
                         else:
                             print_and_save(f'Unknown ref type {reference["type"]}')
                     except mongoengine.errors.FieldDoesNotExist as e:
-                        print_and_save(f'[ERROR for {model.__name__} {obj.id}] {traceback.format_exc()}')
+                        print_and_save(
+                            f"[ERROR for {model.__name__} {obj.id}] {traceback.format_exc()}"
+                        )
 
         for key, nb_errors in errors[model].items():
-            print(f'{key}: {nb_errors}')
+            print(f"{key}: {nb_errors}")
             total += nb_errors
 
-    print(f'\n Total errors: {total}')
+    print(f"\n Total errors: {total}")
 
     if total > 0:
         try:
             import sentry_sdk
+
             with sentry_sdk.push_scope() as scope:
                 scope.set_extra("errors", Log.errors)
                 sentry_sdk.capture_message(f"{total} integrity errors", "fatal")
         except ImportError:
             print("`sentry_sdk` not installed. The errors weren't reported")
 
+
 @grp.command()
-@click.option('--models', multiple=True, default=[], help='Model(s) to check')
+@click.option("--models", multiple=True, default=[], help="Model(s) to check")
 def check_integrity(models):
-    '''Check the integrity of the database from a business perspective'''
+    """Check the integrity of the database from a business perspective"""
     check_references(models)

udata/commands/dcat.py

@@ -2,50 +2,54 @@ import logging
 
 import click
 import mongoengine
-
 from rdflib import Graph
 
-from udata.commands import cli, green, yellow, cyan, echo, magenta
+from udata.commands import cli, cyan, echo, green, magenta, yellow
 from udata.core.dataset.factories import DatasetFactory
 from udata.core.dataset.rdf import dataset_from_rdf
-from udata.harvest.backends.dcat import DcatBackend, CswDcatBackend, CswIso19139DcatBackend
+from udata.harvest.backends.dcat import (
+    CswDcatBackend,
+    CswIso19139DcatBackend,
+    DcatBackend,
+)
 from udata.rdf import namespace_manager
 
 log = logging.getLogger(__name__)
 
 
-@cli.group('dcat')
+@cli.group("dcat")
 def grp():
-    '''DCAT diagnosis operations'''
+    """DCAT diagnosis operations"""
     pass
 
 
 @grp.command()
-@click.argument('url')
-@click.option('-q', '--quiet', is_flag=True, help='Ignore warnings')
-@click.option('-r', '--rid', help='Inspect specific remote id (contains)')
-@click.option('-c', '--csw', is_flag=True, help='The target is a CSW endpoint with DCAT output')
-@click.option('-i', '--iso', is_flag=True, help='The target is a CSW endpoint with ISO output')
-def parse_url(url, csw, iso, quiet=False, rid=''):
-    '''Parse the datasets in a DCAT format located at URL (debug)'''
+@click.argument("url")
+@click.option("-q", "--quiet", is_flag=True, help="Ignore warnings")
+@click.option("-r", "--rid", help="Inspect specific remote id (contains)")
+@click.option("-c", "--csw", is_flag=True, help="The target is a CSW endpoint with DCAT output")
+@click.option("-i", "--iso", is_flag=True, help="The target is a CSW endpoint with ISO output")
+def parse_url(url, csw, iso, quiet=False, rid=""):
+    """Parse the datasets in a DCAT format located at URL (debug)"""
     if quiet:
-        verbose_loggers = ['rdflib', 'udata.core.dataset']
+        verbose_loggers = ["rdflib", "udata.core.dataset"]
         [logging.getLogger(l).setLevel(logging.ERROR) for l in verbose_loggers]
 
     class MockSource:
-        url = ''
+        url = ""
 
     class MockJob:
         items = []
 
     class MockDatasetFactory(DatasetFactory):
-        '''Use DatasetFactory without .save()'''
+        """Use DatasetFactory without .save()"""
+
         @classmethod
         def _create(cls, model_class, *args, **kwargs):
             instance = model_class(*args, **kwargs)
             return instance
 
-    echo(cyan('Parsing url {}'.format(url)))
+    echo(cyan("Parsing url {}".format(url)))
     source = MockSource()
     source.url = url
     if csw:
@@ -56,7 +60,7 @@ def parse_url(url, csw, iso, quiet=False, rid=''):
         backend = DcatBackend(source, dryrun=True)
     backend.job = MockJob()
     format = backend.get_format()
-    echo(yellow('Detected format: {}'.format(format)))
+    echo(yellow("Detected format: {}".format(format)))
     graphs = backend.parse_graph(url, format)
 
     # serialize/unserialize graph like in the job mechanism
@@ -68,25 +72,27 @@ def parse_url(url, csw, iso, quiet=False, rid=''):
 
     for item in backend.job.items:
         if not rid or rid in item.remote_id:
-            echo(magenta('Processing item {}'.format(item.remote_id)))
-            echo('Item kwargs: {}'.format(yellow(item.kwargs)))
+            echo(magenta("Processing item {}".format(item.remote_id)))
+            echo("Item kwargs: {}".format(yellow(item.kwargs)))
             node = backend.get_node_from_item(graph, item)
             dataset = MockDatasetFactory()
             dataset = dataset_from_rdf(graph, dataset, node=node)
-            echo('')
-            echo(green('Dataset found!'))
-            echo('Title: {}'.format(yellow(dataset)))
-            echo('License: {}'.format(yellow(dataset.license)))
-            echo('Description: {}'.format(yellow(dataset.description)))
-            echo('Tags: {}'.format(yellow(dataset.tags)))
-            echo('Resources: {}'.format(yellow(
-                [(r.title, r.format, r.url) for r in dataset.resources]
-            )))
+            echo("")
+            echo(green("Dataset found!"))
+            echo("Title: {}".format(yellow(dataset)))
+            echo("License: {}".format(yellow(dataset.license)))
+            echo("Description: {}".format(yellow(dataset.description)))
+            echo("Tags: {}".format(yellow(dataset.tags)))
+            echo(
+                "Resources: {}".format(
+                    yellow([(r.title, r.format, r.url) for r in dataset.resources])
+                )
+            )
 
             try:
                 dataset.validate()
             except mongoengine.errors.ValidationError as e:
                 log.error(e, exc_info=True)
             else:
-                echo(green('Dataset is valid ✅'))
-            echo('')
+                echo(green("Dataset is valid ✅"))
+            echo("")