-
Notifications
You must be signed in to change notification settings - Fork 0
/
_db_queries.py
82 lines (62 loc) · 2.22 KB
/
_db_queries.py
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
import sqlite3, math
from Table import Table
con = sqlite3.connect('test.db')
c = con.cursor()
def createMaterialsTabel():
c.execute('''CREATE TABLE materials (id integer, name text, damage integer, defense integer, durability integer, weapon integer, armour integer)''')
def addMaterials():
c.execute("INSERT INTO materials VALUES (0, 'Rusty', 1, 2, 2, 1, 1)")
c.execute("INSERT INTO materials VALUES (1, 'Bronze', 1, 2, 2, 1, 1)")
c.execute("INSERT INTO materials VALUES (2, 'Iron', 3, 4, 3, 1, 1)")
c.execute("INSERT INTO materials VALUES (3, 'Silver', 5, 2, 1, 1, 1)")
c.execute("INSERT INTO materials VALUES (4, 'Mythril', 3, 5, 3, 1, 1)")
c.execute("INSERT INTO materials VALUES (5, 'Mysterious alloy', 5, 6, 2, 1, 1)")
c.execute("INSERT INTO materials VALUES (6, 'Dragon scales', 4, 7, 4, 1, 1)")
c.execute("INSERT INTO materials VALUES (7, 'leather', 0, 2, 15, 0, 1)")
def createWeaponsBlueprintTable():
c.execute('''
CREATE TABLE weaponBlueprints
(
id integer,
name text,
damage integer,
durability integer
)
''')
def createWeaponsTable():
c.execute('''
CREATE TABLE weapons
(
weaponBlueprintId integer,
materialId integer,
foreign key (weaponBlueprintId) references weaponBlueprints,
foreign key (materialId) references materials
)
''')
def showTable(tableName):
headers = c.execute("PRAGMA table_info(" + tableName + ")").fetchall()
contents = c.execute('''
SELECT *
FROM ''' + tableName ).fetchall()
sanitizedHeaders = ['' for i in range(len(headers))]
for index, header in enumerate(headers):
sanitizedHeaders[index] = header[1]
rows = [sanitizedHeaders]
for content in contents:
rows.append(content)
table = Table(rows, tableName)
table.print()
def showTableNames():
cursor = c.execute('''
SELECT name
FROM sqlite_master
WHERE type = "table"
''')
table = Table(cursor.fetchall(), "Table Names")
table.print()
showTableNames()
showTable('materials')
showTable('weapons')
showTable('weaponBlueprints')
con.commit()
con.close()