gistfile1.txt

import tornado
from tornado import autoreload, ioloop, web, options, escape, websocket
import re
import json

profane_dict = dict()
re_dict = dict()

def re_compile(word_list):
	exp = r'\b%s\b' %'|'.join(word_list)
	r = re.compile(exp, re.IGNORECASE)
	return r

def codify_doc(doc, re_obj):
	match = re_obj.match(doc)
	if match:
		return True
	else:
		return False

class WSHandler(websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hello World")
      
    def on_message(self, message):
        print 'message received %s' % message
 
    def on_close(self):
      print 'connection closed'


class MainHandler(tornado.web.RequestHandler):
    def get(self):
        self.write("Hello, world")

class CodifyProfane(tornado.web.RequestHandler):
	def get(self):
		client_id = self.get_argument('client_id', '')
		doc = client_id = self.get_argument('doc', '')

		re_obj = re_dict[client_id]


		if client_id in profane_dict:
			response = { 'client_id': client_id,
						'profane_code': codify_doc(doc, re_obj)			
						}
		else:
			response = "client not found"
		self.write(response)

class ProfaneList(tornado.web.RequestHandler):
	def get(self):
		client_id = self.get_argument('client_id', '')
		if client_id in profane_dict:
			response = { 'client_id': client_id,
						'swear_dict': profane_dict[client_id]
						}
		else:
			response = { 'client_id': client_id,
						'swear_dict': ''
						}
		self.write(response)


class ProfaneListInit(tornado.web.RequestHandler):
	global profane_dict
	global re_dict
	def post(self):
		client_id = self.get_argument('client_id', '')
		profane_list = json.loads(self.get_argument('profane_list', ''))

		if client_id in profane_dict:
			self.write("Client ID exists, use update")
		else:
			profane_dict[client_id] ={
				'profane_list': profane_list
			}

			word_list = []
			for key,value in profane_list.iteritems():
				word_list.append(key)

			re_obj = re_compile(word_list)
			re_dict[client_id] = re_obj


class ProfaneListUpdate(tornado.web.RequestHandler):
	global profane_dict
	global re_dict
	def post(self):
		client_id = self.get_argument('client_id', '')
		u_type = self.get_argument('u_type', '')
		term = self.get_argument('term', '')

		self.write(u_type)

		if client_id in profane_dict:
			if u_type == 'add':
				if term not in profane_dict[client_id]['profane_list']:
					profane_dict[client_id]['profane_list'][term] = []
			if u_type == 'remove':
				if term in profane_dict[client_id]['profane_list']:
					profane_dict[client_id]['profane_list'].pop(term, None)

			word_list = []
			for key,value in profane_dict[client_id]['profane_list'].iteritems():
				word_list.append(key)

			re_obj = re_compile(word_list)
			re_dict[client_id] = re_obj

		else:
			self.write("Client does not exist")
 	get = post



application = web.Application([
    (r"/", MainHandler),
    (r"/profanity/", ProfaneList),
    (r"/profanity/update/", ProfaneListUpdate),
    (r"/profanity/initialize/", ProfaneListInit),
    (r"/profanity/codify/", CodifyProfane),
    (r'/ws', WSHandler),
])


if __name__ == "__main__":
    tornado.options.parse_command_line()
    app = application
    app.listen(8888)
    ioloop = tornado.ioloop.IOLoop().instance()
    autoreload.start(ioloop)
    ioloop.start()