Added basic chat example.
This commit is contained in:
parent
a2706cfd25
commit
f40536802b
26
examples/chat app/client/client.py
Normal file
26
examples/chat app/client/client.py
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
""" client communication for this simple chat example application.
|
||||||
|
This is a class derived from enetcomponents.client.client, which sends all data over pubsub via the topic response.
|
||||||
|
"""
|
||||||
|
from enetcomponents import client
|
||||||
|
from pubsub import pub
|
||||||
|
|
||||||
|
class client(client.client):
|
||||||
|
|
||||||
|
def network(self, event, data):
|
||||||
|
""" This functions receives data from an enet server in the following protocol:
|
||||||
|
dict(action="some_command", **kwargs)
|
||||||
|
This function will send all data to whatever listener in the pubsub stack by using the topic "response".
|
||||||
|
"""
|
||||||
|
f = data.get("action")
|
||||||
|
if f == None:
|
||||||
|
print("Error: Invalid data in protocol. %r" % (data))
|
||||||
|
return
|
||||||
|
pub.sendMessage("response", data=data)
|
||||||
|
|
||||||
|
def connected(self, peer):
|
||||||
|
pub.sendMessage("ask_login")
|
||||||
|
|
||||||
|
def disconnected(self, peer):
|
||||||
|
pub.sendMessage("disconnected")
|
||||||
|
|
90
examples/chat app/client/gui.py
Normal file
90
examples/chat app/client/gui.py
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
""" Dfinition of all GUI components. """
|
||||||
|
import wx
|
||||||
|
|
||||||
|
class loginDialog(wx.Dialog):
|
||||||
|
|
||||||
|
def __init__(self, title="Login"):
|
||||||
|
super(loginDialog, self).__init__(parent=None, id=wx.ID_ANY)
|
||||||
|
self.SetTitle(title)
|
||||||
|
panel = wx.Panel(self)
|
||||||
|
sizer = wx.BoxSizer(wx.VERTICAL)
|
||||||
|
label1 = wx.StaticText(panel, wx.ID_ANY, "Username: ")
|
||||||
|
self.username = wx.TextCtrl(panel, wx.ID_ANY, size=(380, -1))
|
||||||
|
s = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
s.Add(label1, 0, wx.ALL, 5)
|
||||||
|
s.Add(self.username, 0, wx.ALL, 5)
|
||||||
|
sizer.Add(s, 0, wx.ALL, 5)
|
||||||
|
label1 = wx.StaticText(panel, wx.ID_ANY, "Server: ")
|
||||||
|
self.server = wx.TextCtrl(panel, wx.ID_ANY, "localhost", size=(380, -1))
|
||||||
|
s = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
s.Add(label1, 0, wx.ALL, 5)
|
||||||
|
s.Add(self.server, 0, wx.ALL, 5)
|
||||||
|
sizer.Add(s, 0, wx.ALL, 5)
|
||||||
|
label1 = wx.StaticText(panel, wx.ID_ANY, "Port: ")
|
||||||
|
self.port = wx.SpinCtrl(panel, wx.ID_ANY, size=(380, -1))
|
||||||
|
self.port.SetRange(1024, 65000)
|
||||||
|
self.port.SetValue(33333)
|
||||||
|
s = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
s.Add(label1, 0, wx.ALL, 5)
|
||||||
|
s.Add(self.port, 0, wx.ALL, 5)
|
||||||
|
sizer.Add(s, 0, wx.ALL, 5)
|
||||||
|
|
||||||
|
# label2 = wx.StaticText(panel, wx.ID_ANY, "Password: ")
|
||||||
|
# self.password = wx.TextCtrl(panel, wx.ID_ANY, size=(380, -1), style=wx.TE_PASSWORD)
|
||||||
|
# ss = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
# ss.Add(label2, 0, wx.ALL, 5)
|
||||||
|
# ss.Add(self.password, 0, wx.ALL, 5)
|
||||||
|
# sizer.Add(ss, 0, wx.ALL, 5)
|
||||||
|
ok = wx.Button(panel, wx.ID_OK, "Log in")
|
||||||
|
ok.SetDefault()
|
||||||
|
cancel = wx.Button(panel, wx.ID_CANCEL)
|
||||||
|
self.SetEscapeId(wx.ID_CANCEL)
|
||||||
|
bs = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
bs.Add(ok, 0, wx.ALL, 5)
|
||||||
|
bs.Add(cancel, 0, wx.ALL, 5)
|
||||||
|
sizer.Add(bs, 0, wx.ALL, 5)
|
||||||
|
panel.SetSizer(sizer)
|
||||||
|
self.SetClientSize(sizer.CalcMin())
|
||||||
|
|
||||||
|
class appFrame(wx.Frame):
|
||||||
|
def __init__(self):
|
||||||
|
super(appFrame, self).__init__(parent=None, title="Chat Window")
|
||||||
|
self.Maximize(True)
|
||||||
|
self.panel = wx.Panel(self)
|
||||||
|
self.sizer = wx.BoxSizer(wx.VERTICAL)
|
||||||
|
self.sb = self.CreateStatusBar()
|
||||||
|
lbl = wx.StaticText(self.panel, wx.ID_ANY, "menu")
|
||||||
|
self.list = wx.ListBox(self.panel, wx.ID_ANY)
|
||||||
|
self.sizer.Add(lbl, 0, wx.GROW)
|
||||||
|
self.sizer.Add(self.list, 1, wx.GROW)
|
||||||
|
lbl = wx.StaticText(self.panel, -1, "Chat")
|
||||||
|
self.chat = wx.TextCtrl(self.panel, -1)
|
||||||
|
sizerchat = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
sizerchat.Add(lbl, 0, wx.ALL, 5)
|
||||||
|
sizerchat.Add(self.chat, 0, wx.ALL, 5)
|
||||||
|
self.sizer.Add(sizerchat, 0, wx.ALL, 5)
|
||||||
|
lbl1 = wx.StaticText(self.panel, wx.ID_ANY, "History")
|
||||||
|
self.history = wx.TextCtrl(self.panel, wx.ID_ANY, style=wx.TE_READONLY|wx.TE_MULTILINE, size=(500, 300))
|
||||||
|
box = wx.BoxSizer(wx.HORIZONTAL)
|
||||||
|
box.Add(lbl1, 0, wx.ALL, 5)
|
||||||
|
box.Add(self.history, 0, wx.ALL, 5)
|
||||||
|
self.sizer.Add(box, 0, wx.ALL, 5)
|
||||||
|
self.panel.SetSizerAndFit(self.sizer)
|
||||||
|
|
||||||
|
def get_item(self):
|
||||||
|
return self.list.GetSelection()
|
||||||
|
|
||||||
|
def add_message(self, message, reverse=False):
|
||||||
|
old_line = self.history.GetNumberOfLines()
|
||||||
|
point = self.history.GetInsertionPoint()
|
||||||
|
if reverse:
|
||||||
|
self.history.SetValue(message+"\n"+self.history.GetValue())
|
||||||
|
else:
|
||||||
|
self.history.AppendText(message+"\n")
|
||||||
|
self.history.SetInsertionPoint(point)
|
||||||
|
new_line = self.history.GetNumberOfLines()#.count("\n")
|
||||||
|
return (old_line, new_line)
|
||||||
|
|
||||||
|
def show_connection_error(self):
|
||||||
|
msg = wx.MessageDialog(None, "Connection error. Try again", "error", style=wx.ICON_ERROR).ShowModal()
|
91
examples/chat app/client/main.py
Normal file
91
examples/chat app/client/main.py
Normal file
@ -0,0 +1,91 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import sys
|
||||||
|
import threading
|
||||||
|
import wx
|
||||||
|
import client
|
||||||
|
import output
|
||||||
|
import gui
|
||||||
|
from pubsub import pub
|
||||||
|
|
||||||
|
# Client instance.
|
||||||
|
c = None
|
||||||
|
|
||||||
|
# thread to keep the client running.
|
||||||
|
t = None
|
||||||
|
|
||||||
|
class controller(object):
|
||||||
|
def __init__(self, window):
|
||||||
|
super(controller, self).__init__()
|
||||||
|
self.window = window
|
||||||
|
self.connect_events()
|
||||||
|
self.window.Show()
|
||||||
|
|
||||||
|
def connect_events(self):
|
||||||
|
self.window.chat.Bind(wx.EVT_CHAR_HOOK, self.on_process)
|
||||||
|
pub.subscribe(self.response, "response")
|
||||||
|
pub.subscribe(self.ask_login, "ask_login")
|
||||||
|
pub.subscribe(self.disconnected, "disconnected")
|
||||||
|
|
||||||
|
def on_process(self, event):
|
||||||
|
key = event.GetKeyCode()
|
||||||
|
if key == wx.WXK_RETURN:
|
||||||
|
self.send_message()
|
||||||
|
event.Skip()
|
||||||
|
|
||||||
|
def send_message(self):
|
||||||
|
global c
|
||||||
|
message = self.window.chat.GetValue()
|
||||||
|
if message == "" or message == None:
|
||||||
|
return wx.Bell()
|
||||||
|
# Otherwise, message does exist.
|
||||||
|
data = dict(action="send_message", message=message)
|
||||||
|
c.send_data(0, data)
|
||||||
|
self.window.chat.ChangeValue("")
|
||||||
|
|
||||||
|
def response(self, data):
|
||||||
|
command = data.get("action")
|
||||||
|
if hasattr(self, "cmd_"+command):
|
||||||
|
getattr(self, "cmd_"+command)(data)
|
||||||
|
|
||||||
|
def cmd_connected(self, data):
|
||||||
|
connected = data.get("nickname")
|
||||||
|
msg = "{} has entered this platform".format(connected)
|
||||||
|
self.window.add_message(msg)
|
||||||
|
|
||||||
|
def cmd_message(self, data):
|
||||||
|
msg = data.get("message")
|
||||||
|
nickname = data.get("nickname")
|
||||||
|
msg = "{0}: {1}".format(nickname, msg)
|
||||||
|
output.speak(msg)
|
||||||
|
self.window.add_message(msg)
|
||||||
|
|
||||||
|
def ask_login(self):
|
||||||
|
global c
|
||||||
|
data = dict(action="login", nickname=self.username)
|
||||||
|
c.send_data(0, data)
|
||||||
|
|
||||||
|
def disconnected(self):
|
||||||
|
self.window.show_connection_error()
|
||||||
|
wx.GetApp().ExitMainLoop()
|
||||||
|
|
||||||
|
def setup():
|
||||||
|
global c, t
|
||||||
|
output.setup()
|
||||||
|
app = wx.App()
|
||||||
|
d = gui.loginDialog()
|
||||||
|
f = gui.appFrame()
|
||||||
|
mainController = controller(f)
|
||||||
|
if d.ShowModal() != wx.ID_OK:
|
||||||
|
return
|
||||||
|
username = d.username.GetValue()
|
||||||
|
server = bytes(d.server.GetValue(), "utf-8")
|
||||||
|
port = int(d.port.GetValue())
|
||||||
|
mainController.username = username
|
||||||
|
d.Destroy()
|
||||||
|
c = client.client(host=server, port=port)
|
||||||
|
t = threading.Thread(target=c.run)
|
||||||
|
t.start()
|
||||||
|
app.MainLoop()
|
||||||
|
c.close()
|
||||||
|
|
||||||
|
setup()
|
31
examples/chat app/client/output.py
Normal file
31
examples/chat app/client/output.py
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
# *- coding: utf-8 -*-
|
||||||
|
""" Simple (really simple) module to provide text to speech output in a chat window."""
|
||||||
|
import logging as original_logging
|
||||||
|
logger = original_logging.getLogger('core.output')
|
||||||
|
from accessible_output2 import outputs
|
||||||
|
import sys
|
||||||
|
|
||||||
|
speaker = None
|
||||||
|
retries = 0
|
||||||
|
|
||||||
|
def speak(text, interrupt=0):
|
||||||
|
global speaker, retries
|
||||||
|
if not speaker:
|
||||||
|
setup()
|
||||||
|
try:
|
||||||
|
speaker.speak(text, interrupt)
|
||||||
|
except:
|
||||||
|
if retries < 5:
|
||||||
|
retries = retries + 1
|
||||||
|
speak(text)
|
||||||
|
|
||||||
|
def setup ():
|
||||||
|
global speaker
|
||||||
|
logger.debug("Initializing output subsystem.")
|
||||||
|
try:
|
||||||
|
speaker = outputs.auto.Auto()
|
||||||
|
except:
|
||||||
|
logger.exception("Output: Error during initialization.")
|
||||||
|
|
||||||
|
def enable_sapi():
|
||||||
|
speaker = outputs.sapi.SAPI5()
|
7
examples/chat app/requirements.txt
Normal file
7
examples/chat app/requirements.txt
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
pyenet
|
||||||
|
wxpython==4.0.3
|
||||||
|
pypubsub
|
||||||
|
pywin32
|
||||||
|
git+https://code.manuelcortez.net/manuelcortez/libloader
|
||||||
|
git+https://code.manuelcortez.net/manuelcortez/platform_utils
|
||||||
|
git+https://code.manuelcortez.net/manuelcortez/accessible_output2
|
51
examples/chat app/server/server.py
Normal file
51
examples/chat app/server/server.py
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
from enetcomponents import server
|
||||||
|
|
||||||
|
class channel(server.channel):
|
||||||
|
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super(channel, self).__init__(*args, **kwargs)
|
||||||
|
self.nickname = None
|
||||||
|
|
||||||
|
def network(self, event, data):
|
||||||
|
f = data.get("action")
|
||||||
|
if f == None:
|
||||||
|
print("Error: Invalid data in protocol. %r" % (data))
|
||||||
|
return
|
||||||
|
if hasattr(self, "cmd_"+f) == False:
|
||||||
|
print("Error: function cmd_{} does not exist".format(f))
|
||||||
|
return
|
||||||
|
getattr(self, "cmd_"+f)(data)
|
||||||
|
|
||||||
|
def cmd_login(self, data):
|
||||||
|
nickname = data.get("nickname")
|
||||||
|
self.nickname = nickname
|
||||||
|
self.room = "public"
|
||||||
|
d = dict(action="connected", nickname=nickname)
|
||||||
|
self.server.send_to_all(0, d)
|
||||||
|
|
||||||
|
def cmd_send_message(self, data):
|
||||||
|
data.update(nickname=self.nickname, action="message")
|
||||||
|
for channel in self.server.peers:
|
||||||
|
if channel.room == self.room:
|
||||||
|
channel.send_data(0, data)
|
||||||
|
|
||||||
|
class server(server.server):
|
||||||
|
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super(server, self).__init__(*args, **kwargs)
|
||||||
|
self.rooms = list()
|
||||||
|
|
||||||
|
def connected(self, peer):
|
||||||
|
p = channel(self, peer)
|
||||||
|
self.peers[p] = True
|
||||||
|
|
||||||
|
def disconnected(self, peer):
|
||||||
|
for channel in self.peers:
|
||||||
|
if peer.incomingPeerID == channel.peer.incomingPeerID:
|
||||||
|
del self.peers[channel]
|
||||||
|
break
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
print("Starting chat server...")
|
||||||
|
s = server()
|
||||||
|
s.run()
|
Loading…
Reference in New Issue
Block a user