A pythonic tool for batch loading data files (json, parquet, csv, tsv) into ElasticSearch
Project description
Main features:
Batch upload CSV (actually any *SV) files to Elasticsearch
Batch upload JSON files / JSON lines to Elasticsearch
Batch upload parquet files to Elasticsearch
Pre defining custom mappings
Delete index before upload
Index documents with _id from the document itself
Load data directly from url
And more
Installation
Usage
(venv)/tmp $ elasticsearch_loader --help Usage: elasticsearch_loader [OPTIONS] COMMAND [ARGS]... Options: --bulk-size INTEGER How many docs to collect before writing to ElasticSearch --concurrency INTEGER How much worker threads to start --es-host TEXT Elasticsearch cluster entry point. eg. http://localhost:9200 --consistency TEXT Consistency level passed to bulk api --index TEXT Destination index name [required] --delete Delete index before import? --type TEXT Docs type [required] --id-field TEXT Specify field name that be used as document id --index-settings-file FILENAME Specify path to json file containing index mapping and settings --help Show this message and exit. Commands: csv json FILES with the format of [{"a": "1"}, {"b":... parquet
Examples
Load 2 CSV files
elasticsearch_loader --index incidents --type incident csv file1.csv file2.csv
Load 2 JSON lines files
elasticsearch_loader --index incidents --type incident json --json-lines file1.json.lines file2.json.lines
Load 2 parquet files
elasticsearch_loader --index incidents --type incident parquet file1.parquet file2.parquet
Load CSV from github repo (actually any http/https is ok)
elasticsearch_loader --index data --type avg_height --id-field country json https://raw.githubusercontent.com/samayo/country-data/master/src/country-avg-male-height.json
Load data from stdin
generate_data | elasticsearch_loader --index data --type incident csv -
Read _id from incident_id field elasticsearch_loader --id-field incident_id --index incidents --type incident csv file1.csv file2.csv
Change bulk size
elasticsearch_loader --bulk-size 300 --index incidents --type incident csv file1.csv file2.csv
Change index concurrency
elasticsearch_loader --concurrency 20 --index incidents --type incident csv file1.csv file2.csv
Load custom mappings
elasticsearch_loader --index-settings-file samples/mappings.json --index incidents --type incident csv file1.csv file2.csv
Tests and sample data
Tests are located under test and can run by runnig tox input format can be found under samples
TODO
[x] parquet support
[x] progress bar
[ ] DLQ style out file for docs that didn’t got in
[x] Python3 support
[x] pep8 test
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Hashes for elasticsearch-loader-0.1.7.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 01bbfd2f5269b192fcf69f88b662a2c9c0cfcc2cf4b5515cfca39d8abec2ec3b |
|
MD5 | 4462348eed6d55bacc95e015d74f10c2 |
|
BLAKE2b-256 | 779d18d27cdfcf076f9eb045449cea111890159573350cba60343dc498e53458 |