#!/bin/env python3 import requests from math import sin, cos from time import sleep import numpy as np from argparse import ArgumentParser # import pyaudio def sine(args): x = 0.0 while True: try: requests.get(f"http://{args.address}:{args.port}/add?chart={args.chart}&series={args.series}&x={x}&y={sin(x)}") x += 0.01 sleep(0.01) except KeyboardInterrupt: break def cosine(args): x = 0.0 while True: try: requests.get(f"http://{args.address}:{args.port}/add?chart={args.chart}&series={args.series}&x={x}&y={cos(x)}") x += 0.01 sleep(0.01) except KeyboardInterrupt: break def sine_array(args): start = 0 while True: try: x = np.linspace(start, start+0.5, args.num) y = np.sin(x) start += 0.5 data = { "chart": args.chart, "series": args.series, "points": [ {"x": i[0], "y":i[1]} for i in zip(x,y) ] } r = requests.post(f"http://{args.address}:{args.port}/add", json=data) sleep(args.delay) except KeyboardInterrupt: break def star(args): points = [ (2/4,2/2), (3/4,0), (0,1/2), (4/4,1/2), (1/4,0), (2/4,2/2) ] for point in points: requests.get(f"http://{args.address}:{args.port}/add?chart={args.chart}&series={args.series}&x={point[0]}&y={point[1]}") # def audio(): # p = pyaudio.PyAudio() # stream = p.open( # input=True, # input_device_index=0, # format=pyaudio.paInt16, # channels=1, # rate=44100, # frames_per_buffer=1024 # ) # stream.start_stream() # while True: # try: # data = stream.read(1024, False) # requests.post( # "http://0.0.0.0:8080/add", # json={ # "chart": 0, # "series": "audio", # "points": [ {"x": i, "y": v} for (i, v) in enumerate(data) ] # }, # headers={"Connection":"close"}) # except KeyboardInterrupt: # break # stream.stop_stream() # stream.close() # p.terminate() if __name__ == "__main__": parser = ArgumentParser() parser.add_argument("-t", action="store", dest="test", help="test type", type=str, choices=["sine", "cosine", "sine_array", "star"], required=False, default="sine") parser.add_argument("-n", action="store", dest="num", help="sin_array size/iteration", type=int, required=False, default=100) parser.add_argument("-d", action="store", dest="delay", help="delay between iteration", type=float, required=False, default=0.25) parser.add_argument("-c", action="store", dest="chart", help="chart id", type=int, required=True) parser.add_argument("-s", action="store", dest="series", help="series name", type=str, required=True) parser.add_argument("-p", action="store", dest="port", help="server port", type=int, required=False, default=8080) parser.add_argument("-a", action="store", dest="address", help="server address", type=str, required=False, default="localhost") args = parser.parse_args() if args.test == "sine": sine(args) if args.test == "cosine": cosine(args) elif args.test == "sine_array": sine_array(args) elif args.test == "star": star(args)