vending/states.py at master · castortut/vending · 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
from tmp_products import products

import logging
import requests
from Bank import Bank


bank = Bank()


# States and their timeouts
IDLE = 0
PRODUCT_CHOSEN = 1
PRODUCT_CHOSEN_TIMEOUT = 200
DISPLAY_TRANS = 2
DISPLAY_TRANS_TIMEOUT = 200
DISPLAY_ACC = 3
DISPLAY_ACC_TIMEOUT = 200


def goto_display_transaction(packet, self):

    result = bank.make_transaction(packet, -products[self.selection][1], self.BANK_MSG_PREFIX + products[self.selection][0])
    if not result.error:
        self.lcd.clear_rows((0, 1, 2, 3))
        if result.success:
            self.lcd.print(0, "Siirto OK")
            self.lcd.print(2, "Tili: " + result.account)
            self.lcd.print(3, "Jaljella: %.2f e" % (result.balance/100))
            logging.info("Transaction complete")
        else:
            self.lcd.clear_rows((1, 2, 3))
            self.lcd.print(1, result.message1)
            if result.message2 is not None:
                self.lcd.print(2, result.message2)
    else:
        self.lcd.clear_rows((0, 1, 2, 3))
        self.lcd.print(1, "Error")

    self.state   = DISPLAY_TRANS
    self.timeout = DISPLAY_TRANS_TIMEOUT


def state_display_trans(self, btn, packet):
    if self.timeout == 0:
        goto_idle(self)


def goto_display_account(packet, self):
    result = bank.get_balance(packet)
    if not result.error:
        if result.success:
            self.lcd.clear_rows((1, 2, 3))
            self.lcd.print(1, "Tili: " + str(result.account))
            self.lcd.print(2, "Saldo: " + "%.2f" % (result.balance/100) + " e")
        else:
            self.lcd.clear_rows((1, 2, 3))
            self.lcd.print(1, result.message1)
            if result.message2 is not None:
                self.lcd.print(2, result.message2)
    else:
        self.lcd.clear_rows((0, 1, 2, 3))
        self.lcd.print(1, "Error")
    self.state = DISPLAY_ACC
    self.timeout = DISPLAY_ACC_TIMEOUT
    logging.info("Displaying account")


def state_display_acc(self, btn, packet):
    if self.timeout == 0:
        goto_idle(self)


def goto_product_chosen(btn, self):
    self.lcd.clear_rows((0, 1, 2, 3))
    self.selection = btn - 1
    try:
        logging.info("Product chosen")
        self.lcd.print(0, "Tuote " + products[self.selection][0])
        self.lcd.print(1, "Hinta: %.2f" % (products[btn - 1][1]/100))
        self.lcd.print(3, "Nayta kortti")
    except IndexError:
        logging.info("Invalid product chosen")
        self.lcd.clear_row(2)
        self.lcd.print(1, "Ei tuotetta")
    self.state = PRODUCT_CHOSEN
    self.timeout = PRODUCT_CHOSEN_TIMEOUT


def state_product_chosen(self, btn, packet):
    if btn and btn != (self.selection + 1):
        goto_product_chosen(btn, self)
    elif packet:
        goto_display_transaction(packet, self)
    elif self.timeout == 0:
        goto_idle(self)


def goto_idle(self):
    self.lcd.clear_rows((0, 1, 2, 3))
    self.lcd.print(1, "Valitse tuote")
    self.state = IDLE


def state_idle(self, btn, packet):
    if btn:
        goto_product_chosen(btn, self)
    elif packet:
        goto_display_account(packet, self)
    else:
        pass