#!/usr/bin/env python

# -*- coding: utf-8 -*-

import socket

import StringIO

import contextlib

import traceback

import sys

import re

from sys import argv

script, PORT, token_file, sep = argv

##### set up a server

HOST = 'localhost'

PORT = int(PORT)

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # Socket created

s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

# the SO_REUSEADDR flag tells the kernel to reuse a local socket in TIME_WAIT state,

# without waiting for its natural timeout to expire.

# see: https://docs.python.org/2/library/socket.html

try:

s.bind((HOST, PORT))

except socket.error , msg:

print 'Bind failed. Error Code : ' + str(msg[0]) + '; Message: ' + msg[1]

quit()

s.listen(10) # Socket now listening

@contextlib.contextmanager

def stdoutIO(stdout=None):

'''

store outputs from stdout

'''

old = sys.stdout

if stdout is None:

stdout = StringIO.StringIO()

sys.stdout = stdout

yield stdout

sys.stdout = old

####### now keep talking with the client

while 1:

### wait to accept a connection

conn, addr = s.accept() # Connected with + addr[0] + str(addr[1])

input_data = conn.recv(1024000)

if re.sub('\s', '', input_data) == 'quit()':

break

### write the codes into a file

with open(token_file, 'w') as f:

f.write(input_data)

### print codes; execute codes

with stdoutIO() as output:

print input_data

print sep

try:

execfile(token_file)

except:

traceback.print_exc()

### send output

output_data = output.getvalue()

conn.send(output_data)

conn.close()

###### clsoe & quit

conn.close()

s.shutdown(2)

s.close()