1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
"""Generate python files to access the stiebel eltron heat pumps."""
from pathlib import Path
import csv
from dataclasses import dataclass
from jinja2 import Environment, FileSystemLoader, Template, select_autoescape
from pystiebeleltron import RegisterType
ENERGY_DATA_BLOCK_NAME = "Energy Data"
@dataclass
class ModbusFile:
"""Register file class."""
name: str
path: Path
register_type: RegisterType
def python_name(name: str, suffix: str = "") -> str:
"""Generate a valid python variable name."""
if suffix == "":
result = name.strip()
else:
result = name + "_" + suffix.strip()
return result.replace(" ", "_")
def python_class_name(name: str) -> str:
"""Generate a valid python variable name."""
return name.replace(" ", "")
def float_or_none(value: str) -> float | None:
"""Calculate a float value or none."""
if value == "":
return None
return float(value)
def get_base_address(registers: list) -> int:
"""Get the base address from a register block."""
result = 35000
for register in registers:
result = min(result, int(register[0]))
return result - 1
def get_block_len(registers: list) -> int:
"""Get the base address from a register block."""
base = 35000
highest = 0
for register in registers:
base = min(base, int(register[0]))
highest = max(highest, int(register[0]))
return highest - base + 1
def generate_heatpump(
root_path: Path,
template: Template,
modbus_files: list[ModbusFile],
heatpump_type: str,
):
"""Generate the python file for a heat pump."""
register_blocks = []
for modbus_file in modbus_files:
with modbus_file.path.open("r") as f:
reader = csv.reader(f)
data = list(reader)
api_data = data[1:]
register_blocks.append(
{
"name": modbus_file.name,
"count": get_block_len(api_data),
"base_address": get_base_address(api_data),
"register_block": api_data,
"register_type": modbus_file.register_type,
}
)
generated_content = template.render(registers=register_blocks, heatpump_type=heatpump_type)
generated_path = root_path / f"pystiebeleltron/{heatpump_type.lower()}.py"
with generated_path.open("w") as f:
f.write(generated_content)
def main() -> None:
"""Generate the python files."""
root = Path.cwd()
scripts_path = Path(__file__).parent
env = Environment(
loader=FileSystemLoader(str(scripts_path / "templates")),
autoescape=select_autoescape(),
)
env.globals.update(python_name=python_name, python_class_name=python_class_name, float_or_none=float_or_none, ENERGY_DATA_BLOCK_NAME=ENERGY_DATA_BLOCK_NAME)
wpm_template = env.get_template("wpmtemplate.j2")
wpm_modbus_files = [
ModbusFile(
name="System Values",
path=root / "api/wpm_system_values.csv",
register_type=RegisterType.INPUT_REGISTER,
),
ModbusFile(
name="System Parameters",
path=root / "api/wpm_system_parameters.csv",
register_type=RegisterType.HOLDING_REGISTER,
),
ModbusFile(
name="System State",
path=root / "api/wpm_system_state.csv",
register_type=RegisterType.INPUT_REGISTER,
),
ModbusFile(
name="Energy Data",
path=root / "api/wpm_energy_data.csv",
register_type=RegisterType.INPUT_REGISTER,
),
]
generate_heatpump(root, wpm_template, wpm_modbus_files, "Wpm")
lwz_template = env.get_template("lwztemplate.j2")
lwz_modbus_files = [
ModbusFile(
name="System Values",
path=root / "api/lwz_system_values.csv",
register_type=RegisterType.INPUT_REGISTER,
),
ModbusFile(
name="System Parameters",
path=root / "api/lwz_system_parameters.csv",
register_type=RegisterType.HOLDING_REGISTER,
),
ModbusFile(
name="System State",
path=root / "api/lwz_system_state.csv",
register_type=RegisterType.INPUT_REGISTER,
),
ModbusFile(
name="Energy Data",
path=root / "api/lwz_energy_data.csv",
register_type=RegisterType.INPUT_REGISTER,
),
]
generate_heatpump(root, lwz_template, lwz_modbus_files, "Lwz")
if __name__ == "__main__":
main()
print("Done!")
|
