annotate service/air_particles/air_particles.py @ 372:c52e7abdd6b1

pms5003 -> ttyUSB -> influxdb Ignore-this: d0c8851b557a88361a1755b93af4fc58
author drewp@bigasterisk.com
date Wed, 21 Nov 2018 22:44:08 -0800
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
372
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
1 from __future__ import print_function
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
2 import sys
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
3 import logging
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
4 import serial
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
5 import time
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
6 from influxdb import InfluxDBClient
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
7
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
8 logging.basicConfig(level=logging.INFO)
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
9 log = logging.getLogger()
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
10
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
11 location, = sys.argv[1:]
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
12 min_period = 5
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
13
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
14 ser_port = "/dev/ttyUSB0"
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
15 ser = serial.Serial(ser_port, baudrate=9600, stopbits=1, parity="N", timeout=2)
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
16
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
17 influx = InfluxDBClient('bang6', 9060, 'root', 'root', 'main')
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
18
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
19 last_write = 0
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
20 while True:
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
21 ser.flushInput()
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
22 s = map(ord, ser.read(32))
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
23 if s[:2] != [0x42, 0x4d]:
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
24 log.warn('unknown packet header: %s' % s)
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
25 continue
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
26
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
27 cs = (s[30] * 256 + s[31]) # check sum
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
28 check = 0
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
29 for i in range(30):
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
30 check += s[i]
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
31 if check != cs:
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
32 log.warn('checksum mismatch: %s' % s)
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
33 continue
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
34
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
35 sample = {
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
36 # PM1, PM2.5 and PM10 values for standard particle in ug/m^3
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
37 'pm1_0_std': s[4] * 256 + s[5],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
38 'pm2_5_std': s[6] * 256 + s[7],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
39 'pm10_0_std': s[8] * 256 + s[9],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
40
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
41 # PM1, PM2.5 and PM10 values for atmospheric conditions in ug/m^3
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
42 'pm1_0_atm': s[10] * 256 + s[11],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
43 'pm2_5_atm': s[12] * 256 + s[13],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
44 'pm10_0_atm': s[14] * 256 + s[15],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
45
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
46 # Number of particles bigger than 0.3 um, 0.5 um, etc. in #/cm^3
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
47 'part_0_3': s[16] * 256 + s[17],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
48 'part_0_5': s[18] * 256 + s[19],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
49 'part_1_0': s[20] * 256 + s[21],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
50 'part_2_5': s[22] * 256 + s[23],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
51 'part_5_0': s[24] * 256 + s[25],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
52 'part_10_0': s[26] * 256 + s[27],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
53 }
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
54
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
55 now = int(time.time())
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
56 if now < last_write + min_period:
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
57 continue
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
58 if last_write == 0:
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
59 log.info('sending first sample: %s', sample)
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
60 last_write = now
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
61 influx.write_points([{'measurement': 'air_particles',
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
62 "fields": sample,
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
63 "time": now,
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
64 }],
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
65 tags=dict(location=location),
c52e7abdd6b1 pms5003 -> ttyUSB -> influxdb
drewp@bigasterisk.com
parents:
diff changeset
66 time_precision='s')