basic.py

@@ -3,14 +3,42 @@
 #Hostname: learn.selfmade.ninja
 
 import socket
+from threading import Thread
 
 HOST = ''
 PORT = 3074
 
+def start_chat_thread(conn, addr):
+	t = ChatMsgThread(conn, addr)
+	t.start()
+	print("Thread created and offloaded connection with IP {}:{}".format(addr[0], addr[1]))
+	
+class ChatMsgThread(Thread):
+	def __init__(self, conn, addr):
+		Thread.__init__(self)
+		self.conn = conn
+		self.addr = addr
+		#print("IP {} connected".format(addr[0]))
+		
+	def run(self):
+		self.conn.sendall(b"Send your message, I will be able to see it here.\n")
+		while(True):
+			try:
+				data = self.conn.recv(2048)
+				if not data:
+					self.conn.close()
+					continue
+				else:
+					try:
+						data = data.decode()
+						print(data)
+					except:
+						pass
+			except:
+				pass
+						
+
 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-					#AF - Address Family
-					#INET - Internet
-					
 s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 s.bind((HOST, PORT))
 s.listen()
@@ -19,10 +47,7 @@ while(True):
 	#conn - the connection object with the client
 	#addr - array 0 - IP address of the client 
 	#		array 1 - The back port with which the connection is established
-	print("IP {} connected".format(addr[0]))
-	print(addr)
-	conn.sendall(b"Hello, welcome to this workshop")
-	conn.close()
+	start_chat_thread(conn, addr)
 s.close()
-	
-	
\ No newline at end of file
+
+

client.py

@@ -15,13 +15,13 @@ s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 					
 s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 s.connect((HOST, PORT))
-
+print(s.recv(2048).decode())
 while(True):
 	msg = input("Message: ")
 	if(msg == "quit"):
 		break
 	msg = name + ": " + msg
-	s.send(msg.encode())
+	s.sendall(msg.encode())
 s.close()
 	
 

mqtt-chat-npy.py

+#!/usr/bin/env python3
+
+import paho.mqtt.client as mqtt
+from threading import Thread
+import npyscreen
+
+HOST = 'broker.hivemq.com'
+PORT = 1883
+TIMEOUT = 60
+f = None
+TOPIC = 'lahtp-sockets-group'
+
+class SubThread(Thread):
+	def __init__(self):
+		Thread.__init__(self)
+		
+class ChatForm(npyscreen.SplitForm, npyscreen.ActionForm):
+	def create(self):
+		client.connect(HOST, PORT, TIMEOUT)
+		self.draw_line_at = 18
+		self.MOVE_LINE_ON_RESIZE = True
+		self.chat_screen = self.add(npyscreen.MultiLine, name="MultiLineChatScreen", height=18)
+		self.chat_screen.display()
+		self.message_field = self.add(npyscreen.TitleText, name="Send: ", editable=True)
+		
+	def on_ok(self):
+		#npyscreen.notify_confirm(self.message_field.value)
+		client.publish(TOPIC, self.message_field.value)
+		self.message_field.value = ""
+		
+		
+		
+	def on_cancel(self):
+		exit(0)
+		
+	def add_msg(message):
+		if(self.chat_screen.value is None):
+			self.chat_screen.value = message
+		else:
+			self.chat_screen.value = self.chat_screen.value + "\n" + message
+		self.chat_screen.display()
+
+def on_connect(client, userdata, flags, rc):
+	npyscreen.notify("Connection Established", title="Info")
+
+def on_message(client, userdata, message):
+	f = open("log.txt", "a")
+	f.write(message.payload)
+	#npyscreen.notify(message, title="New Message")
+	if f is not None:
+		f.add_msg(message.payload)
+	
+class App(npyscreen.NPSAppManaged):
+	def onStart(self):
+		f = self.addForm("MAIN", ChatForm)
+
+		
+if __name__=="__main__":
+	client = mqtt.Client()
+	client.on_connect = on_connect
+	client.on_message = on_message
+	#f.chat_screen.value = "Welcome.\n"
+	app = App()
+	app.run()
\ No newline at end of file

mqtt_basics.py

+#!/usr/bin/env python3
+
+import paho.mqtt.client as mqtt
+
+HOST = 'broker.hivemq.com'
+PORT = 1883
+TIMEOUT = 60
+
+def on_message(client, userdata, message):
+	print("{}: {}".format(message.topic, str(message.payload)))
+	
+def on_connect(client, userdata, flags, rc):
+	print("Connected.")
+	client.subscribe('lahtp-sockets-group')
+
+client = mqtt.Client()
+client.on_message = on_message
+client.on_connect = on_connect
+
+client.connect(HOST, PORT, TIMEOUT)
+client.loop_forever()