def _hash_passwords(): try: db = connect_db() cur = db.execute("SELECT id, password FROM users") rs = cur.fetchall() for row in rs: (password, salt) = hash_password(row['password']) db.execute( "UPDATE users SET password = ?, password_salt = ? WHERE id = ?", (password, salt, row['id'])) db.commit() db.close() except Exception as ex: print(str(ex)) os.exit(1)
def execute(cls, sql, params=None): db = connect_db() return db.execute(sql, params)
def do(cls, sql, params=None): db = connect_db() cursor = cls.execute(sql, params) db.commit() return cursor
def _run_sql(filename): db = connect_db() with open(filename, 'r') as f: db.cursor().executescript(f.read()) db.commit()
def generate_sample_users(): with app.app_context(): g.db = connect_db() for i in (1, 2, 3): s = 'user{}'.format(i) User(username=s, password=generate_password_hash(s)).save()