email-assistant/viewers/main/viewer/main.py

import os
import sys

import flask
import requests


app = flask.Flask(__name__)
app.config['DEBUG'] = True

indexer_url = os.environ.get('INDEXER_URL', 'http://indexer')


@app.route('/healthcheck')
def healthcheck():
    return 'OK'

@app.route('/')
def home():
    return flask.render_template('home.html')


@app.route('/shipping')
def get_tokens():
    resp = requests.get(
        indexer_url+'/token',
        params={
            'filter_type': 'SHIPPING',
            'desc': True,
        },
    )
    resp.raise_for_status()
    tokens = resp.json().get('tokens')
    for token in tokens:
        try:
            resp = requests.get(
                'http://viewer_package_tracking:3000/info/'+token['token']
            )
            resp.raise_for_status()
            print('Response: ', resp.text, file=sys.stderr)
            info = resp.json()
            token['metadata'].update(info)
        except Exception as e:
            print('Error', e, file=sys.stderr)
            pass
    return flask.render_template('shipping.html', trackers=tokens)


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)