Py_Expense_template/expense.py at master · TktMaxime/Py_Expense_template · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from PyInquirer import prompt
import csv
from user import read_users, add_user


expense_questions = [
    {
        "type": "input",
        "name": "amount",
        "message": "New Expense - Amount: ",
    },
    {
        "type": "input",
        "name": "label",
        "message": "New Expense - Label: ",
    },
    {
        "type": "input",
        "name": "spender",
        "message": "New Expense - Spender: ",
    },

]


def add_stop_option():
    L = read_users()
    L.append("Stop")
    return L


users = [
    {
        'type': 'rawlist',
        'name': 'users',
        'message': 'Select users or select Stop when you are done : ',
        'choices': add_stop_option(),
    },
]


def export_expense(infos):
    with open('expense_report.csv', 'a') as f:  # You will need 'wb' mode in Python 2.x
        w = csv.DictWriter(f, infos.keys())
        w.writeheader()
        w.writerow(infos)


def new_expense(*args):
    infos = prompt(expense_questions)

    # Writing the informations on external file might be a good idea ¯\_(ツ)_/¯
    print("Expense Added !")
    tmp = ""
    List_of_users_involved = []
    while tmp != "Stop":
        user = prompt(users)
        tmp = user["users"]
        if tmp == "Stop":
            break
        List_of_users_involved.append(tmp)
    add_user(infos["spender"])
    List_of_users_involved.append(infos["spender"])

    cost_of_each = int(infos["amount"]) // (len(List_of_users_involved))
    print("Each people need to refund : " +
          str(cost_of_each) + " to " + infos["spender"])

    infos["people_involved"] = List_of_users_involved
    export_expense(infos)
    return True