def check_login(login, password):
if login and password:
password = generate_password_hash(password)
account = db.session.query(Accounts.status).filter_by(name=login.encode("utf-8"), password=password).limit(
1).first()
if account:
if account.status != BANNED_ACCOUNT:
return account.status
return None
def add_account(name, password):
password = generate_password_hash(password)
try:
user = Accounts(name, str(password), USER_ACCOUNT)
db.session.add(user)
db.session.commit()
except:
db.session.rollback()
raise
def add_account(name, password):
password = generate_password_hash(password)
try:
user = Accounts(name, str(password), USER_ACCOUNT)
db.session.add(user)
db.session.commit()
except:
db.session.rollback()
raise
def check_login(login, password):
if login and password:
password = generate_password_hash(password)
account = db.session.query(Accounts.status).filter_by(
name=login.encode("utf-8"), password=password).limit(1).first()
if account:
if account.status != BANNED_ACCOUNT:
return account.status
return None
def update_account(user_id, name, password):
account = get_account(user_id)
if account:
try:
account.name = name
if password != "":
password = generate_password_hash(password)
account.password = str(password)
db.session.commit()
except:
db.session.rollback()
raise
def update_account(user_id, name, password):
account = get_account(user_id)
if account:
try:
account.name = name
if password != "":
password = generate_password_hash(password)
account.password = str(password)
db.session.commit()
except:
db.session.rollback()
raise
def signup():
form = RegisterForm()
if form.validate_on_submit():
hashed_password = generate_password_hash(form.password.data)
new_user = User(username=form.username.data,
email=form.email.data,
password=hashed_password)
db.session.add(new_user)
db.session.commit()
session['id'] = new_user.id
login_user(new_user, remember=False)
return redirect(url_for('board'))
return render_template('signup.html', form=form)
def generate_account():
try:
name = None
password = None
status = 2
need_help = False
need_show_hash_string = False
opts, args = getopt.getopt(sys.argv[1:], "n:p:s:hv")
if not opts:
show_args_error()
return
for key, value in opts:
if key == "-n":
name = value
if key == "-p":
password = value
if key == "-s":
if value == "admin":
status = 2
if value == "user":
status = 1
if key == "-h":
need_help = True
if key == "-v":
need_show_hash_string = True
if need_help:
show_help()
return
if name and password:
hash_str = generate_password_hash(password)
if need_show_hash_string:
show_hash_string(hash_str)
show_query(name, hash_str, status)
else:
show_args_error()
except getopt.GetoptError as err:
print str(err)
def set_password(self, password):
self.pw_hash = generate_password_hash(password)