summaryrefslogtreecommitdiff
path: root/python/gras/stats/__init__.py
blob: 2ed52eae77f30d18c7363b2ee02af13b24bb90b7 (plain)
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
import time
import BaseHTTPServer

import os
__path__ = os.path.abspath(os.path.dirname(__file__))

get_stats_registry = [lambda: ""]

class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):

    def do_HEAD(s):
        s.send_response(200)
        s.send_header("Content-type", "text/html")
        s.end_headers()

    def do_GET(s):
        """Respond to a GET request."""
        if s.path.endswith('stats.xml'):
            s.send_response(200)
            s.send_header("Content-type", "text/xml")
            s.end_headers()
            s.wfile.write(get_stats_registry[0]())
            return
        s.send_response(200)
        s.send_header("Content-type", "text/html")
        s.end_headers()
        path = s.path
        if path.startswith('/'): path = path[1:]
        if not path: path = 'main.html'
        s.wfile.write(open(os.path.join(__path__, path)).read())

def http_server(args, get_stats_xml):
    get_stats_registry[0] = get_stats_xml
    server_class = BaseHTTPServer.HTTPServer
    httpd = server_class(args, MyHandler)
    return httpd