diff options
Diffstat (limited to 'python/fatcat_tools/transforms')
-rw-r--r-- | python/fatcat_tools/transforms/elasticsearch.py | 5 | ||||
-rw-r--r-- | python/fatcat_tools/transforms/entities.py | 2 |
2 files changed, 4 insertions, 3 deletions
diff --git a/python/fatcat_tools/transforms/elasticsearch.py b/python/fatcat_tools/transforms/elasticsearch.py index c16053ec..11e133b7 100644 --- a/python/fatcat_tools/transforms/elasticsearch.py +++ b/python/fatcat_tools/transforms/elasticsearch.py @@ -279,7 +279,7 @@ def _rte_container_helper(container: ContainerEntity, release_year: Optional[int Container metadata sub-section of release_to_elasticsearch() """ this_year = datetime.date.today().year - t = dict() + t: Dict[str, Any] = dict() t["publisher"] = container.publisher t["container_name"] = container.name # this is container.ident, not release.container_id, because there may @@ -325,7 +325,8 @@ def _rte_container_helper(container: ContainerEntity, release_year: Optional[int if c_extra.get("sherpa_romeo"): if c_extra["sherpa_romeo"].get("color") == "white": t["is_oa"] = False - if c_extra.get("default_license") and c_extra.get("default_license").startswith("CC-"): + c_default_license = c_extra.get("default_license") + if c_default_license and c_default_license.startswith("CC-"): t["is_oa"] = True if c_extra.get("doaj"): if c_extra["doaj"].get("as_of"): diff --git a/python/fatcat_tools/transforms/entities.py b/python/fatcat_tools/transforms/entities.py index e5da633f..4066297b 100644 --- a/python/fatcat_tools/transforms/entities.py +++ b/python/fatcat_tools/transforms/entities.py @@ -58,7 +58,7 @@ def entity_to_toml( def entity_from_toml( - toml_str: str, entity_type: Any, api_client: Optional[List[str]] = None + toml_str: str, entity_type: Any, api_client: Optional[ApiClient] = None ) -> Any: obj = toml.loads(toml_str) return entity_from_dict(obj, entity_type, api_client=api_client) |