-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathgroup_5_publisher.py
327 lines (277 loc) · 14 KB
/
group_5_publisher.py
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
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
import json
import threading
import random
from tkinter import *
from tkinter import messagebox
from tkinter.ttk import *
import argparse
import time
import sys
import paho.mqtt.client as mqtt
import logging
import socket
from group_5_data_generator import SensorSimulator
# Configure logging
logging.basicConfig(filename='publisher.log', level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s')
class PublisherGUI(Tk):
sensors_address = {
"LivingRoom": "123.89.46.72",
"Kitchen": "123.89.46.44",
"Bath Room": "123.89.46.56",
"Master Bedroom": "123.89.46.98",
"Dining Room": "123.89.46.34",
"Play Room": "123.89.46.65",
"Laundry": "123.89.46.89"
}
def __init__(self, topic_name):
super().__init__()
self.title(f'Temperature Publisher - {topic_name}')
self.__topic = topic_name
# Initialize UI
self.create_vars()
self.create_ui()
self.configureResizable()
def generate_client_id():
timestamp = int(time.time())
random_value = random.randint(1000, 9999)
return f'pub_demo_{timestamp}_{random_value}'
# Create Mqtt client
self.mqttc = mqtt.Client(
mqtt.CallbackAPIVersion.VERSION2,
client_id= generate_client_id(),
protocol=mqtt.MQTTv5
)
# Register callbacks
self.mqttc.on_connect = self.on_connect
self.mqttc.on_disconnect = self.on_disconnect
self.mqttc.on_message = self.on_message
self.mqttc.on_publish = self.on_publish
def configureResizable(self):
max_row = 12
max_col = 2
for row_index in range(max_row):
Grid.rowconfigure(self, index=row_index, weight=1)
Grid.columnconfigure(self, index=0, weight=1)
Grid.columnconfigure(self, index=1, weight=3)
def create_vars(self):
self.__flag_status = False
self.__name = StringVar()
self.__time = StringVar()
self.__base = StringVar()
self.__min = StringVar()
self.__max = StringVar()
self.__delta = StringVar()
self.__min_step = StringVar()
self.__max_step = StringVar()
self.__min_cycle = StringVar()
self.__max_cycle = StringVar()
self.__squiggle = BooleanVar()
self.__button_name = StringVar(value='Start')
self.__status = StringVar(value='Standby')
# dropdown options
self.__minMaxValues = [18, 18.5, 19, 19.5, 20, 20.5, 21]
self.__stepsValues = [0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1]
self.__cycleValues = [1, 2, 3, 4]
self.__sensors_name = list(self.sensors_address.keys())
self.__time_intervals = [0.25, 0.5, 1, 1.5, 2, 2.5]
self.__max_iteration = 100
def create_ui(self, parent=None):
if not parent:
parent = self
Label(
parent,
text='Publisher Parameters',
font=('Arial Bold', 14),
anchor='c',
justify='center',
).grid(row=0, columnspan=2, sticky='nsew', pady=(20, 10))
Label(parent, text='Sensor Name:', font=('Arial', 10), anchor='w').grid(row=1, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Time Interval (seconds):', font=('Arial', 10), anchor='w').grid(row=2, column=0, sticky='nsew', pady=(0, 5), padx=(3, 0))
Label(parent, text='Starting Temp:', font=('Arial' ,10), anchor='w').grid(row=3, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Minimum Temp:', font=('Arial' ,10), anchor='w').grid(row=4, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Max Temp:', font=('Arial' ,10), anchor='w').grid(row=5, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Delta (Range: -1-1):', font=('Arial' ,10), anchor='w').grid(row=6, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Min Step:', font=('Arial' ,10), anchor='w').grid(row=7, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Max Step:', font=('Arial' ,10), anchor='w').grid(row=8, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Min Cycle:', font=('Arial' ,10), anchor='w').grid(row=9, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Max Cycle:', font=('Arial' ,10), anchor='w').grid(row=10, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
Label(parent, text='Squiggle:', font=('Arial' ,10), anchor='w').grid(row=11, column=0, sticky='nsew', pady=(0, 5), padx=(3,0))
# Create dropdowns and other widgets
nameComboBox = Combobox(parent, width=20, textvariable=self.__name)
nameComboBox['values'] = self.__sensors_name
nameComboBox.current(5)
nameComboBox.grid(row=1, column=1, sticky='W', pady=(0, 1))
timeComboBox = Combobox(parent, width=20, textvariable=self.__time)
timeComboBox['values'] = self.__time_intervals
timeComboBox.current(0)
timeComboBox.grid(row=2, column=1, sticky='W', pady=(0, 1))
minComboBox = Combobox(parent, width=20, textvariable=self.__base)
minComboBox['values'] = self.__minMaxValues
minComboBox.current(1)
minComboBox.grid(row=3, column=1, sticky='W', pady=(0, 5))
minComboBox = Combobox(parent, width=20, textvariable=self.__min)
minComboBox['values'] = self.__minMaxValues
minComboBox.current(0)
minComboBox.grid(row=4, column=1, sticky='W', pady=(0, 5))
maxComboBox = Combobox(parent, width=20, textvariable=self.__max)
maxComboBox['values'] = self.__minMaxValues
maxComboBox.current(6)
maxComboBox.grid(row=5, column=1, sticky='W', pady=(0, 5))
e = Entry(
parent,
textvariable=self.__delta,
width=23
)
e.insert(0, '1')
e.grid(row=6, column=1, sticky=W, pady=(0, 5))
minStepComboBox = Combobox(parent, width=20, textvariable=self.__min_step)
minStepComboBox['values'] = self.__stepsValues
minStepComboBox.current(0)
minStepComboBox.grid(row=7, column=1, sticky='W', pady=(0, 5))
maxStepComboBox = Combobox(parent, width=20, textvariable=self.__max_step)
maxStepComboBox['values'] = self.__stepsValues
maxStepComboBox.current(6)
maxStepComboBox.grid(row=8, column=1, sticky='W', pady=(0, 5))
minCycleComboBox = Combobox(parent, width=20, textvariable=self.__min_cycle)
minCycleComboBox['values'] = self.__cycleValues
minCycleComboBox.current(0)
minCycleComboBox.grid(row=9, column=1, sticky='W', pady=(0, 5))
maxCycleComboBox = Combobox(parent, width=20, textvariable=self.__max_cycle)
maxCycleComboBox['values'] = self.__cycleValues
maxCycleComboBox.current(3)
maxCycleComboBox.grid(row=10, column=1, sticky='W', pady=(0, 5))
check = Checkbutton(parent, text="Squiggle", variable=self.__squiggle)
check.grid(row=11, column=1, sticky='W', pady=(0, 5))
Button(parent, textvariable=self.__button_name, command=self.btn_click).grid(row=12, columnspan=2, pady=(10, 20))
Label(parent, textvariable=self.__status, font=('Arial', 10), anchor='w').grid(row=13, columnspan=2, sticky='nsew')
def btn_click(self):
if self.__button_name.get() == 'Start':
self.__button_name.set('Stop')
self.__flag_status = True
# parse values to float for validation
try:
parsedBase = float(self.__base.get())
parsedMin = float(self.__min.get())
parsedMax = float(self.__max.get())
parsedMinStep = float(self.__min_step.get())
parsedMaxStep = float(self.__max_step.get())
parsedMinCycle = int(self.__min_cycle.get())
parsedMaxCycle = int(self.__max_cycle.get())
parsedName = self.__name.get()
parsedInterval = float(self.__time.get())
# Validate delta value
parsedDelta = float(self.__delta.get())
if parsedDelta < -1 or parsedDelta > 1:
messagebox.showinfo(title='Information', message="Delta must be from -1 to 1")
logging.warning("Delta value out of range: %s", parsedDelta)
return
if parsedMaxCycle <= parsedMinCycle:
messagebox.showinfo(title='Information', message="Max cycle must be greater than min cycle")
logging.warning("Max cycle not greater than min cycle: Min %d, Max %d", parsedMinCycle, parsedMaxCycle)
return
# Connect to MQTT broker
try:
self.mqttc.connect(host='broker.hivemq.com', port=1883)
logging.info("MQTT client connected")
except (socket.error, mqtt.MQTTException) as e:
messagebox.showinfo(title='Information', message=f'Error connecting to MQTT broker: {e}')
logging.error("MQTT connection error: %s", e)
return
# Start data publishing thread
thread = threading.Thread(
target=self.run,
args=[parsedBase, parsedMin, parsedMax, parsedDelta, parsedMinStep, parsedMaxStep, parsedMinCycle, parsedMaxCycle, parsedName, parsedInterval])
thread.daemon = True
thread.start()
except ValueError as e:
messagebox.showinfo(title='Information', message=f'Error parsing input: {e}')
logging.error("ValueError: %s", e)
except Exception as e:
messagebox.showinfo(title='Information', message=f'Unexpected error: {e}')
logging.error("Unexpected error: %s", e)
else:
self.__button_name.set('Start')
self.__status.set('Standby')
self.__flag_status = False
def run(self, parsedBase, parsedMin, parsedMax, parsedDelta, parsedMinStep, parsedMaxStep, parsedMinCycle, parsedMaxCycle, parsedName, parsedInterval):
tempGenerator = SensorSimulator(
min_value=parsedMin, max_value=parsedMax, delta=parsedDelta,
base_value=parsedBase, min_step=parsedMinStep, max_step=parsedMaxStep,
min_cycle=parsedMinCycle, max_cycle=parsedMaxCycle,
squiggle=self.__squiggle.get()
)
miss_transmission = random.randint(1, self.__max_iteration)
wild_transmission = random.randint(1, self.__max_iteration)
iteration = 0
while self.__flag_status:
iteration += 1
if iteration > self.__max_iteration:
iteration = 0
miss_transmission = random.randint(1, self.__max_iteration)
wild_transmission = random.randint(1, self.__max_iteration)
if miss_transmission == iteration:
time.sleep(parsedInterval)
continue
temp = tempGenerator.value
if wild_transmission == iteration:
temp = temp * random.randint(2,10)
try:
packetId = int(time.time() * 1000)
msg_dict = {
"packetId": packetId,
"name": parsedName,
"ipv4": self.sensors_address[parsedName],
"temp": temp,
"interval": parsedInterval
}
data = json.dumps(msg_dict, indent=4, sort_keys=True, default=str)
self.mqttc.publish(topic=self.__topic, payload=data, qos=0)
self.__status.set(f'Packet Sending: {msg_dict["packetId"]}')
logging.info("Published message: %s", msg_dict)
time.sleep(parsedInterval)
except (KeyboardInterrupt, SystemExit):
self.mqttc.disconnect()
logging.info("MQTT client disconnected due to exit")
sys.exit()
except mqtt.MQTTException as e:
messagebox.showinfo(title='Information', message=f'Error publishing message: {e}')
logging.error("MQTT publish error: %s", e)
break
except socket.error as e:
messagebox.showinfo(title='Information', message=f'Network error: {e}')
logging.error("Network error: %s", e)
break
def on_connect(self, mqttc, userdata, flags, rc):
logging.info('Connected to MQTT broker. Return code: %s', rc)
def on_disconnect(self, client, userdata, flags, reason, properties):
logging.info('Disconnected from MQTT broker. Return code: %s', reason)
def on_message(self, mqttc, userdata, msg):
logging.info('Received message. Topic: %s, Message: %s', msg.topic, msg.payload)
def on_publish(self, client, userdata, mid, reason, properties):
logging.info('Published message. MID: %s, Reason: %s', mid, reason)
def publish(self, topic, payload, qos=0):
try:
self.mqttc.publish(topic=topic, payload=payload, qos=qos)
logging.info('Published message to topic: %s', topic)
except mqtt.MQTTException as e:
logging.error("MQTT publish error: %s", e)
except socket.error as e:
logging.error("Network error: %s", e)
return
def disconnect(self):
try:
self.mqttc.disconnect()
logging.info('MQTT client disconnected')
except mqtt.MQTTException as e:
logging.error("MQTT disconnection error: %s", e)
except socket.error as e:
logging.error("Network error during disconnection: %s", e)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--topic", help="Topic name", required=True, type=str)
args = parser.parse_args()
rmPub = PublisherGUI(topic_name=args.topic)
rmPub.geometry("400x300")
rmPub.minsize(width=450, height=400)
rmPub.mainloop()