app.py

#### Richard Keit <rajkeit@gmail.com> ######

import os
import socket
import string
import psycopg2
import psycopg2.extras
from flask import Flask, render_template, request, redirect, url_for, abort, session
import HTML
import datetime
from time import gmtime, strftime
import time
import sys
app = Flask(__name__)

containeridx = os.environ["HOSTNAME"]

#databasehost = os.environ["PGDB_PORT_5432_TCP_ADDR"]
databasehost = socket.gethostbyname("pgdb")



########## Database Connection String ###########
conn_string = "dbname='ipss' user='postgres' host=" + (databasehost) + " password='postgres'"
	# get a connection, if a connect cannot be made an exception will be raised here
conn = psycopg2.connect(conn_string)
 
	# conn.cursor will return a cursor object, you can use this cursor to perform queries
try:
    conn = psycopg2.connect(conn_string)
except:
    print "I am unable to connect to the database"

##################################################


########## Insert into Database ##################
xcursor = conn.cursor()
xcursor.execute("INSERT INTO temp.containers (containerid, time) VALUES (%s, %s)", (containeridx, strftime("%Y-%m-%d %H:%M:%S", gmtime())))

conn.commit()
xcursor.close()
conn.close()


###################################################
########## Shared File Creation ###################

filename = "/sharedvol/sharedfile"

f = open(filename, "ab")
f.write(" " + containeridx + strftime("%Y-%m-%d %H:%M:%S") + "\n")
f.close()

####################################################

@app.route('/')
def index():
 return 'Please direct to either /db or /sharedfile'


@app.route('/db')
def db():
 connS = psycopg2.connect(conn_string)

 cursor = connS.cursor('cursor_unique_name', cursor_factory=psycopg2.extras.DictCursor)

 cursor.execute('SELECT * FROM temp.containers')

 row_count = 0
 t = HTML.Table(header_row=['ContainerID', 'timestamp'])

 for row in cursor:
   row_count += 1
   t.rows.append([row_count, row])
 htmlcode = str(t)
 print htmlcode
 cursor.close()
 connS.close()
 return htmlcode




@app.route('/sharedfile')
def file():
 row_count = 0
 t = HTML.Table(header_row=['Entry #', 'Entry'])
 testsite_array = []
 with open(filename) as my_file:
  for line in my_file:
   row_count += 1
   t.rows.append([row_count, line])
   
 htmlcode = str(t)
 return htmlcode

###################################################3

if __name__ == '__main__':
    app.run(debug=False, host='0.0.0.0')