client.py

#  client.py
#  
#  Copyright 2015 Ericson Willians (Rederick Deathwill) <EricsonWRP@ERICSONWRP-PC>
#  
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#  
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#  
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
#  
#  

import os
import json
import socket
import tkinter as tk	
import threading

CONNECTION_PORT = 9311

class Serializable:

	def __init__(self):
		self.file = None
		
	def serialize(self, path, mode):    
		try:
			self.file = open(path, mode)
		except:
			raise FileNotFoundError()
		if self.file:
			return self.file

class App(Serializable, threading.Thread):
	
	def __init__(self):
		Serializable.__init__(self)
		if os.path.isfile("user.cfg"):
			self.data = self.load("user.cfg")
		else:
			self.data = {
				"HOST": "127.0.0.1",
				"NICKNAME": "Unknown User"
			}
		self.is_running = True
		self.write("user.cfg")		
		threading.Thread.__init__(self)
		self.start()
		
	def write(self, path):
		self.serialize(path, "w").write(json.dumps(self.data))
		self.file.close()
			
	def load(self, path):
		json_data = open(path, "r")
		self.data = json.load(json_data)
		json_data.close()
		return self.data
	
	def center(self, win):
		win.update_idletasks()
		width = win.winfo_width()
		height = win.winfo_height()
		x = (win.winfo_screenwidth() // 2) - (width // 2)
		y = (win.winfo_screenheight() // 2) - (height // 2)
		win.geometry("{}x{}+{}+{}".format(width, height, x, y))
		
	def run(self):
		self.root = tk.Tk()
		self.root.protocol("WM_DELETE_WINDOW", self.quit)
		self.root.title("Social Pie Client")
		self.root.grid()
		self.root.grid_columnconfigure(0, weight=1)
		for n in range(7):
			self.root.grid_rowconfigure(n, weight=2)
		self.create_widgets()
		self.console.insert(tk.END, "Social Pie initiated.\n")
		self.connected = False
		self.center(self.root)
		self.root.mainloop()
	
	def quit(self):
		self.is_running = False
		self.root.destroy()

	def create_widgets(self):
		self.console = tk.Text(self.root, bg="#000", fg="#0F0", highlightcolor="#F00", highlightthickness=2)
		self.console.grid(column=0, row=0, padx=25, pady=10, sticky=tk.W+tk.E)
		self.msg_label = tk.Label(self.root, text="Message: ")
		self.msg_label.grid(column=0, row=1, pady=10, sticky=tk.W+tk.E)
		self.msg_area = tk.Text(self.root, height=6, width=58, bg="#000", fg="#0F0", highlightcolor="#F00", highlightthickness=2)
		self.msg_area.grid(column=0, row=2, padx=25, pady=10, sticky=tk.W+tk.E)
		self.msg_area.bind("<Return>", self.send)
		self.connect_button = tk.Button(self.root, text="Connect", command=self.connect)
		self.connect_button.grid(column=0, row=3, padx=30, pady=10, sticky=tk.W)
		self.quit_button = tk.Button(self.root, text="Exit", command=self.quit)
		self.quit_button.grid(column=0, row=3, padx=110, pady=10, sticky=tk.W)
	
	def send(self, x=None):
		self.sock.sendall(bytes(json.dumps({
			"data": self.msg_area.get("1.0", tk.END),
			"user": self.data["NICKNAME"]
		}), "utf-8"))
		
	def connect(self, x=None):
		if not self.connected:
			self.host = self.data["HOST"]
			self.console.insert(tk.END, "Connecting to {host}...\n".format(host = self.host))
			done = False
			try:
				self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
				self.sock.connect((self.host, CONNECTION_PORT))
				self.console.insert(tk.END, "Connection established.\n")
				self.connect_button.config(text="Disconnect")
				self.connected = True
			except Exception as e:
				self.console.insert(tk.END, "Error trying to connect to {host}: {msg}\n".format(host = self.host, msg = str(e)))
		else:
			self.sock.close()
			self.rsock.close()
			self.console.insert(tk.END, "Connection with {host} destroyed.\n".format(host = self.host))
			self.connect_button.config(text="Connect")
			self.connected = False

if __name__ == '__main__':
	app = App()
	while app.is_running:
		try:
			received = app.sock.recv(10000).decode("utf-8")
			app.console.insert(tk.END, "{msg}\n".format(msg = received))
			self.msg_area.delete("1.0", tk.END)
			self.msg_area.focus()
		except:
			pass