aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2021-11-02 14:28:57 -0700
committerBryan Newbold <bnewbold@robocracy.org>2021-11-02 17:02:49 -0700
commita0d89d08b40c07b13a76a8afd020b0d1e62e58aa (patch)
treee44c6fce15cda3359d82e0cd991a79ebf70cfb02
parent8e2fd41d10725b65c787ae56cb9320fbcc182288 (diff)
downloadfatcat-a0d89d08b40c07b13a76a8afd020b0d1e62e58aa.tar.gz
fatcat-a0d89d08b40c07b13a76a8afd020b0d1e62e58aa.zip
small python tweaks for annotations, imports
-rw-r--r--python/fatcat_tools/importers/common.py2
-rw-r--r--python/fatcat_tools/importers/crossref.py6
-rw-r--r--python/fatcat_tools/transforms/elasticsearch.py2
3 files changed, 7 insertions, 3 deletions
diff --git a/python/fatcat_tools/importers/common.py b/python/fatcat_tools/importers/common.py
index 2c4dd496..9d22ce83 100644
--- a/python/fatcat_tools/importers/common.py
+++ b/python/fatcat_tools/importers/common.py
@@ -7,7 +7,7 @@ import sqlite3
import datetime
import subprocess
from collections import Counter
-from typing import Dict, Any, List, Optional
+from typing import Dict, Any, List, Optional, Tuple
import lxml
import xml.etree.ElementTree as ET
diff --git a/python/fatcat_tools/importers/crossref.py b/python/fatcat_tools/importers/crossref.py
index d4b4a4c7..bd72a781 100644
--- a/python/fatcat_tools/importers/crossref.py
+++ b/python/fatcat_tools/importers/crossref.py
@@ -1,7 +1,11 @@
import sqlite3
import datetime
+from typing import Dict, Optional, Any
+
import fatcat_openapi_client
+from fatcat_openapi_client import ReleaseEntity
+
from .common import EntityImporter, clean
@@ -452,7 +456,7 @@ class CrossrefImporter(EntityImporter):
if not extra:
extra = None
- re = fatcat_openapi_client.ReleaseEntity(
+ re = ReleaseEntity(
work_id=None,
container_id=container_id,
title=title,
diff --git a/python/fatcat_tools/transforms/elasticsearch.py b/python/fatcat_tools/transforms/elasticsearch.py
index 26eacded..bfc18f83 100644
--- a/python/fatcat_tools/transforms/elasticsearch.py
+++ b/python/fatcat_tools/transforms/elasticsearch.py
@@ -4,7 +4,7 @@ from typing import Dict, List, Any, Optional
import tldextract
-from fatcat_openapi_client import ReleaseEntity, ContainerEntity
+from fatcat_openapi_client import ReleaseEntity, ContainerEntity, EntityEdit, ChangelogEntry, FileEntity
def check_kbart(year: int, archive: dict) -> Optional[bool]: