PythonServer/server.py at master · noobskylimit/PythonServer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
import datetime
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
import SocketServer
import json
import cgi


class Server(BaseHTTPRequestHandler):
    def _set_headers(self):
        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()

    def do_HEAD(self):
        self._set_headers()

    # GET sends back a Hello world message
    def do_GET(self):
        self._set_headers()
        self.wfile.write(json.dumps({'hello': 'world','hello': 'earth', 'received': 'ok'}))

    # POST echoes the message adding a JSON field
    def do_POST(self):
        ctype, pdict = cgi.parse_header(self.headers.getheader('content-type'))

        # refuse to receive non-json content
        if ctype != 'application/json':
            self.send_response(400)
            self.end_headers()
            return

        # read the message and convert it into a python dictionary
        length = int(self.headers.getheader('content-length'))
        message = json.loads(self.rfile.read(length))

        if message.has_key('after'):
            self.after =  message.get('after')
        else:
            after = datetime.datetime.utcnow()
            self.after = after.strftime("%Y-%m-%d %H:%M:%S")

        payload = {
            "success": "0",
            "im": "empty!",
        }
        bevor = datetime.datetime.utcnow()
        bevor = bevor.strftime("%Y-%m-%d %H:%M:%S")

        if message.has_key('url'):
            # add a property to the object, just to mess with data
            payload = {
                "success": "1",
                "data": {
                    "task_uuid": "efb625b9942e000004751393ce42030d"
                }}

        elif message.has_key("username"):
            payload = {"success": "1",}


        elif message.has_key("uuid"):
            with open(r'report.json', 'r') as json_file:
                payload = json.load(json_file)

        elif message.has_key("moreData"):
            payload = {
                "success": 1,
                "data": {
                    "tasks": [
                        "7777777777777777777777777777777",
                        "8888888888888888888888888888888",
                        "9999999999999999999999999999999"
                    ],
                    "after": self.after,
                    "more_results_available": 1,
                    "before": bevor
                }
            }
        elif message.has_key("noMore"):
              payload = {
                "success": 1,
                "data": {
                    "tasks": [
                        "111111111111111111111111111111111",
                        "efb625b9942e000004751393ce42030d",
                        "222222222222222222222222222222222"
                    ],
                    "after": self.after,
                    "more_results_available": 0,
                    "before": "2016 - 03 - 11 20:45:22"
                }
            }

        else:
            print("in Server Else!")
            payload = {
                "success": 1,
                "data": {
                    "task": ["efb625b9942e000004751393ce42030d"],
                    "after": "2016 - 03 - 11 20:00:00",
                    "more_results_available": 0,
                    "before": "2016 - 03 - 11 20:45:22"
                }}


        # send the message back
        self._set_headers()
        self.wfile.write(json.dumps(payload))


def run(server_class=HTTPServer, handler_class=Server, port=8008):
    server_address = ('', port)
    httpd = server_class(server_address, handler_class)

    print 'Starting httpd on port %d...' % port
    httpd.serve_forever()


if __name__ == "__main__":
    from sys import argv

    if len(argv) == 2:
        run(port=int(argv[1]))
    else:
        run()