meshenger/meshenger_clientserve.py

#!/usr/bin/python

import logging
import cgi
import os
import socket
from BaseHTTPServer import HTTPServer
import SimpleHTTPServer
import urlparse
import unicodedata
import logging, logging.config

logging.config.fileConfig('pylog.conf')
logger = logging.getLogger('meshenger'+'.clientserve')

# this variable is set from main, to be called when new messages available
build_index_callback = None


class ClientServeHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):

  messageDir = "msg"

  """
  Serve index and messages
  """
  def do_GET(self):

    if self.path == '/':
      self.send_response(200)
      self.send_header('Content-type', 'text/html')
      self.end_headers()
      f = os.path.relpath('webapp.html')
      # f = os.path.join('/root/meshenger/',"webapp.html")
      with open( f, 'r') as the_file:
        self.wfile.write(the_file.read())

    elif self.path == '/index' or self.path.startswith( "/"+self.messageDir ):
      return SimpleHTTPServer.SimpleHTTPRequestHandler.do_GET(self)
    elif self.path == '/log':
      self.send_response(200)
      self.send_header('Content-type', 'text-html')
      self.end_headers()
      f = os.path.relpath('log/meshenger.log')
      with open( f, 'r') as the_file:
        self.wfile.write(the_file.read())
    else:
      self.send_response(200) #serve the webapp on every url requested
      self.send_header('Content-type', 'text/html')
      self.end_headers()
      f = os.path.join( "webapp.html")
      with open( f, 'r') as the_file:
        self.wfile.write(the_file.read())


  """
  Allow clients to post messages
  """
  def do_POST(self):
    if self.path == '/send':

      length = int(self.headers['Content-Length'])
      post_data = urlparse.parse_qs(self.rfile.read(length).decode('utf-8'))

      self.writeMessage( post_data["time"][0].encode('ascii','ignore'), post_data["message"][0])

      self.send_response(200)
      self.send_header('Content-type', 'text/html')
      self.end_headers()
      self.wfile.write('message created')

      #let main rebuild message index
      if build_index_callback:
        build_index_callback()

  def writeMessage(self, time, message):

    f = os.path.join( self.messageDir, time)
    if os.path.isfile( f ):
      return
    with open( f, 'a') as the_file:
      the_file.write(message)


class ClientServe():
  def __init__(self, port):
    server = HTTPServer( ('', port), ClientServeHandler)
    server.serve_forever()


def main():
  clientServe = ClientServe(80)

if __name__ == '__main__':
  main()