import psycopg2
from flask import Flask, request, render_template, redirect, url_for
import time

app = Flask(__name__)

i=int()
e=int()
b=int()

@app.route("/", methods=['GET', 'POST'])
def home():
    return redirect(url_for('display'))

@app.route('/display', methods=['POST'])
def hello():
    date=request.form['date']
    description = request.form['description']
    transaction = request.form['transaction']
    amount = request.form['amount']
    intoTable(description,transaction,amount,date)
    return redirect(url_for('display'))

def intoTable(de,t,a,d):
    global i,e,b
    conn = psycopg2.connect("dbname=one user=postgres password=12345678")
    cur = conn.cursor()
    if(d==""):
        d= time.strftime("%Y-%m-%d")
    if(t=='credit'):
        i = i + int(a)
        b = b + int(a)
        cur.execute("INSERT INTO bs(date, description, credit, debit, balance, total_income, total_expense) VALUES(%s, %s, %s, %s, %s, %s, %s);", (d, de, a, 0, b, i, e))
    else:
        e = e + int(a)
        b = b - int(a)
        cur.execute("INSERT INTO bs(date, description, credit, debit, balance, total_income, total_expense) VALUES(%s, %s, %s, %s, %s, %s, %s);", (d, de, 0, a, b, i, e))
    conn.commit()
    conn.close()

@app.route('/display', methods=['GET','POST'])
def display():
    conn = psycopg2.connect("dbname=one user=postgres password=12345678")
    cur=conn.cursor()
    cur.execute("SELECT id, date, description, credit, debit, balance, total_income, total_expense FROM bs;")
    rows = cur.fetchall()
    conn.close()
    return render_template('table.html', items=rows)

if __name__=='__main__':
    app.run(port=4998)