__main__.py

#!/usr/bin/env python3

import connexion

from access_node import encoder
from flask_cors import CORS

from access_node.models.nodes import nodes
import json

import time
import requests
import psycopg2

def ConnectToDatabase(postgres_username, postgres_password):
    database_host = 'database'
    with open('database_host.txt') as database_host_file:
        database_host = database_host_file.readline()
    return psycopg2.connect(database="postgres", user="postgres",
                       password="postgres", host=database_host, port="5432")


def main():
    # Wait for simulation nodes to post to database
    time.sleep(5)

    # get simulation nodes
    con = ConnectToDatabase('postgres', 'postgres')
    cur = con.cursor()
    # NEST
    cur.execute("SELECT address FROM nest_simulation_node")
    nodes.nest_simulation_nodes = [i[0] for i in cur.fetchall()]
    # Arbor
    cur.execute("SELECT address FROM nest_simulation_node")
    nodes.arbor_simulation_nodes = [i[0] for i in cur.fetchall()]
    con.close()


    # run acces_node
    app = connexion.App(__name__, specification_dir='./swagger/')
    app.app.json_encoder = encoder.JSONEncoder
    app.add_api('swagger.yaml', arguments={
                'title': 'In-Situ Pipeline REST API'})
    CORS(app.app)
    app.run(port=8080)


if __name__ == '__main__':
    main()