- Added support to python 2

grupoamudi · Nov 6, 2017 · f892a62 · f892a62
1 parent bf4cfb2
commit f892a62
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/peopleCounter/reader.py b/peopleCounter/reader.py
@@ -10,7 +10,9 @@
 ser = serial.Serial(serialport, 115200, timeout=1)
 while True:
     data = ser.read(50)
-    if len(data) == 5 and data[0] == 48 and data[1] == 48 and data[2] == 48:
+    if (len(data) == 5 and data[0] == '0' and data[1] == '0' and data[2] == '0') or\
+       (len(data) == 5 and data[0] == 48  and data[1] == 48  and data[2] == 48 ):
+
         nmacs = struct.unpack("<H", data[3:])[0]
         line = str(datetime.datetime.now()) + ' ' + str(nmacs) + '\n'
         f = open('counter.txt', 'a')
@@ -20,4 +22,7 @@
         f.write(struct.pack("<H", nmacs))
         f.close()
         print(line)
+    else:
+        if len(data) == 5:
+            print(data[0])