nortti
/
buranun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add foreign key constraint to enforce proper parent field in users table

This commit is contained in:
Juhani Krekelä 2018-05-10 00:12:39 +03:00
parent 1980ef2c08
commit c2ee55e5e9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
database.py
View File

@ -39,6 +39,7 @@ def add_user(db, *, username, password, email, parent, status):
# Add the user into the database
cursor = db.cursor()
cursor.execute('PRAGMA foreign_keys = ON;') # Fail if we insert a user with bogus parent field
cursor.execute('INSERT INTO users VALUES (?, ?, ?, ?, ?, ?, ?);', (userid, parent, status, password, username, email, ''))
def initialize_users(db, admin_user, admin_password):
@ -58,7 +59,9 @@ def initialize_users(db, admin_user, admin_password):
username text NOT NULL,
email text NOT NULL,
comment text NOT NULL
comment text NOT NULL,
FOREIGN KEY(parent) REFERENCES users(id)
);''')
add_user(db, username = admin_user, password = admin_password, email = '', parent = None, status = userstatus.admin)