KKnGDvwN

1import sqlite3 as db
2from datetime import datetime
3def init():
4    conn = db.connect("expense.db")
5    cur = conn.cursor()
6    sql = '''
7    create table if not exists expense (
8        amount number,
9        category string,
10        message string,
11        date string
12        )
13    '''
14    cur.execute(sql)
15    conn.commit()
16
17def log(amount, category, message = ""):
18    date = str(datetime.now())
19    conn = db.connect("expense.db")
20    cur = conn.cursor()
21    sql = '''
22    insert into expense values (
23         {},
24        '{}',
25        '{}',
26        '{}'
27    )
28    '''.format(amount, category, message, date)
29    cur.execute(sql)
30    conn.commit()
31
32def view(category=None):
33    conn = db.connect("expense.db")
34    cur = conn.cursor()
35    if category:
36        sql = '''
37        select * from expense where category = '{}'
38        '''.format(category)
39        sql2 = '''
40        select sum(amount) from expense where category = '{}'
41        '''.format(category)
42    else :
43        sql = '''
44        select * from expense
45        '''
46        sql2 = '''
47        select sum(amount) from expense
48        '''.format(category)
49    cur.execute(sql)
50    results = cur.fetchall()
51    cur.execute(sql2)
52    total_amount = cur.fetchone()[0]
53    return results,total_amount