summaryrefslogtreecommitdiffstats
path: root/python/fatcat_web/entity_helpers.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/fatcat_web/entity_helpers.py')
-rw-r--r--python/fatcat_web/entity_helpers.py21
1 files changed, 18 insertions, 3 deletions
diff --git a/python/fatcat_web/entity_helpers.py b/python/fatcat_web/entity_helpers.py
index ab4c16b5..26371341 100644
--- a/python/fatcat_web/entity_helpers.py
+++ b/python/fatcat_web/entity_helpers.py
@@ -1,11 +1,26 @@
-from flask import abort
-from fatcat_openapi_client import *
+from fatcat_openapi_client import (
+ ContainerEntity,
+ CreatorEntity,
+ FileEntity,
+ FilesetEntity,
+ ReleaseEntity,
+ ReleaseExtIds,
+ WebcaptureEntity,
+ WorkEntity,
+)
from fatcat_openapi_client.rest import ApiException, ApiValueError
-from fatcat_tools.transforms import *
+from flask import abort
+
+from fatcat_tools.transforms import (
+ container_to_elasticsearch,
+ file_to_elasticsearch,
+ release_to_elasticsearch,
+)
from fatcat_web import api
from fatcat_web.hacks import strip_extlink_xml, wayback_suffix
+
def enrich_container_entity(entity):
if entity.state in ('redirect', 'deleted'):
return entity