5dbddc2c7e
Replace the timeframe filter to date filter Change-Id: Iaae7efe8e850db2c631e254f067028b9ef2ec701
73 lines
1.9 KiB
Python
Executable File
73 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import datetime
|
|
|
|
from bson import json_util
|
|
from flask import Flask, request, render_template, send_from_directory
|
|
import pymongo
|
|
|
|
import config
|
|
import db_helper
|
|
import logger
|
|
|
|
|
|
cfg = config.Config()
|
|
|
|
app = Flask(__name__)
|
|
app.debug = True
|
|
|
|
logger.init(cfg)
|
|
|
|
db = db_helper.DBHelper(cfg).get()
|
|
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return render_template('index.html', host=request.host)
|
|
|
|
|
|
@app.route('/static/<path:path>')
|
|
def send_js(path):
|
|
# TODO: We should probably use a real webserver for this..
|
|
return send_from_directory('static', path)
|
|
|
|
|
|
@app.route('/ci-accounts', methods=['GET'])
|
|
def ci_accounts():
|
|
return json_util.dumps(db.ci_accounts.find())
|
|
|
|
|
|
@app.route('/results', methods=['GET'])
|
|
def results():
|
|
# TODO: We should have a cache for these requests
|
|
# so we don't get hammered by reloading pages
|
|
project = request.args.get('project', None)
|
|
count = request.args.get('count', None)
|
|
skip = request.args.get('skip', None)
|
|
timeframe = request.args.get('timeframe', None)
|
|
start = request.args.get('start', None)
|
|
end = request.args.get('end', None)
|
|
|
|
return query_results(project, count, skip, timeframe, start, end)
|
|
|
|
def query_results(project, count, skip, timeframe, start, end):
|
|
query = {}
|
|
date_format = '%Y-%m-%d'
|
|
if project:
|
|
query['project'] = project
|
|
if timeframe:
|
|
num_hours = int(timeframe)
|
|
current_time = datetime.datetime.utcnow()
|
|
start_time = current_time - datetime.timedelta(hours=num_hours)
|
|
query['created'] = {'$gt': start_time}
|
|
elif start and end:
|
|
start = datetime.datetime.strptime(start, date_format)
|
|
end = datetime.datetime.strptime(end, date_format)
|
|
query['created'] = {'$gte': start, '$lt': end}
|
|
records = db.test_results.find(query).sort('created', pymongo.DESCENDING)
|
|
return json_util.dumps(records)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0')
|