1
0
mirror of https://github.com/pommi/python-itho-wpu.git synced 2024-11-23 14:12:14 +01:00

feat(itho-wpu): initial i2c master functionality

Writing directly to the /dev/i2c-1 device.

Support for 4 actions:
- getregelaar
- getserial
- getdatatype
- getdatalog
This commit is contained in:
Pim van den Berg 2020-10-08 13:39:10 +02:00
parent 4e6d4c8666
commit 91913c4fe7
2 changed files with 91 additions and 0 deletions

29
i2c_raw.py Normal file
View File

@ -0,0 +1,29 @@
import io
import fcntl
import struct
I2C_SLAVE = 0x0703
class I2CRaw:
def __init__(self, address, bus):
self.fr = io.open(f"/dev/i2c-{bus}", "rb", buffering=0)
self.fw = io.open(f"/dev/i2c-{bus}", "wb", buffering=0)
fcntl.ioctl(self.fr, I2C_SLAVE, address)
fcntl.ioctl(self.fw, I2C_SLAVE, address)
def write_i2c_block_data(self, data):
if type(data) is not list:
return -1
data = bytearray(data)
self.fw.write(data)
return 0
def read_i2c_block_data(self, n_bytes):
data_raw = self.fr.read(n_bytes)
unpack_format = 'B'*n_bytes
return list(struct.unpack(unpack_format, data_raw))
def close(self):
self.fr.close()
self.fw.close()

62
itho-wpu.py Executable file
View File

@ -0,0 +1,62 @@
#!/usr/bin/env python3
import argparse
import i2c_raw
import logging
import sys
consolelogformatter = logging.Formatter("%(asctime)-15s %(levelname)s: %(message)s")
logger = logging.getLogger('stdout')
logger.setLevel(logging.INFO)
stdout_log_handler = logging.StreamHandler(sys.stdout)
stdout_log_handler.setFormatter(consolelogformatter)
logger.addHandler(stdout_log_handler)
def parse_args():
parser = argparse.ArgumentParser(description='Itho WPU i2c master')
actions = [
"getregelaar",
"getserial",
"getdatatype",
"getdatalog",
]
parser.add_argument('--action', nargs='?', required=True,
choices=actions, help="Execute an action")
parser.add_argument('--loglevel', nargs='?',
choices=["debug", "info", "warning", "error", "critical"],
help="Loglevel")
args = parser.parse_args()
return args
class I2CMaster:
def __init__(self, address, bus):
self.i = i2c_raw.I2CRaw(address=address, bus=bus)
def execute_action(self, action):
actions = {
"getregelaar": [0x80, 0x90, 0xE0, 0x04, 0x00, 0x8A],
"getserial": [0x80, 0x90, 0xE1, 0x04, 0x00, 0x89],
"getdatatype": [0x80, 0xA4, 0x00, 0x04, 0x00, 0x56],
"getdatalog": [0x80, 0xA4, 0x01, 0x04, 0x00, 0x55],
}
logger.info(f"Executing action: {action}")
self.i.write_i2c_block_data(actions[action])
def close(self):
self.i.close()
if __name__ == "__main__":
args = parse_args()
if args.loglevel:
logger.setLevel(args.loglevel.upper())
master = I2CMaster(address=0x41, bus=1)
if args.action:
master.execute_action(args.action)
master.close()