mirror of
https://github.com/cwinfo/yggdrasil-map
synced 2024-12-29 02:55:40 +00:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from flask import Flask, render_template, request
|
|
from graphData import insert_graph_data
|
|
|
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--host", type=str, help="host to listen on (default 'localhost')", default="localhost")
|
|
parser.add_argument("--port", type=int, help="port to listen on (default '3000')", default=3000)
|
|
|
|
app = Flask(__name__)
|
|
app.config.from_pyfile('web_config.cfg')
|
|
|
|
def get_ip():
|
|
try:
|
|
ip = request.headers['x-real-ip']
|
|
except KeyError:
|
|
ip = None
|
|
if ip == '10.18.3.20':
|
|
ip = request.headers['x-atomshare-real-ip']
|
|
return ip
|
|
|
|
@app.context_processor
|
|
def add_ip():
|
|
return dict(ip=get_ip())
|
|
|
|
|
|
@app.route('/')
|
|
@app.route('/network')
|
|
def page_network():
|
|
return render_template('network.html', page='network')
|
|
|
|
@app.route('/about')
|
|
def page_about():
|
|
return render_template('about.html', page='about')
|
|
|
|
@app.route('/sendGraph', methods=['POST'])
|
|
def page_sendGraph():
|
|
print "Receiving graph from %s" % (request.remote_addr)
|
|
|
|
data = request.form['data']
|
|
mail = request.form.get('mail', 'none')
|
|
version = int(request.form.get('version', '1'))
|
|
ret = insert_graph_data(ip=get_ip(), config=app.config, data=data, mail=mail, version=version)
|
|
|
|
if ret == None:
|
|
return 'OK'
|
|
else:
|
|
return 'Error: %s' % ret
|
|
|
|
@app.after_request
|
|
def add_header(response):
|
|
response.cache_control.max_age = 300
|
|
return response
|
|
|
|
if __name__ == '__main__':
|
|
args = parser.parse_args()
|
|
app.run(host=args.host, port=args.port)
|