mirror of
https://github.com/pommi/python-itho-wpu.git
synced 2024-11-24 14:22: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:
parent
4e6d4c8666
commit
91913c4fe7
29
i2c_raw.py
Normal file
29
i2c_raw.py
Normal 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
62
itho-wpu.py
Executable 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()
|
Loading…
Reference in New Issue
Block a user