main.py

#!/usr/bin/env python3
import os

from bson import json_util
from flask import (Flask, Response, json, jsonify, render_template,
                   send_from_directory)
from pymongo import MongoClient

mongo_uri = "mongodb://localhost:27017/orbits"
try:
    # This is what it's called by default on Heroku
    mongo_uri = os.environ["MONGODB_URI"]
except KeyError as e:
    pass
mongo_client = MongoClient(mongo_uri)
db = mongo_client.get_database()
czml_collection = db['czml']

app = Flask("tleapp")


@app.route("/")
def index():
    return render_template("index.html")


@app.route("/cesium/<path:path>")
def send_cesium_files(path):
    return send_from_directory("node_modules/cesium/Build/Cesium", path)


@app.route("/orbit/<id>")
def get_orbit(id):
    return Response(json_util.dumps([
        czml_collection.find_one({'id': 'document'}),
        czml_collection.find_one({'$text': {'$search': id}})
    ]), status=200, content_type="application/json")


@app.route("/orbits")
def get_orbits():
    return Response(json_util.dumps(czml_collection.aggregate([
        {'$match': {
            'id': {'$regex': '^((?!deb)(?!r/b).)*$', '$options': 'i'}}},
        {'$limit': 100}
    ])), status=200, content_type="application/json")


if __name__ == "__main__":
    app.run()