udata 9.1.3.dev30862__py2.py3-none-any.whl → 9.1.3.dev30920__py2.py3-none-any.whl

tasks/helpers.py

@@ -1,10 +1,11 @@
 from os.path import abspath, dirname, join
+from typing import Callable
 
 #: Project absolute root path
 ROOT = abspath(join(dirname(__file__), ".."))
 
 
-def color(code):
+def color(code: str) -> Callable:
     """A simple ANSI color wrapper factory"""
     return lambda t: "\033[{0}{1}\033[0;m".format(code, t)
 
@@ -17,25 +18,25 @@ purple = color("1;35m")
 white = color("1;39m")
 
 
-def header(text, *args, **kwargs):
+def header(text: str, *args, **kwargs) -> None:
     """Display an header"""
     text = text.format(*args, **kwargs)
     print(" ".join((blue(">>"), cyan(text))))
 
 
-def info(text, *args, **kwargs):
+def info(text: str, *args, **kwargs) -> None:
     """Display informations"""
     text = text.format(*args, **kwargs)
     print(" ".join((purple(">>>"), text)))
 
 
-def success(text, *args, **kwargs):
+def success(text: str, *args, **kwargs) -> None:
     """Display a success message"""
     text = text.format(*args, **kwargs)
     print(" ".join((green("✔"), white(text))))
 
 
-def error(text, *args, **kwargs):
+def error(text: str, *args, **kwargs) -> None:
     """Display an error message"""
     text = text.format(*args, **kwargs)
     print(red("✘ {0}".format(text)))

udata/api_fields.py

@@ -335,7 +335,7 @@ def wrap_primary_key(
     field_name: str,
     foreign_field: mongoengine.fields.ReferenceField | mongoengine.fields.GenericReferenceField,
     value: str,
-    document_type: type = None,
+    document_type=None,
 ):
     """
     We need to wrap the `String` inside an `ObjectId` most of the time. If the foreign ID is a `String` we need to get

udata/assets.py

@@ -2,7 +2,7 @@ from flask import current_app, url_for
 from flask_cdn import url_for as cdn_url_for
 
 
-def cdn_for(endpoint, **kwargs):
+def cdn_for(endpoint: str, **kwargs) -> str:
     """
     Get a CDN URL for a static assets.
 

udata/commands/fixtures.py

@@ -39,7 +39,7 @@ COMMUNITY_RES_URL = "/api/1/datasets/community_resources"
 DISCUSSION_URL = "/api/1/discussions"
 
 
-DEFAULT_FIXTURE_FILE_TAG: str = "v1.0.0"
+DEFAULT_FIXTURE_FILE_TAG: str = "v2.0.0"
 DEFAULT_FIXTURE_FILE: str = f"https://raw.githubusercontent.com/opendatateam/udata-fixtures/{DEFAULT_FIXTURE_FILE_TAG}/results.json"  # noqa
 
 DEFAULT_FIXTURES_RESULTS_FILENAME: str = "results.json"
@@ -198,8 +198,7 @@ def import_fixtures(source):
                         MessageDiscussionFactory(**message, posted_by=user) for message in messages
                     ],
                 )
-            for dataservice in fixture.get("dataservices", []):
-                # We may be missing dataservices in older fixtures.
+            for dataservice in fixture["dataservices"]:
                 dataservice = remove_unwanted_keys(dataservice, "dataservice")
                 if not dataservice["contact_point"]:
                     DataserviceFactory(**dataservice, datasets=[dataset])

udata/core/dataset/actions.py

@@ -4,12 +4,12 @@ from datetime import datetime
 from flask import current_app, render_template
 
 from udata import i18n
-from udata.models import Discussion, Message
+from udata.models import Dataset, Discussion, Message
 
 log = logging.getLogger(__name__)
 
 
-def archive(dataset, comment=False):
+def archive(dataset: Dataset, comment=False) -> None:
     """Archive a dataset"""
     if dataset.archived:
         log.warning("Dataset %s already archived, bumping date", dataset)

udata/core/reuse/metrics.py

@@ -3,5 +3,5 @@ from udata.models import Reuse
 
 @Reuse.on_create.connect
 @Reuse.on_update.connect
-def update_reuses_dataset_metric(reuse, **kwargs):
+def update_reuses_dataset_metric(reuse: Reuse, **kwargs) -> None:
     reuse.count_datasets()

udata/core/reuse/permissions.py

@@ -3,10 +3,11 @@ from udata.core.organization.permissions import (
     OrganizationAdminNeed,
     OrganizationEditorNeed,
 )
+from udata.core.reuse.models import Reuse
 
 
 class ReuseEditPermission(Permission):
-    def __init__(self, reuse):
+    def __init__(self, reuse: Reuse) -> None:
         needs = []
 
         if reuse.organization:

udata/core/reuse/search.py

@@ -38,7 +38,7 @@ class ReuseSearch(ModelSearchAdapter):
     }
 
     @classmethod
-    def is_indexable(cls, reuse):
+    def is_indexable(cls, reuse: Reuse) -> bool:
         return reuse.deleted is None and len(reuse.datasets) > 0 and not reuse.private
 
     @classmethod
@@ -55,7 +55,7 @@ class ReuseSearch(ModelSearchAdapter):
         return reuses.order_by(sort).skip(offset).limit(args["page_size"]), reuses.count()
 
     @classmethod
-    def serialize(cls, reuse):
+    def serialize(cls, reuse: Reuse) -> dict:
         organization = None
         owner = None
         if reuse.organization:

udata/core/reuse/tasks.py

@@ -10,7 +10,7 @@ log = get_logger(__name__)
 
 
 @job("purge-reuses")
-def purge_reuses(self):
+def purge_reuses(self) -> None:
     for reuse in Reuse.objects(deleted__ne=None):
         log.info(f"Purging reuse {reuse}")
         # Remove followers
@@ -32,7 +32,7 @@ def purge_reuses(self):
 
 
 @task
-def notify_new_reuse(reuse_id):
+def notify_new_reuse(reuse_id: int) -> None:
     reuse = Reuse.objects.get(pk=reuse_id)
     for dataset in reuse.datasets:
         if dataset.organization: