forked from protonphoton/LJ
65 lines
1.5 KiB
Python
Executable File
65 lines
1.5 KiB
Python
Executable File
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
# -*- mode: Python -*-
|
|
|
|
|
|
'''
|
|
|
|
redis exporter
|
|
v0.1.0
|
|
|
|
A basic exporter
|
|
|
|
LICENCE : CC
|
|
|
|
by cocoa
|
|
|
|
|
|
'''
|
|
from __future__ import print_function
|
|
import argparse
|
|
import ast
|
|
import json
|
|
import os
|
|
import redis
|
|
import sys
|
|
import time
|
|
|
|
argsparser = argparse.ArgumentParser(description="Redis exporter LJ")
|
|
argsparser.add_argument("-i","--ip",help="IP address of the Redis server ",default="127.0.0.1",type=str)
|
|
argsparser.add_argument("-p","--port",help="Port of the Redis server ",default="6379",type=str)
|
|
argsparser.add_argument("-k","--key",help="Redis key to update",default="0",type=str)
|
|
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose")
|
|
args = argsparser.parse_args()
|
|
|
|
ip = args.ip
|
|
port = args.port
|
|
key = args.key
|
|
verbose=args.verbose
|
|
|
|
def debug(*args, **kwargs):
|
|
if( verbose == False ):
|
|
return
|
|
print(*args, file=sys.stderr, **kwargs)
|
|
|
|
r=redis.StrictRedis(host=ip, port=port, db=0)
|
|
|
|
try:
|
|
while True:
|
|
line = sys.stdin.readline()
|
|
if line == "":
|
|
time.sleep(0.01)
|
|
line = line.rstrip('\n')
|
|
line=line[1:-1]
|
|
# Decode as list of lists
|
|
pointsList = ast.literal_eval(line)
|
|
# convert to list of tuples
|
|
pointsList = [tuple(elem) for elem in pointsList]
|
|
# Convert to JSON string
|
|
line = json.dumps( pointsList )
|
|
if r.set(key,line)==True:
|
|
debug("exports::redis set("+str(key)+") to "+line)
|
|
except EOFError:
|
|
debug("break")# no more information
|
|
|