mirror of
https://github.com/janishutz/BiogasControllerApp.git
synced 2025-11-25 13:54:24 +00:00
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from typing import Optional
|
|
import serial
|
|
import struct
|
|
import serial.tools.list_ports
|
|
|
|
|
|
class Com:
|
|
def __init__(self, filters: Optional[list[str]] = None) -> None:
|
|
self._serial: Optional[serial.Serial] = None
|
|
self._filters = filters if filters != None else [ 'USB-Serial Controller', 'Prolific USB-Serial Controller' ]
|
|
self._port_override = ''
|
|
|
|
def set_port_override(self, override: str) -> None:
|
|
"""Set the port override, to disable port search"""
|
|
self._port_override = override
|
|
|
|
def get_comport(self) -> str:
|
|
"""Find the comport the microcontroller has attached to"""
|
|
if self._port_override != '':
|
|
return self._port_override
|
|
|
|
# Catch all errors and simply return an empty string if search unsuccessful
|
|
try:
|
|
# Get an array of all used comports
|
|
ports = [comport.device for comport in serial.tools.list_ports.comports()]
|
|
|
|
# Filter for specific controller
|
|
for comport in ports:
|
|
for filter in self._filters:
|
|
if ( filter in comport ):
|
|
return comport
|
|
except:
|
|
pass
|
|
|
|
return ''
|
|
|
|
def connect(self, baud_rate: int, port_override: Optional[str] = None) -> bool:
|
|
"""Try to find a comport and connect to the microcontroller. Returns the success as a boolean"""
|
|
comport = self.get_comport()
|
|
|
|
# Comport search returns empty string if search unsuccessful
|
|
if comport == '':
|
|
try:
|
|
self._serial = serial.Serial(comport, baud_rate, timeout=5)
|
|
except:
|
|
return False
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def close(self) -> None:
|
|
"""Close the serial connection, if possible"""
|
|
if self._serial != None:
|
|
try:
|
|
self._serial.close()
|
|
except:
|
|
pass
|
|
|
|
def receive(self, byte_count: int) -> bytes:
|
|
"""Recieve bytes from microcontroller over serial. Returns bytes. Might want to decode using functions from lib.tools"""
|
|
if self._serial == None:
|
|
self.connect(19200)
|
|
if self._serial != None:
|
|
return self._serial.read(byte_count)
|
|
else:
|
|
raise Exception('ERR_CONNECTION')
|
|
|
|
def send(self, msg: str) -> None:
|
|
"""Send a string over serial connection."""
|
|
if self._serial == None:
|
|
self.connect(19200)
|
|
if self._serial != None:
|
|
self._serial.write(msg.encode())
|
|
|
|
def send_float(self, msg: float) -> None:
|
|
"""Send a float number over serial connection"""
|
|
if self._serial == None:
|
|
self.connect(19200)
|
|
if self._serial != None:
|
|
self._serial.write(bytearray(struct.pack('>f', msg))[0:3])
|