def add_user(db, user):
'''Add a user to the database
'''
username = user['username']
password = user['password']
email = user['email']
lulebo_username = user['lulebo_username']
lulebo_password = user['lulebo_password']
print('Add user ' + username + '')
try:
user_models.User.get_by_name(username)
print('\tSkipping -- user _does_ exist.')
return False
except user_models.UserNotFoundError:
pass
try:
u = user_models.User(username=username,
password=password,
email=email,
uuid=str(uuid.uuid4()),
lulebo_username=lulebo_username,
lulebo_password=lulebo_password)
db.session.add(u)
db.session.commit()
except Exception as e:
print(e)
print('\tSkipping -- unknown error.')
return False
return True
def post(self, request):
username = request.data.get('username')
password = request.data.get('password')
email = request.data.get('email')
mobile = request.data.get('mobile')
newpassword = encrypt(password)
user_object = models.User(username=username,
password=newpassword,
mobile=mobile,
email=email,
state=False)
user_object.save()
returndata = {}
data = {}
meta = {}
data['id'] = user_object.id
data['role_id'] = user_object.rid
data['username'] = user_object.username
data['mobile'] = user_object.mobile
data['email'] = user_object.email
meta['msg'] = '用户创建成功'
meta['status'] = 201
returndata['data'] = data
returndata['meta'] = meta
return Response(returndata, status=201)
def get_or_make_user(code, mode):
try:
user = m.User.objects.get(code=code, mode=mode)
except m.User.DoesNotExist:
user = m.User(code=code, mode=mode)
user.save()
return user
def create_user():
while True:
name = input("Username: "******"User successfully created.")
return
def createUser(obj):
if userExists(obj['username'], obj['email']):
return errorhelper.generateError('username or password already exist',
400)
user_object = models.User(
email=obj['email'],
username=obj['username'],
admin=obj['admin'],
)
security.hashPassword(user_object, obj['password'])
models.db.session.add(user_object)
return helpers.commitResponse(models.db.session, {}, user_object)