From f208054f6883589239f21d5e46545960c45ece18 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Wed, 17 Nov 2021 16:11:44 -0800
Subject: python: fix api annotations (DefaultApi not ApiClient)

---
 python/fatcat_tools/workers/changelog.py     | 6 +++---
 python/fatcat_tools/workers/elasticsearch.py | 2 +-
 python/fatcat_tools/workers/worker_common.py | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

(limited to 'python/fatcat_tools/workers')

diff --git a/python/fatcat_tools/workers/changelog.py b/python/fatcat_tools/workers/changelog.py
index ff358c66..3ea592bb 100644
--- a/python/fatcat_tools/workers/changelog.py
+++ b/python/fatcat_tools/workers/changelog.py
@@ -3,7 +3,7 @@ import time
 from typing import Any, Dict, List, Optional
 
 from confluent_kafka import Consumer, KafkaException, Producer
-from fatcat_openapi_client import ApiClient, ReleaseEntity
+from fatcat_openapi_client import DefaultApi, ReleaseEntity
 
 from fatcat_tools.transforms import release_ingest_request, release_to_elasticsearch
 
@@ -18,7 +18,7 @@ class ChangelogWorker(FatcatWorker):
 
     def __init__(
         self,
-        api: ApiClient,
+        api: DefaultApi,
         kafka_hosts: str,
         produce_topic: str,
         poll_interval: float = 10.0,
@@ -89,7 +89,7 @@ class EntityUpdatesWorker(FatcatWorker):
 
     def __init__(
         self,
-        api: ApiClient,
+        api: DefaultApi,
         kafka_hosts: str,
         consume_topic: str,
         release_topic: str,
diff --git a/python/fatcat_tools/workers/elasticsearch.py b/python/fatcat_tools/workers/elasticsearch.py
index 71c4dcf6..7c959e93 100644
--- a/python/fatcat_tools/workers/elasticsearch.py
+++ b/python/fatcat_tools/workers/elasticsearch.py
@@ -46,7 +46,7 @@ class ElasticsearchReleaseWorker(FatcatWorker):
         self.elasticsearch_backend = elasticsearch_backend
         self.elasticsearch_index = elasticsearch_index
         self.elasticsearch_release_index = elasticsearch_release_index
-        self.entity_type = ReleaseEntity
+        self.entity_type: Union[ReleaseEntity,ChangelogEntry,ContainerEntity] = ReleaseEntity
         self.transform_func: Callable = release_to_elasticsearch
         self.api_host = api_host
         self.query_stats = query_stats
diff --git a/python/fatcat_tools/workers/worker_common.py b/python/fatcat_tools/workers/worker_common.py
index 5239465b..f6b0293e 100644
--- a/python/fatcat_tools/workers/worker_common.py
+++ b/python/fatcat_tools/workers/worker_common.py
@@ -1,7 +1,7 @@
 from typing import Any, Dict, Optional
 
 from confluent_kafka import Consumer, KafkaException, Message, TopicPartition
-from fatcat_openapi_client import ApiClient
+from fatcat_openapi_client import DefaultApi
 
 
 def most_recent_message(topic: str, kafka_config: Dict[str, Any]) -> Message:
@@ -58,7 +58,7 @@ class FatcatWorker:
         kafka_hosts: str,
         produce_topic: Optional[str] = None,
         consume_topic: Optional[str] = None,
-        api: Optional[ApiClient] = None,
+        api: Optional[DefaultApi] = None,
     ) -> None:
         if api:
             self.api = api
-- 
cgit v1.2.3