server_proj.py

import socket		# for sockets
import re			# for regex
import picamera
import threading	# for threads
import os			# for using os.urandom()
import hashlib		# for hashing the passwords
import random		# for generating random confirmation code
import string		# for string.ascii_uppercase and string.digits
import smtplib		# for sending mails
import urllib		# for reading html files
from email.mime.text import MIMEText


#######################################################################################
#		IMPORTANT NOTES:
#
#	-Each message that the server sends to client MUST be ended with \n (flushed)
#		Thus a message cannot contain \n because otherwise it will be sent separately
#
#
#
######################################################################################


PORT = 5013
APP_NAME = 'AwesomeChatApp'
MAX_EMAIL_CHARS = 50
MIN_USERNAME_CHARS = 5
MAX_USERNAME_CHARS = 20
MIN_PASSWORD_CHARS = 10
MAX_PASSWORD_CHARS = 30
RANDOM_CHARS_COUNT = 32
CONFIRMATION_CODE_LENGTH = 7
EMAIL_FROM = APP_NAME
EMAIL_FROM_ADDRESS = '[email protected]'
SMTP_SERVER = "smtp.mail.yahoo.com"
SMTP_PORT = 587
SMTP_USERNAME = "[email protected]"
SMTP_PASSWORD = "Amiga1200"
BUFFER_SIZE=10240


server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind(("", PORT))
server_socket.listen(5)

clientThreads = []


class MessageType:
	CLOSE_CONNECTION = 'CLOSE_CONNECTION'
	NORMAL_MESSAGE = 'NORMAL_MESSAGE'
	QUERY = 'QUERY'
	RESPONSE_SUCCESS = 'RESPONSE_SUCCESS'
	RESPONSE_FAILED = 'RESPONSE_FAILED',
	FRIEND_REQUEST = 'FRIEND_REQUEST'


class Operation:
	CREATE_USER = 'CREATE_USER'
	TRY_LOGIN = 'TRY_LOGIN'
	VERIFY_CODE = 'VERIFY_CODE'
	LOGOUT = 'LOGOUT'
	CHECK_FRIEND_REQUESTS = 'CHECK_FRIEND_REQUESTS'
	SEND_IMAGE='SEND_IMAGE'


class ResponseMessage:
	NO_FRIEND_REQUESTS = 'NO_FRIEND_REQUESTS'




def removeClient(userIDtoRemove):
	print 'Removing user with id: ', userIDtoRemove
	for index, client in enumerate(clientThreads):
		if client.userID == userIDtoRemove:
			del clientThreads[index]
			print 'client: ', client.userID
			print 'index: ', index
			print ' has been removed!'
			break



def showClients():
	print 'Showing the clients...'
	for index, client in enumerate(clientThreads):
		print 'client: ', client.userID
		print 'index: ', index
	print ''




### Client thread ###


class ClientThread(threading.Thread):
	userID = None
	isOnline = False


	def __init__(self, socket, ip, port):
		threading.Thread.__init__(self)

		self.socket = socket
		self.ip = ip
		self.port = port

		print 'New thread started for: %s:%s' %  (str(ip), str(port))


	def run(self):

		while True:

			print '\nReceiving data from Client...'
			data = None
			try:
				data = self.socket.recv(512)
			except:
				print 'Connection lost!'
				removeClient(self.userID)
				break;
			print 'Received data: ', data

			showClients()

			# extract 'messageType' from the received message
			messageTypeAndRest = data.split(':', 1)
			messageType = messageTypeAndRest[0]
			rest = messageTypeAndRest[1]

			if messageType == MessageType.CLOSE_CONNECTION:
				print 'MessageType is CLOSE_CONNECTION'
				print 'Closing connection...'
				self.socket.close()
				break

			elif messageType == MessageType.NORMAL_MESSAGE:
				print 'MessageType is NORMAL_MESSAGE'

			elif messageType == MessageType.QUERY:
				print 'MessageType is QUERY'
				# extract 'operation' from the received message
				operationAndParams = messageTypeAndRest[1].split(':', 1)
				operation = operationAndParams[0]

				if operation == Operation.SEND_IMAGE:
					# get image
					# decompose image in bytes
					# in a while loop send 1024 bytes each time
					# when nothing to send, exit while
					# optional: send message that image has been successfully sent

					#activez camera si fac poza
					#stochez poza pe disk in acelasi folder cu server_proj.py + extensia .png
					#filename4324.png, filename142.png (random)
					#SAU VEZI DACA SE SUPRASCRIE POZA si ii spun captured_image.png
					print("About to take a picture.");
					with picamera.PiCamera() as camera:
						camera.capture("poza.png")
					print("Picture taken.")
					avatar_file = open("poza.png", "rb")
					avatar_file_bytes = avatar_file.read(BUFFER_SIZE)
					while (avatar_file_bytes):
						print 'Sending %s bytes...', BUFFER_SIZE
						send_message = '%s%s' % (avatar_file_bytes, '\n')
						self.socket.send(avatar_file_bytes)
						avatar_file_bytes = avatar_file.read(BUFFER_SIZE)
						data = self.socket.recv(512)
						print 'Received data: ', data

					print 'file opened!'
					avatar_file.close()
					responseMsj='Picture successully sent!\n'
					response = '%s:%s' % (MessageType.RESPONSE_SUCCESS, responseMsj)
					self.socket.send(response)




		print 'Connection has been closed.'


print "TCPServer Waiting for client on port ", PORT


"""
Accept for connecions and for each
connection create a Client socket
"""
while True:
	(client_socket, (ip, port)) = server_socket.accept()
	print "I got a connection from %s:%s" % (str(ip), str(port))

	newThread = ClientThread(client_socket, ip, port)
	newThread.start()
	clientThreads.append(newThread)