__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'
    try:
        with open('database_host.txt') as database_host_file:
            database_host = database_host_file.readline().rstrip('\n')
    except:
        pass
    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(0.1)

    # # 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()