Update database.py

database.py

@@ -32,13 +32,15 @@ def init_db(app):
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 udid TEXT NOT NULL UNIQUE,
                 paid BOOLEAN DEFAULT FALSE,
-                usage_limit INTEGER DEFAULT 4
+                usage_limit INTEGER DEFAULT 4,
+                invoice TEXT DEFAULT NULL
             )
         ''')
         
         cursor.execute('SELECT COUNT(*) FROM users')
         if cursor.fetchone()[0] == 0:
-            cursor.executemany('INSERT INTO users (udid, paid, usage_limit) VALUES (?, ?, ?)',
-                             [('test_udid_1', True, 4), ('test_udid_2', False, 4)])
+            cursor.executemany('INSERT INTO users (udid, paid, usage_limit, invoice) VALUES (?, ?, ?, ?)',
+                             [('test_udid_1', True, 4, 'test_invoice_1'), 
+                              ('test_udid_2', False, 4, None)])
         
         db.commit()