summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2019-05-30 12:50:22 -0700
committerBryan Newbold <bnewbold@robocracy.org>2019-05-30 12:50:22 -0700
commit393172a29e91bf1236e546b4a99f445dbbe849ef (patch)
treea2fb74c676b195c068f98fee5d76afb531dead78
parenteee39965eee92b5005df0d967be779c2f2bb15f8 (diff)
downloadfatcat-393172a29e91bf1236e546b4a99f445dbbe849ef.tar.gz
fatcat-393172a29e91bf1236e546b4a99f445dbbe849ef.zip
fix syntax bugs (container elastic worker)
-rw-r--r--python/fatcat_tools/workers/elasticsearch.py10
-rwxr-xr-xpython/fatcat_worker.py2
2 files changed, 6 insertions, 6 deletions
diff --git a/python/fatcat_tools/workers/elasticsearch.py b/python/fatcat_tools/workers/elasticsearch.py
index 6e336e52..fc96e3d4 100644
--- a/python/fatcat_tools/workers/elasticsearch.py
+++ b/python/fatcat_tools/workers/elasticsearch.py
@@ -4,7 +4,7 @@ import time
import requests
from pykafka.common import OffsetType
-from fatcat_client import ReleaseEntity, ApiClient
+from fatcat_client import ReleaseEntity, ContainerEntity, ApiClient
from fatcat_tools import *
from .worker_common import FatcatWorker
@@ -62,10 +62,10 @@ class ElasticsearchContainerWorker(ElasticsearchReleaseWorker):
elasticsearch_backend="http://localhost:9200", elasticsearch_index="fatcat"):
super().__init__(kafka_hosts=kafka_hosts,
consume_topic=consume_topic,
- poll_interval,
- offset,
- elasticsearch_backend,
- elasticsearch_index)
+ poll_interval=poll_interval,
+ offset=offset,
+ elasticsearch_backend=elasticsearch_backend,
+ elasticsearch_index=elasticsearch_index)
self.entity_type = ContainerEntity
self.elasticsearch_document_name = "container"
self.transform_func = container_to_elasticsearch
diff --git a/python/fatcat_worker.py b/python/fatcat_worker.py
index 4455d29e..b9e5b8fd 100755
--- a/python/fatcat_worker.py
+++ b/python/fatcat_worker.py
@@ -6,7 +6,7 @@ import datetime
import raven
from fatcat_tools import public_api
-from fatcat_tools.workers import ChangelogWorker, EntityUpdatesWorker, ElasticsearchReleaseWorker
+from fatcat_tools.workers import ChangelogWorker, EntityUpdatesWorker, ElasticsearchReleaseWorker, ElasticsearchContainerWorker
# Yep, a global. Gets DSN from `SENTRY_DSN` environment variable
sentry_client = raven.Client()