#!/usr/bin/env python
# -*- coding: utf-8 -*-
host = '0.0.0.0'
port = 9001
db_host = 'localhost'
db_name = 'stalker'
db_user = 'stalker'
db_pass = '*******'
# ---
import MySQLdb
from flask import Flask, request, abort
from datetime import datetime, timedelta
app = Flask(__name__)
db = MySQLdb.connect (host = db_host, user = db_user, passwd = db_pass, db = db_name)
@app.route('/')
def page(path = ''):
serial = request.args.get('serial', None)
if not serial:
abort(403)
ip = request.args.get('ip', None)
if not ip:
abort(403)
cursor = db.cursor()
timeout = datetime.now() - timedelta(minutes=10)
query = 'SELECT id FROM users WHERE serial_number=%s AND ip=%s AND keep_alive>%s'
cursor.execute(query, (serial, ip, timeout.strftime("%Y-%m-%d %H:%M:%S")))
row = cursor.fetchone()
cursor.close()
if row == None:
abort(403)
return 'OK', 200
if __name__ == '__main__':
app.run(host=host, port=port)
Comments