balance-application/generate.py

import random, string
from getpass import getpass
from util.auth_lib import hash
from service.auth_service import AuthData, AuthService

def gen_salt(length = 20):
	letters = string.ascii_lowercase + string.digits + string.punctuation
	return "".join(random.choice(letters) for i in range(length))

def _gen_token():
	deps = string.ascii_lowercase + string.ascii_uppercase + string.digits + string.punctuation
	token = "".join(random.choice(deps) for i in range(20))

	sec = open("./secret_token.txt", "w")
	sec.write(token)

	sec.close()

def _new_account():
	name = input("input your display name: ")
	username = input("input your username: ")
	password = getpass("input your password: ")
	passchk = getpass("type password one more time: ")
	salt = gen_salt()

	if password != passchk:
		return
		
	hashed_password = hash(password, salt)
	packed = AuthData(
		name=name,
		username=username,
		password=hashed_password,
		salt=salt
	)

	service = AuthService()
	service.create(data=packed)

def on_load(conn, cur):
	_gen_token()
	cur.execute(
		"""
		create table account(
			name varchar(25),
			username varchar(25) not null,
			password varchar(100) not null,
			salt varchar(50),
			primary key(username)
		);
		"""
	)
	cur.execute(
		"""
		create table balset(
			id serial primary key,
			uid varchar(25) not null,
			name varchar(50),
			date bigint,
			price bigint,
			buy boolean,
			memo varchar(300),
			constraint FK_Account_ID
				foreign key (uid)
				references account(username)
				on delete CASCADE
		);
		"""
	)
	conn.commit()
	_new_account()