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