influx_test_data.py

Go to the documentation of this file.

#
# Copyright (c) 2020 Seagate Technology LLC and/or its Affiliates
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# For any questions about this software or licensing,
# please email opensource@seagate.com or cortx-questions@seagate.com.
#

import time
import argparse
import datetime
from influxdb import InfluxDBClient

USER = ''
PASSWORD = ''
DBNAME = 'm0play'

class InfluxDB:
    def __init__(self, user, password, dbname, host, port, bucket_size):
        self.client = InfluxDBClient(host, port, user, password, dbname)
        self.bucket = []
        self.bucket_size = bucket_size

    def createDB(self):
        self.client.create_database(self.dbname)

    def openDB(self):
        self.client.switch_database(self.dbname)

    def closeDB(self):
        client.write_points(self.bucket)

    def writePoint(self, point):
        self.bucket.append(point)
        if len(self.bucket) < bucket_size:
            client.write_points(self.bucket)


def main(host='localhost', port=8086):
    now = datetime.datetime.today()

    point = {
        "measurement": 'meas0',
        "time": now.isoformat(timespec='milliseconds')
        "fields": {
            "id": 11100111
        },
        "tags": {
            "type": 'XXX00YYY'
        }
    }

    db = InfluxDB(USER, PASSWORD, DBNAME, host, port, 10000)
    db.createDB()
    db.openDB()
    db.writePoint(point)
    db.closeDB()


def parse_args():
    """Parse the args."""
    parser = argparse.ArgumentParser(
        description='example code to play with InfluxDB')
    parser.add_argument('--host', type=str, required=False,
                        default='localhost',
                        help='hostname influxdb http API')
    parser.add_argument('--port', type=int, required=False, default=8086,
                        help='port influxdb http API')
    return parser.parse_args()


if __name__ == '__main__':
    args = parse_args()
    main(host=args.host, port=args.port)