CheckMeIn/guests.py at master · alan412/CheckMeIn · 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
import sqlite3
import os
from collections import namedtuple
import datetime
from enum import IntEnum

Guest = namedtuple('Guest', ['guest_id', 'displayName'])


class Status(IntEnum):
    inactive = 0
    active = 1


class Guests(object):
    def __init__(self):
        self.date = 0
        self.num = 1

    def migrate(self, dbConnection, db_schema_version):
        if db_schema_version <= 2:
            dbConnection.execute('''CREATE TABLE guests
                                (guest_id TEXT UNIQUE,
                                 displayName TEXT,
                                 email TEXT,
                                 firstName TEXT,
                                 lastName TEXT,
                                 whereFound TEXT,
                                 status INTEGER default 1
                                 )''')
        if db_schema_version <= 5:
            dbConnection.execute(
                "ALTER TABLE guests ADD COLUMN newsletter INTEGER default 0")

    def injectData(self, dbConnection, data):
        for datum in data:
            dbConnection.execute(
                "INSERT INTO guests VALUES (?,?,?,?,?,?,?,?)",
                (datum["guest_id"], datum["displayName"], datum["email"],
                 datum["firstName"], datum["lastName"], datum["whereFound"],
                 datum["status"], datum["newsletter"]))

    def add(self, dbConnection, displayName, first, last, email, whereFound,
            newsletter):
        if self.date != datetime.date.today():
            self.date = datetime.date.today()
            self.num = 1
        else:
            self.num = self.num + 1

        while self.num < 10000:
            try:
                guest_id = self.date.strftime("%Y%m%d") + '{0:04d}'.format(
                    self.num)
                # zero padded up to 9999 for each day
                dbConnection.execute(
                    "INSERT INTO guests VALUES (?,?,?,?,?,?,?,?)",
                    (guest_id, displayName, email, first, last, whereFound,
                     Status.active, newsletter))
            except sqlite3.DatabaseError:
                self.num = self.num + 1
            else:
                return guest_id

    def getName(self, dbConnection, guest_id):
        data = dbConnection.execute(
            "SELECT displayName FROM guests WHERE guest_id==?",
            (guest_id, )).fetchone()
        if data is None:
            return ('Invalid: ' + guest_id, None)
        else:
            # Add code here for inactive
            return ('', data[0])

    def getEmail(self, dbConnection, guest_id):
        data = dbConnection.execute(
            "SELECT email FROM guests WHERE guest_id==?",
            (guest_id, )).fetchone()
        if data is None:
            return ('Invalid: ' + guest_id, None)
        else:
            # Add code here for inactive
            return ('', data[0])

    def getList(self, dbConnection):
        guestList = []

        for row in dbConnection.execute(
            "SELECT * FROM guests WHERE status is NOT ? ORDER BY displayName",
                (Status.inactive, )):
            guestList.append(Guest(row[0], row[1]))
        return guestList

    def getGuests(self, dbConnection, numDays):
        guestList = []
        for row in dbConnection.execute(
            '''SELECT DISTINCT guest_id, displayName FROM guests
                 INNER JOIN visits ON guest_id = visits.barcode
                 WHERE start > ?
                 ORDER BY displayName''',
                (datetime.datetime.now() - datetime.timedelta(numDays), )):
            guestList.append(Guest(row[0], row[1]))
        return guestList