import tempfile
 
# The path for storing any files with randomly generated (temp) names.
path = 'data'
tempfile.tempdir = path
 
# Generates a random filename not already in use.
filename = tempfile.mktemp()
 
# Writes the given text to the file.
def write(text=""):
    fileHandle = open(filename, 'a')
    fileHandle.write(text)
    fileHandle.close()

import pickle
 
# Pickles a python object (e.g. a dictionary).
def p(obj):
    fileHandle = open(filename, 'a')
    pickle.dump(obj, fileHandle)
    fileHandle.close()
 
sourceFolder = 'data'
 
# Gets data out of a pickled file.
def unpickle(filename):
    file = sourceFolder + '/' + filename
 
    fileHandle = open(file, 'r')
    data = pickle.load(fileHandle)
    fileHandle.close()
 
    return data

import sqlite3
class DataAccess():
    my_database="budget.db"
    def __init__(self):
        self.connection=sqlite3.connect(self.my_database)
    def create_database(self):
        self.connection.execute("create table account(id integer primary key, balance number);")
        self.connection.execute("create table transact(id integer primary key, narrativ text, amount number);")
        self.connection.execute("insert into account(balance) values(0);")
        self.connection.commit()
    def input_transaction(self, amount, narrativ):
        sqlstr="insert into transact (narrativ, amount) values(\"%s\",%s); " % (narrativ,amount)
        self.connection.execute(sqlstr)
        self.connection.execute("update account set balance=balance+%s; " % (amount))
        self.connection.commit()
    def get_balance(self):
        cu=self.connection.cursor()
        cu.execute("select * from account;")
        return cu.fetchall()
 
class BussinesLogic():
    def __init__(self):
        self.database=DataAccess()
    def report_balance(self):
        return self.database.get_balance()[0][1]
    def input_expense(self,expense, narrativ):
        self.database.input_transaction(-expense,narrativ)
    def create_database(self):
        self.database.create_database()
 
 
import sys
class UserInterface():
    def __init__(self, bl):
        self.bl=bl
        self.menu=dict()
        self.set_menu()
    def set_menu(self):
        self.menu[4]=("Report Balance",self.display_balance)
        self.menu[1]=("Input expense",self.get_expense)
        self.menu[2]=("Exit program",self.exit)
        self.menu[3]=("Create_database (budget.db)",self.bl.create_database)
    def display_balance(self):
        print ("your balance is: %s"%self.bl.report_balance())
    def get_expense(self):
        print("please input the data")
        amount=int(raw_input("amount:"))
        narrativ=raw_input("narrativ:")
        self.bl.input_expense(amount,narrativ)
    def exit(self):
        print ("Goodbye")
        sys.exit(0)
    def display_menu(self):
        print ("Personal Budgeting")
        print ("Please choose a function!")
        for k,v in self.menu.iteritems():
            print k,v[0]
    def main(self):
        choice=-1
        while True:
            self.display_menu()
            choice=int(raw_input("?:"))
            self.menu[choice][1]()
 
if __name__=="__main__":
    BL=BussinesLogic()
    UI=UserInterface(BL)
    UI.main()

 
def delEntry(index):
    fname="budget.csv"
    entries=open(fname).readlines()
    del entries[index]
    fo=open("budget.csv","w")
    fo.writelines(entries)
    fo.close()
 
def delExpense():
    delE = int(raw_input('What expense would you like to delete? '))
    delEntry(delE)
delExpense()

def get_transactions(self):
    ct=self.connection.cursor()
    ct.execute("select * from transact;")
    return ct.fetchall()
 
def report_transactions(self):
    return self.database.get_transactions()
 
def display_transactions(self):
    print "Your tranaction History is:",self.bl.report_transactions()
 
self.menu[5]=("Transaction History",self.display_transactions)