import csv
import elasticsearch
import random


def get_data(filename):
    # Load and parse the CSV as a dict
    data = []

    with open(filename) as f:
        reader = csv.reader(f, delimiter="\t")

        rows = list(reader)

        for x in range(0, 100):
            for row in rows:
                try:
                    data.append({
                        "name": row[0],
                        "iterations": random.randrange(0, 1000),
                        "speed": random.randrange(0, 10)
                    })
                except IndexError:
                    continue

    return data


def get_json_write_data(row):
    json = {
        "measurement": row["name"],
        "tags": {},
        "fields": {
            "iterations": row["iterations"],
            "speed": row["speed"]
        }
    }

    return json


def main():
    filename = "data.csv"  # TODO: Read this from a command line argument

    data = get_data(filename)

    client = elasticsearch.Elasticsearch()

    n = 0
    for row in data:
        print row

        client.index(index='data', doc_type='blog', id=n, body=row)
        n += 1


if __name__ == "__main__":
    main()