def run(self):
admin = Admin.where('admin_user_id', '=', OWNER_USER_ID).first()
if not admin:
owner = Admin()
owner.admin_user_id = OWNER_USER_ID
owner.is_owner = True
owner.save()
def add_admin(update: Update, _context: CallbackContext) -> None:
user_id = update.message.text.partition(' ')[2]
user_id = int(user_id)
if is_user_owner(update.effective_user.id):
admin = Admin()
admin.admin_user_id = user_id
admin.save()
update.message.reply_text(f"User {user_id} has been added as admins")
def create():
data = request.json
name = data.get('name')
email = data.get('email')
password = data.get('password')
if name and email and password:
admin = Admin(name=name, email=email, password=password)
if admin.save():
token = create_access_token(identity=admin.id)
return jsonify({
"auth_token": token,
"message": "Successfully created an admin and signed in",
"status": "success",
"admin": {
"id": admin.id,
"name": admin.name,
"email": admin.email
}
})
elif admin.errors != 0:
return jsonify({
"message": [error for error in admin.errors],
"status": "failed"
})
else:
return jsonify({
"message": "All fields are required!",
"status": "failed"
})
def post(self, request):
# 注册
username = request.json.get("username")
password = request.json.get("password")
nickname = request.json.get("nickname")
account_type = request.json.get("account_type")
if not all([username, password, nickname, account_type]):
return JsonErrorResponse(
"username, password, nickname, account_type are needed", 400)
new_admin = Admin(username=username,
password=password,
nickname=nickname,
account_type=account_type)
try:
new_admin.save()
except Exception, e:
print e
return JsonErrorResponse("Fail" + e.message)
def post(self, request):
# 注册
username = request.json.get("username")
password = request.json.get("password")
nickname = request.json.get("nickname")
account_type = request.json.get("account_type")
if not all([username, password, nickname, account_type]):
return JsonErrorResponse("username, password, nickname, account_type are needed", 400)
new_admin = Admin(
username=username,
password=password,
nickname=nickname,
account_type=account_type
)
try:
new_admin.save()
except Exception, e:
print e
return JsonErrorResponse("Fail" + e.message)
def admin_register():
"""
Easy register of a new admin.
Beta system
"""
from models.admin import Admin
if "admin-session" in session:
return redirect(url_for("admin.new_debts"))
if request.method == "POST":
names = request.form["names"]
last_names = request.form["lastNames"]
username = request.form["email"]
password = request.form["password"]
params = [names, last_names, username, password]
for n in params:
if len(n) == 0:
return render_template("register_adm.html", error_params=True)
allAdmins = [admin.username for admin in Admin.query.all()]
if username in allAdmins:
return render_template("register_adm.html", error_exist=True)
else:
md5Pwd = md5(password.encode('utf-8')).hexdigest()
try:
newAdmin = Admin(names, last_names, username, md5Pwd)
except:
return render_template("register_adm.html", error_data=True)
newAdmin.save()
session['admin-session'] = username
return redirect(url_for('admin.new_debts'))
return render_template("register_adm.html")
# ran_image = []
# ran_description = sample(["ngoan hiền", "dễ thương", "lễ phép với gia đình"], 2)
# ran_measurement = [randint(80,100), randint(50,70), randint(80,100)]
# new_service = Service(
# name = fake.name(),
# yob = randint(1990,2000),
# gender = randint(0,1),
# height = randint(150,190),
# phone = fake.phone_number(),
# address = fake.address(),
# status = choice([True,False]),
# description = "{0}, {1}".format(*ran_description),
# measurements = "{0} - {1} - {2}".format(*ran_measurement),
# image = image
# )
# for i in range(50):
# new_customer = Customer(
# name = fake.name(),
# gender = randint(0, 1),
# email = fake.email(),
# phone = fake.phone_number(),
# job = fake.job(),
# company = fake.company()
# )
# # new_customer.save()
new_admin = Admin(name="admin", password="******")
new_admin.save()
class Controller:
def __init__(self):
self.admin = Admin()
def login(self, request):
data = json.loads(request.get_data())
result = self.admin.findByUsernameAndPassword(data)
if result == None:
return response(result, code=50000, message='账号或密码错误！')
else:
loginTime = datetime.datetime.utcnow()
self.admin.update_one(result['username'], {'loginTime': loginTime})
auth = {'username': result['username'], 'role': result['role']}
data = {}
data['token'] = Jwt.jwtEncode(auth).decode('utf-8')
return response(data, code=20000, message='登陆成功！')
def getInfo(self, request):
p = Jwt.authHeader(request)
res = json.loads(p)
if res['code'] == 20000:
result = self.admin.findByUsername(res['data']['username'])
return response(result, code=20000, message='获取成功！')
else:
return p
def logout(self):
return response({}, code=20000, message='登出成功！')
def getList(self, request):
p = Jwt.authHeader(request)
res = json.loads(p)
if res['code'] == 20000:
data = json.loads(request.get_data())
result = self.admin.find(data)
return response(result, code=20000, message='获取成功！')
else:
return p
def delete(self, request):
p = Jwt.authHeader(request)
res = json.loads(p)
if res['code'] == 20000:
data = json.loads(request.get_data())
result = self.admin.delete(data)
print(type(result))
return response({}, code=20000, message='删除成功！')
else:
return p
def edit(self, request):
p = Jwt.authHeader(request)
res = json.loads(p)
if res['code'] == 20000:
data = json.loads(request.get_data())
if 'username' in data:
result = self.admin.update_one(data['username'], data)
else:
data['username'] == res['username']
result = self.admin.update_one(data['username'], data)
print(type(result))
return response({}, code=20000, message='修改成功！')
else:
return p
def create(self, request):
p = Jwt.authHeader(request)
res = json.loads(p)
if res['code'] == 20000:
data = json.loads(request.get_data())
result = self.admin.save(data)
if result == None:
return response({}, code=50000, message='该用户名已存在！')
return response({}, code=20000, message='创建成功！')
else:
return p
def changePwd(self, request):
p = Jwt.authHeader(request)
res = json.loads(p)
if res['code'] == 20000:
admin = dict()
data = json.loads(request.get_data())
admin['username'] = res['data']['username']
admin['oldPass'] = data['oldpass']
admin['password'] = data['password']
result = self.admin.update_pwd(admin)
if result != None:
return response({}, code=20000, message='修改成功！')
else:
return response({}, code=50000, message='修改失败,请检查原密码！')
else:
return p