try new data class

This commit is contained in:
DasMoorhuhn 2024-06-22 20:46:25 +02:00
parent 208a82142c
commit 2b1c12f273
6 changed files with 46 additions and 46 deletions

View File

@ -2,7 +2,7 @@
<module type="PYTHON_MODULE" version="4"> <module type="PYTHON_MODULE" version="4">
<component name="NewModuleRootManager"> <component name="NewModuleRootManager">
<content url="file://$MODULE_DIR$" /> <content url="file://$MODULE_DIR$" />
<orderEntry type="jdk" jdkName="Python 3.12" jdkType="Python SDK" /> <orderEntry type="inheritedJdk" />
<orderEntry type="sourceFolder" forTests="false" /> <orderEntry type="sourceFolder" forTests="false" />
</component> </component>
</module> </module>

2
.idea/misc.xml generated
View File

@ -3,5 +3,5 @@
<component name="Black"> <component name="Black">
<option name="sdkName" value="Python 3.12 (autopicture-v3)" /> <option name="sdkName" value="Python 3.12 (autopicture-v3)" />
</component> </component>
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12" project-jdk-type="Python SDK" /> <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10" project-jdk-type="Python SDK" />
</project> </project>

View File

@ -1,12 +1,30 @@
from datetime import datetime
class Data: class Data:
def __init__(self, data:dict): def __init__(self, data):
self.timestamp = data['timestamp'] self.timestamp = None
self.mac = data['mac'] self.mac = None
self.temperature = data['temperature'] self.temperature = None
self.humidity = data['humidity'] self.humidity = None
self.battery_percent = data['battery_percent'] self.battery_percent = None
self.battery_volt = data['battery_volt'] self.battery_volt = None
self.count = data['count'] self.count = None
self.parse_data(data)
def parse_data(self, val):
data_bytes = [int(val[i:i + 2], 16) for i in range(0, len(val), 2)]
if data_bytes[8] > 127:
print(data_bytes)
data_bytes[8] -= 256
self.timestamp = datetime.now().astimezone().replace(microsecond=0).isoformat()
self.mac = ":".join(["{:02X}".format(data_bytes[i]) for i in range(2, 8)])
self.temperature = (data_bytes[8] * 256 + data_bytes[9]) / 10
self.humidity = data_bytes[10]
self.battery_percent = data_bytes[11]
self.battery_volt = (data_bytes[12] * 256 + data_bytes[13]) / 1000
self.count = data_bytes[14]
def print_data(self): def print_data(self):
print(self.to_json()) print(self.to_json())

View File

@ -1,6 +1,5 @@
from bluepy.btle import DefaultDelegate from bluepy.btle import DefaultDelegate
from bluepy.btle import Scanner from bluepy.btle import Scanner
from datetime import datetime
from data_class import Data from data_class import Data
from devices import get_devices as get_device_from_config from devices import get_devices as get_device_from_config
@ -18,10 +17,11 @@ class ScanDelegate(DefaultDelegate):
for (sdid, desc, val) in dev.getScanData(): for (sdid, desc, val) in dev.getScanData():
if self.is_temperature(sdid, val): if self.is_temperature(sdid, val):
data_obj = Data(self.parse_data(val)) data_obj = Data(val)
if self.is_atc_device(dev, data_obj): if self.is_atc_device(dev, data_obj):
device_from_config = self.get_device(dev) device_from_config = self.get_device(dev)
devices.append([dev, data_obj]) devices.append([dev, data_obj, device_from_config])
@staticmethod @staticmethod
def is_temperature(sdid, val): def is_temperature(sdid, val):
@ -42,35 +42,22 @@ class ScanDelegate(DefaultDelegate):
print(f'\tTemp: {data_obj.temperature}°C, Humid: {data_obj.humidity}%, Batt: {data_obj.battery_percent}%\n') print(f'\tTemp: {data_obj.temperature}°C, Humid: {data_obj.humidity}%, Batt: {data_obj.battery_percent}%\n')
return True return True
@staticmethod
def parse_data(val):
bytes = [int(val[i:i + 2], 16) for i in range(0, len(val), 2)]
if bytes[8] > 127:
bytes[8] -= 256
return {
'timestamp': datetime.now().astimezone().replace(microsecond=0).isoformat(),
'mac': ":".join(["{:02X}".format(bytes[i]) for i in range(2, 8)]),
'temperature': (bytes[8] * 256 + bytes[9]) / 10,
'humidity': bytes[10],
'battery_percent': bytes[11],
'battery_volt': (bytes[12] * 256 + bytes[13]) / 1000,
'count': bytes[14],
}
@staticmethod @staticmethod
def get_device(dev): def get_device(dev):
return next((d for d in get_device_from_config() if d.mac == dev.addr.upper()), None) return next((d for d in get_device_from_config() if d.mac == dev.addr.upper()), None)
class Discovery: def cleanup():
def __init__(self): global devices
pass devices = []
def start_discovery(self, timeout=20.0):
global devices
print(f'Start discovery with timout {timeout}s...')
scanner = Scanner().withDelegate(ScanDelegate()) def start_discovery(timeout=20.0):
scanner.scan(timeout=timeout, passive=True) cleanup()
global devices
print(f'Start discovery with timout {timeout}s...')
return devices scanner = Scanner().withDelegate(ScanDelegate())
scanner.scan(timeout=timeout, passive=False)
return devices

View File

@ -1,8 +1,9 @@
from time import sleep from time import sleep
from discovery import start_discovery
def start_loop(interval=60): def start_loop(interval=60):
while True: while True:
from discovery import start_discovery
start_discovery() start_discovery()
sleep(interval) sleep(interval)

View File

@ -1,13 +1,7 @@
from discovery import Discovery from discovery import start_discovery
from data_class import Data from data_class import Data
from loop import start_loop from loop import start_loop
from time import sleep from time import sleep
while True: devices = start_discovery()
discovery = Discovery()
devices = discovery.start_discovery()
sleep(10)