Start work on boards in the database
This commit is contained in:
parent
59d9ea4bf4
commit
1980ef2c08
30
database.py
30
database.py
|
@ -14,7 +14,8 @@ class userstatus(enum.Enum):
|
||||||
csprng = random.SystemRandom()
|
csprng = random.SystemRandom()
|
||||||
|
|
||||||
def add_user(db, *, username, password, email, parent, status):
|
def add_user(db, *, username, password, email, parent, status):
|
||||||
"""Add a user to the database"""
|
"""Add a user to the database
|
||||||
|
Will not commit the changes itself, so run .commit() on the database object yourself"""
|
||||||
global csprgn
|
global csprgn
|
||||||
|
|
||||||
assert type(username) == str
|
assert type(username) == str
|
||||||
|
@ -26,6 +27,9 @@ def add_user(db, *, username, password, email, parent, status):
|
||||||
# Generate a user ID. SQLite uses 64 bit signed ints, so generate at max 2⁶³-1
|
# Generate a user ID. SQLite uses 64 bit signed ints, so generate at max 2⁶³-1
|
||||||
userid = csprng.randrange(2**63)
|
userid = csprng.randrange(2**63)
|
||||||
|
|
||||||
|
# Unicode normalize the username
|
||||||
|
username = unicodedata.normalize('NFKC', username)
|
||||||
|
|
||||||
# First unicode normalize the password, then hash it with argon2
|
# First unicode normalize the password, then hash it with argon2
|
||||||
password = unicodedata.normalize('NFKC', password)
|
password = unicodedata.normalize('NFKC', password)
|
||||||
password = argon2.hash(password)
|
password = argon2.hash(password)
|
||||||
|
@ -36,10 +40,9 @@ def add_user(db, *, username, password, email, parent, status):
|
||||||
# Add the user into the database
|
# Add the user into the database
|
||||||
cursor = db.cursor()
|
cursor = db.cursor()
|
||||||
cursor.execute('INSERT INTO users VALUES (?, ?, ?, ?, ?, ?, ?);', (userid, parent, status, password, username, email, ''))
|
cursor.execute('INSERT INTO users VALUES (?, ?, ?, ?, ?, ?, ?);', (userid, parent, status, password, username, email, ''))
|
||||||
db.commit()
|
|
||||||
|
|
||||||
def initialize_users(db, admin_user, admin_password):
|
def initialize_users(db, admin_user, admin_password):
|
||||||
"""Creates a bare-bones database with only admin user
|
"""Creates a bare-bones user table with only admin user
|
||||||
This should never be run outside of the initialization script"""
|
This should never be run outside of the initialization script"""
|
||||||
|
|
||||||
cursor = db.cursor()
|
cursor = db.cursor()
|
||||||
|
@ -58,6 +61,25 @@ def initialize_users(db, admin_user, admin_password):
|
||||||
comment text NOT NULL
|
comment text NOT NULL
|
||||||
);''')
|
);''')
|
||||||
|
|
||||||
|
add_user(db, username = admin_user, password = admin_password, email = '', parent = None, status = userstatus.admin)
|
||||||
|
|
||||||
db.commit()
|
db.commit()
|
||||||
|
|
||||||
add_user(db, username = admin_user, password = admin_password, email = '', parent = None, status = userstatus.admin)
|
def initialize_boards(db, boards):
|
||||||
|
"""Creates a table of boards
|
||||||
|
This should never be run outside of the initialization script"""
|
||||||
|
|
||||||
|
cursor = db.cursor()
|
||||||
|
|
||||||
|
cursor.execute('''CREATE TABLE boards (
|
||||||
|
id integer NOT NULL PRIMARY KEY,
|
||||||
|
name text NOT NULL
|
||||||
|
);''')
|
||||||
|
|
||||||
|
# .executemany() wants them in the format [("board1",), ("board2",), …]
|
||||||
|
boards = [(board_name,) for board_name in boards]
|
||||||
|
|
||||||
|
# Use NULL to have SQLite generate the IDs automatically
|
||||||
|
cursor.executemany('INSERT INTO boards VALUES (NULL, ?);', boards)
|
||||||
|
|
||||||
|
db.commit()
|
||||||
|
|
|
@ -9,6 +9,7 @@ def generate_nav(*, soup):
|
||||||
# TODO: Don't generate link to a board if we're at the index
|
# TODO: Don't generate link to a board if we're at the index
|
||||||
nav_tag = soup.new_tag('nav')
|
nav_tag = soup.new_tag('nav')
|
||||||
|
|
||||||
|
# TODO: Read these from the database
|
||||||
for board in ['a', 'b', 'his', 'g']:
|
for board in ['a', 'b', 'his', 'g']:
|
||||||
url = config.url_prefix + '/' + urllib.parse.quote(board, safe = '') + '/'
|
url = config.url_prefix + '/' + urllib.parse.quote(board, safe = '') + '/'
|
||||||
a_tag = soup.new_tag('a', href = url)
|
a_tag = soup.new_tag('a', href = url)
|
||||||
|
|
|
@ -7,3 +7,6 @@ if __name__ == '__main__':
|
||||||
username = input('admin username: ')
|
username = input('admin username: ')
|
||||||
password = input('admin password: ')
|
password = input('admin password: ')
|
||||||
database.initialize_users(db, username, password)
|
database.initialize_users(db, username, password)
|
||||||
|
|
||||||
|
boards = input('boards: ').split()
|
||||||
|
database.initialize_boards(db, boards)
|
||||||
|
|
Loading…
Reference in New Issue