· 7 years ago · Nov 28, 2018, 11:18 PM
1import sqlite3
2
3def load_database():
4 with sqlite3.connect("NEA.db") as db:
5 cursor = db.cursor()
6
7 sql_users = """
8 CREATE TABLE IF NOT EXISTS Users(
9 UserID INTEGER PRIMARY KEY AUTOINCREMENT,
10 Email TEXT,
11 HashedPassword TEXT,
12 AccountFunds NUMERIC
13 );
14 """
15 cursor.execute(sql_users)
16
17 sql_location = """
18 CREATE TABLE IF NOT EXISTS Location(
19 LocationID INTEGER PRIMARY KEY AUTOINCREMENT,
20 LongLatCoordinates TEXT,
21 LocationDescription TEXT
22 );
23 """
24 cursor.execute(sql_location)
25
26 sql_advertisements = """
27 CREATE TABLE IF NOT EXISTS Advertisements(
28 AdvertID INTEGER PRIMARY KEY AUTOINCREMENT,
29 UserID INTEGER,
30 LocationID INTEGER,
31 ImageLink TEXT,
32 ViewsRemaining INTEGER,
33 TotalViews INTEGER,
34 DayDisplayed TEXT,
35 OverlayPosition TEXT,
36 OverlayColour INTEGER,
37 OverlayContWalkingDist INTEGER,
38 OverlayContOpenTime TEXT,
39 FOREIGN KEY(UserID) REFERENCES Users(UserID),
40 FOREIGN KEY(LocationID) REFERENCES Location(LocationID)
41 );
42 """
43 cursor.execute(sql_advertisements)