diff --git a/rp_get_sayma_data.py b/rp_get_sayma_data.py index 51e3f1c..b98705b 100644 --- a/rp_get_sayma_data.py +++ b/rp_get_sayma_data.py @@ -114,13 +114,25 @@ def main(): print("Succesfully written y2 raw data from RP {}.".format(rp.name)) if args.txt: + header = '''\ + RedPitaya Oscilloscope Reading + + RP Hostname: {} ({}) + RP Channel: Input {} ({}) + Date & Time: {} + ------ + ''' y1 = [float(i) for i in y1_raw.split(',')] y2 = [float(i) for i in y2_raw.split(',')] with open(os.path.join(args.dir, 'rp_{}_y1_raw.txt'.format(rp.name)), 'w') as f: + f.write(''.join([l.lstrip(' ') for l in header.format( + RP_IP_ADDRS[rp.name], rp.name, 1, ch1_gain.upper(), now_iso).splitlines(True)])) for i in y1: f.write(str(i) + '\n') print("Succesfully written y1 human-readable data from {}.".format(rp.name)) with open(os.path.join(args.dir, 'rp_{}_y2_raw.txt'.format(rp.name)), 'w') as f: + f.write(''.join([l.lstrip(' ') for l in header.format( + RP_IP_ADDRS[rp.name], rp.name, 2, ch2_gain.upper(), now_iso).splitlines(True)])) for i in y2: f.write(str(i) + '\n') print("Succesfully written y2 human-readable data from {}.".format(rp.name))