rendered paste body#! usr/bin/env python
"Import modules from python."
import os, sys
import math, random
"Import the pygame module."
try:
import pygame
from pygame.locals import *
except ImportError:
print 'This requires PyGame. You can download it at pygame.org.'
"Import pygu!"
try:
import pygu
from pygu.locals import *
except ImportError:
print 'This requires PyGU.'
from socket import *
import socket as sock
import thread, time
import ConfigParser
class Cursor(pygu.gui.Cursor):
def __init__(self):
pygu.gui.Cursor.__init__(self, False)
self.image = pygu.image.load('data/Mouse Cursor.bmp', True)
self.cursor = pygu.image.load('data/Mouse Cursor.bmp', True)
self.cursor2 = pygu.image.load('data/Text Cursor.bmp', True)
def change_type(self, type):
if type == 'text':
self.image = self.cursor2
if type == 'cursor':
self.image = self.cursor
def draw(self, surface):
if pygame.mouse.get_focused():
surface.blit(self.image, (self.rect[0], self.rect[1]-8))
def now():
return time.strftime("%I:%M:%S %p",time.localtime())
def send():
global nick
if window.get_widget(1).text is not "":
if window.get_widget(1).text.startswith('/me'):
message = str("[" + now() + "]" + " * " + nick + window.get_widget(1).text[3:])
if s.send(message):
window.get_widget(1).text = ""
elif window.get_widget(1).text.startswith('/nick '):
oldnick = nick
nick = window.get_widget(1).text[6:]
sendadminote(oldnick + " is now known as " + nick)
window.get_widget(1).text = ""
else:
message = str("[" + now() + "]" + " <" + nick + "> " + window.get_widget(1).text)
if s.send(message):
window.get_widget(1).text = ""
def sendadminote(msg):
s.sendto("[" + now() + "]" + " * " + str(msg), addr)
def show_msg(msg):
if not msg.startswith("New user: "):
window.get_widget(0).add_message(msg)
else:
users.append(msg[10:])
def change_nick():
global nick
oldnick = nick
nick = window.get_widget(2).text
if nick is not oldnick:
sendadminote(oldnick + " is now known as " + nick)
def ChatRead():
while 1:
data = s.recv(1024)
show_msg(data)
print `data`
#s.close()
return
def ChatWrite():
while 1:
s.send(data)
time.sleep(0.1)
s.close()
return
def connect():
window.get_widget(0).add_message("Connecting to: " + host + " at port " + str(port))
draw()
draw()
s = socket(AF_INET, SOCK_STREAM)
s.connect(addr)
s.send("Name: " + nick)
time.sleep(0.1)
draw()
thread.start_new(ChatRead, ())
thread.start_new(ChatWrite, ())
return s
def draw():
screen.fill(BLUE_BG)
window.render(screen)
window.update(event)
ren = font2.render("PyChat", 1, WHITE)
screen.blit(ren, (256-ren.get_width()/2, -3))
ren = font.render(now(), 1, BLACK)
screen.blit(ren, (451, 36))
ren = font.render(now(), 1, WHITE)
screen.blit(ren, (450, 35))
ren = font.render("Msg:", 1, BLACK)
screen.blit(ren, (6, 352))
ren = font.render("Msg:", 1, WHITE)
screen.blit(ren, (5, 351))
ren = font.render("Nick:", 1, BLACK)
screen.blit(ren, (6, 379))
ren = font.render("Nick:", 1, WHITE)
screen.blit(ren, (5, 378))
screen.blit(logo, (5, 5))
cursor.update()
cursor.draw(screen)
pygame.display.flip()
pygame.init()
screen = pygu.display.screen((512, 400), 'Michael Burns - Python Powered Chat Program')
font = pygu.font.Font('data/GUI Font.TTF', 8)
font2 = pygu.font.Font('data/Title Font.ttf', 40)
window = pygu.gui.MainWindow()
window.add_widget(pygu.gui.InputBox(font=font, size=[206, 15], pos=[150, 200], fill_color=[25, 50, 125], text_color=WHITE, line_color=BLUE_BUTTON))
clock = pygame.time.Clock()
cursor = Cursor()
pygame.mouse.set_visible(0)
logo = pygu.image.load('data/python-powered-w-100x40.png')
while 1:
clock.tick(60)
event = pygame.event.poll()
if event.type == QUIT:
sys.exit()
if event.type == KEYDOWN:
if event.key == K_ESCAPE:
pygame.quit()
if event.key == K_RETURN or event.key == K_KP_ENTER:
break
if cursor.rect.colliderect(window.get_widget(0).rect):
cursor.change_type('text')
else:
cursor.change_type('cursor')
screen.fill(BLUE_BG)
pygame.draw.rect(screen, [25, 50, 125], [90, 170, 326, 100])
pygame.draw.rect(screen, BLUE_BUTTON, [90, 170, 326, 100], 1)
window.render(screen)
window.update(event)
ren = font2.render("PyChat", 1, WHITE)
screen.blit(ren, (256-ren.get_width()/2, -3))
ren = font.render("Enter a Nickname:", 1, BLACK)
screen.blit(ren, (257-ren.get_width()/2, 182))
ren = font.render("Enter a Nickname:", 1, WHITE)
screen.blit(ren, (256-ren.get_width()/2, 181))
ren = font.render("Press Enter to continue", 1, BLACK)
screen.blit(ren, (257-ren.get_width()/2, 226))
ren = font.render("Press Enter to continue", 1, WHITE)
screen.blit(ren, (256-ren.get_width()/2, 225))
ren = font.render("Make sure you open \"Server.py\" before hitting enter.", 1, BLACK)
screen.blit(ren, (257-ren.get_width()/2, 242))
ren = font.render("Make sure you open \"Server.py\" before hitting enter.", 1, WHITE)
screen.blit(ren, (256-ren.get_width()/2, 241))
ren = font.render("Use \"Settings.ini\" to change the server and ports.", 1, BLACK)
screen.blit(ren, (257-ren.get_width()/2, 252))
ren = font.render("Use \"Settings.ini\" to change the server and ports.", 1, WHITE)
screen.blit(ren, (256-ren.get_width()/2, 251))
screen.blit(logo, (5, 5))
nick = window.get_widget(0).text
cursor.update()
cursor.draw(screen)
pygame.display.flip()
config = ConfigParser.ConfigParser()
config.read("settings.ini")
host = config.get("server", "host")
port = int(config.get("server", "port"))
# Set the socket parameters
host = host
port = port
buf = 1024
addr = (host, port)
users = []
window = pygu.gui.MainWindow()
window.add_widget(pygu.gui.MessageBox(font=font, size=[502, 290], pos=[5,50], fill_color=[25, 50, 125], text_color=WHITE, line_color=BLUE_BUTTON))
window.add_widget(pygu.gui.InputBox(font=font, size=[380, 15], pos=[40, 350], fill_color=[25, 50, 125], text_color=WHITE, line_color=BLUE_BUTTON))
window.add_widget(pygu.gui.InputBox(text=nick, font=font, size=[380, 15], pos=[40, 377], fill_color=[25, 50, 125], text_color=WHITE, line_color=BLUE_BUTTON))
window.add_widget(pygu.gui.Button(text="Send", font=font, command=send, size=[75, 20], pos=[429, 348], button_color=BLUE_BUTTON))
window.add_widget(pygu.gui.Button(text="Change", font=font, command=change_nick, size=[75, 20], pos=[429, 375], button_color=BLUE_BUTTON))
def ChatRead():
while 1:
data = s.recv(1024)
show_msg(data)
print `data`
#s.close()
return
def ChatWrite():
while 1:
s.send(data)
time.sleep(0.1)
s.close()
return
# Create socket
try:
window.get_widget(0).add_message("Connecting to: " + host + " at port " + str(port))
draw()
draw()
s = socket(AF_INET, SOCK_STREAM)
s.connect(addr)
s.send("Name: " + nick)
window.get_widget(0).add_message("Connected to: " + host + " at port " + str(port))
time.sleep(0.1)
sendadminote(nick + " joins the chat")
draw()
thread.start_new(ChatRead, ())
thread.start_new(ChatWrite, ())
except sock.error, (error, text):
t = "ERROR: " + text
window.get_widget(0).add_message(str(t))
print s
while 1:
clock.tick(60)
event = pygame.event.poll()
if event.type == QUIT:
sendadminote(nick + " quit")
#s.close()
sys.exit()
if event.type == KEYDOWN:
if event.key == K_RETURN:
if window.get_widget(1).selected:
send()
if window.get_widget(2).selected:
change_nick()
if cursor.rect.colliderect(window.get_widget(1).rect):
cursor.change_type('text')
elif cursor.rect.colliderect(window.get_widget(2).rect):
cursor.change_type('text')
else:
cursor.change_type('cursor')
draw()