diff options
author | Martin Czygan <martin@archive.org> | 2020-04-20 17:55:44 +0000 |
---|---|---|
committer | Martin Czygan <martin@archive.org> | 2020-04-20 17:55:44 +0000 |
commit | a134944ea6044d488af264e3845dad40536c61f4 (patch) | |
tree | b35191c6b1af0666faffdd2cc6f7c7b2372e3070 /python/fatcat_tools/workers | |
parent | e749b576bdb9bbec3bcef7e4e389e7fbbf916c84 (diff) | |
parent | 9e9f7f1da115458d87f8bcdc011b2843e2b31d3b (diff) | |
download | fatcat-a134944ea6044d488af264e3845dad40536c61f4.tar.gz fatcat-a134944ea6044d488af264e3845dad40536c61f4.zip |
Merge branch 'bnewbold-fix-changelog-es' into 'master'
fixes for changelog elasticsearch worker
See merge request webgroup/fatcat!46
Diffstat (limited to 'python/fatcat_tools/workers')
-rw-r--r-- | python/fatcat_tools/workers/elasticsearch.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/python/fatcat_tools/workers/elasticsearch.py b/python/fatcat_tools/workers/elasticsearch.py index 525f372b..e58b3da1 100644 --- a/python/fatcat_tools/workers/elasticsearch.py +++ b/python/fatcat_tools/workers/elasticsearch.py @@ -19,7 +19,7 @@ class ElasticsearchReleaseWorker(FatcatWorker): def __init__(self, kafka_hosts, consume_topic, poll_interval=10.0, offset=None, elasticsearch_backend="http://localhost:9200", elasticsearch_index="fatcat", - batch_size=200): + batch_size=200, api_host="https://api.fatcat.wiki/v0"): super().__init__(kafka_hosts=kafka_hosts, consume_topic=consume_topic) self.consumer_group = "elasticsearch-updates3" @@ -30,9 +30,11 @@ class ElasticsearchReleaseWorker(FatcatWorker): self.entity_type = ReleaseEntity self.elasticsearch_document_name = "release" self.transform_func = release_to_elasticsearch + self.api_host = api_host def run(self): ac = ApiClient() + api = public_api(self.api_host) def fail_fast(err, partitions): if err is not None: @@ -103,13 +105,20 @@ class ElasticsearchReleaseWorker(FatcatWorker): if entity_dict.get('name') and not entity_dict.get('title'): continue entity = entity_from_json(json_str, self.entity_type, api_client=ac) + if self.elasticsearch_document_name == "changelog": + key = entity.index + # might need to fetch from API + if not (entity.editgroup and entity.editgroup.editor): + entity = api.get_changelog_entry(entity.index) + else: + key = entity.ident # TODO: handle deletions from index bulk_actions.append(json.dumps({ - "index": { "_id": entity.ident, }, + "index": { "_id": key, }, })) bulk_actions.append(json.dumps( self.transform_func(entity))) - print("Upserting, eg, {} (of {} releases in elasticsearch)".format(entity.ident, len(batch))) + print("Upserting, eg, {} (of {} {} in elasticsearch)".format(key, len(batch), self.elasticsearch_document_name)) elasticsearch_endpoint = "{}/{}/{}/_bulk".format( self.elasticsearch_backend, self.elasticsearch_index, |