def create_user():
username = prompt('用户名').strip()
password = prompt_pass('密码').strip()
repassword = prompt_pass('重复密码').strip()
is_super = prompt_bool('超级用户')
is_staff = prompt_bool('后台用户')
if not all([username, password, repassword]):
print('请输入完整数据')
return
if password != repassword:
print('两次输入密码不同')
return
connect_db()
user = User()
user.username = username
user.password = password
user.is_super = is_super
user.is_staff = is_staff
db_session.add(user)
db_session.commit()
print('创建用户成功!')
def create_user():
connect_db()
username = prompt('username')
password = prompt_pass('password')
repassword = prompt_pass('repassword')
is_super = prompt_bool('is super(default: True)', default='y')
is_staff = prompt_bool('is staff(default: True)', default='y')
if password != repassword:
print('the two password is not same!')
return
user = User()
user.username = username
user.password = password
user.is_super = is_super
user.is_staff = is_staff
db_session.add(user)
db_session.commit()
print('create user success')
