def create_admin(): admin = Admin(username='******',password='******') try: db.session.add(admin) db.session.commit() except: db.session.rollback()
def admin():
if request.method == "GET":
return base_req(
response=[asdict(admin) for admin in Admin.query.all()])
if request.method == "POST":
if "email" not in request.json:
abort(400, "Missing key email")
admin = Admin(email=request.json["email"])
db.session.add(admin)
db.session.commit()
return base_req()
if request.method == "DELETE":
if "email" not in request.json:
abort(400, "Missing key email")
search = Admin.query.filter_by(email=request.json["email"]).all()
if not search:
abort(404, "Admin with specified email does not exist")
db.session.delete(
Admin.query.filter_by(email=request.json["email"]).one())
db.session.commit()
return base_req()
def register():
"""
注册管理员账号
:return:
"""
name = request.json.get('name')
password = request.json.get('password')
adm = Admin.query.filter_by(name=name).first()
if not adm:
new_admin = Admin(name=name, password=password)
try:
db.session.add(new_admin)
db.session.commit()
return jsonify(code=201,
msg='注册成功!',
admin={
'name': new_admin.name,
'password': new_admin.password
})
except Exception as e:
traceback.print_exc()
db.session.rollback()
return jsonify(code=500, msg='注册失败!')
else:
return jsonify(code=400,
msg='用户名已存在！',
admin={
'name': adm.name,
'password': adm.password
})
def admin(id=None):
if request.method == 'GET':
if id is not None:
admin = Admin.query.get(id)
if admin:
return jsonify(admin.serialize()), 200
return jsonify({"msg": "This user not found!"}), 404
else:
admins = Admin.query.all()
admins = list(map(lambda admin: admin.serialize(), admins))
return jsonify(admins), 200
elif request.method == 'PUT':
admin = Admin.query.get(id)
admin.user_id = request.json.get("user_id", "")
admin.update()
return jsonify('Actualizado correctamente'), 200
elif request.method == 'DELETE':
admin = Admin.query.get(id)
admin.delete()
return jsonify('Borrado'), 200
elif request.method == "POST":
admin = Admin()
admin.user_id = request.json.get("user_id", "")
admin.save()
return jsonify(admin.serialize()), 201
def add_admins(update, context):
try:
admin_sent_list = update.effective_message.text.split('\n')
except ValueError:
update.message.reply_text('ورودی اشتباه')
return ADD_ADMINS
for admin in admin_sent_list:
if not admin in context.user_data['admins_dict'].values():
if admin.__contains__('0'):
Admin(username=admin.replace('@', '').replace(' 0', ''),
is_super=True).add()
else:
Admin(username=admin.replace('@', '')).add()
return ADD_ADMINS
def create_admin():
data = request.get_json()
if not data:
return {"response": "No input data provided"}, 400
try:
result = AdminSchema().load(data)
except Exception:
return jsonify({'response': "Invalid input"}), 400
if db.session.query(Admin.adminId).filter_by(
username=result["username"]).scalar() is not None:
return jsonify({'response': "Username is already used"}), 409
password = result["password"]
salt = os.urandom(32) # Новая соль для данного пользователя
key = hashlib.pbkdf2_hmac('sha256', password.encode('utf-8'), salt, 10000)
storage = salt + key
admin = Admin(username=result["username"],
email=result["email"],
password=storage)
db.session.add(admin)
db.session.commit()
adminId = db.session.query(
Admin.adminId).filter_by(username=result["username"]).scalar()
encoded_jwt = bu.encode_auth_token(adminId, 1)
return jsonify({
'response':
"Sign up successfully! Please LOG IN to get access token"
}), 201
def init(username, password):
click.echo('Initializing the database...')
db.create_all()
admin = Admin.query.first()
if admin is not None:
click.echo('The administrator already exists, updating...')
admin.username = username
admin.set_password(password)
else:
click.echo('Creating the temporary administrator account...')
admin = Admin(username=username,
blog_title='Bluelog',
blog_sub_title="No, I'm the real thing.",
name='Admin',
about='Anything about you.')
admin.set_password(password)
db.session.add(admin)
category = Category.query.first()
if category is None:
click.echo('Creating the default category...')
category = Category(name='Default')
db.session.add(category)
db.session.commit()
click.echo('Done.')
def register():
form = RegistrationForm(request.form)
if request.method == "POST":
error = not form.validate_on_submit()
else:
error = False
if request.method == 'GET' or error:
return render_template("registration.html", form=form, error=error)
elif request.method == 'POST':
email = form.email.data
name = form.name.data
password = form.password.data
if Admin.query.first():
role = 'normal'
enabled = False
else:
role = 'full'
enabled = True
admin = Admin(name, email, password, enabled, role)
token = utils.generate_confirmation_token(email)
confirm_url = url_for('confirm', token=token, _external=True)
html = render_template('confirmation_email.html',
confirm_url=confirm_url, admin=admin)
message = "Hi there {0}!\n\nThanks for signing up. Please follow this link to activate your account:\n\n{1}\n\nCheers!".format(
admin.name, confirm_url)
subject = "Please confirm your email"
utils.mailgun_send_message(subject, [email], message, html=html)
db.session.add(admin)
db.session.commit()
message = "Please check your inbox for a confirmation email."
return render_template("success.html", message=message)
def otherUserRegistration():
if(session['logged_in'] == True): #if user login
if(request.method == 'GET'):
return render_template('OtherUsersRegistrationPage.html')
elif(request.method == 'POST'):
try:
userId = request.json['user_id']
name = request.json['name']
userNIC = request.json['userNIC']
userType = request.json['userType']
newUser = ""
if(userType == "teacher"): #if a teache, create new teacher object
newUser = Teacher(userId, name, userNIC)
if(userType == "admin"): #if a admin create new admin object
newUser = Admin(userId, name, userNIC)
DataPipelineObj.insert_data(newUser) #Add details to the database
return jsonify({'status':"User successfully registered"})
except:
return jsonify({'status':"Fill the required details"})
else:
abort(405)
else: #if user not login
return render_template('showSignIn.html')
def set_roles(
user_id, is_recruiter=None, is_senior_recruiter=None, is_admin=None,
current_user=None):
user_admin_access_check(current_user)
user = User.get(user_id)
if is_senior_recruiter:
if not user.recruiter:
db.session.add(Recruiter(id=user.id, is_senior=True))
elif not user.recruiter.is_senior:
user.recruiter.is_senior = True
elif is_recruiter:
if not user.recruiter:
db.session.add(Recruiter(id=user.id, is_senior=False))
elif is_recruiter == False and user.recruiter:
remove_recruiter(user.recruiter)
if is_senior_recruiter == False and user.recruiter and user.recruiter.is_senior:
user.recruiter.is_senior = False
if is_admin and not user.admin:
db.session.add(Admin(id=user.id))
elif is_admin == False and user.admin:
db.session.delete(user.admin)
db.session.commit()
if is_recruiter or is_senior_recruiter or is_admin:
delete_any_open_application(user_id)
return {'status': 'ok'}
def init(username, password):
print('Initializing the database...')
db.create_all()
admin = Admin.query.first()
if admin is not None:
print('The administrator already exists, updating...')
admin.username = username
admin.set_password(password)
else:
print('Creating the temporary administrator account...')
admin = Admin(username='******',
blog_title="Flaskblog",
blog_sub_title="No,I am the real thing",
name="Miro",
about="I am a fun guy....")
admin.set_password(password)
db.session.add(admin)
category = Category.query.first()
if category is None:
print('Creating the default category...')
category = Category(name='Default')
db.session.add(category)
db.session.commit()
print('Done.')
def admins():
if request.method == 'POST':
form = AdminForm(request.form)
if form.validate():
admin = Admin()
form.populate_obj(admin)
admin.pw_hash = bcrypt.generate_password_hash(form.password.data)
g.db.session.add(admin)
g.db.session.commit()
login_user(get_admin(g.db, admin.email))
flash('Welcome to xAPI Scavenger Hunt', 'success')
logger.info(
'Admin registration form was submitted successfully for %s',
admin.email)
return make_response(
render_template('settings.html', form=SettingForm()))
logger.info(
'Admin registration form was submitted with'
' invalid information. Errors: %s', form.errors)
flash(
'There was an error creating your admin profile.'
' Please try again.', 'warning')
return render_template('homepage.html',
form=form,
display_login_link=True)
return login()
def login(email, password):
"""管理员登录
当验证登录信息正确后，更新用户上次登录时间为系统当前时间
:param email: Email :type: string.
return: : Admin Obj
"""
admin = Admin.query.filter(Admin.email == email).first()
all_admin = Admin.query.all()
try:
assert all_admin != []
# 管理员status若不为1，则表示不处于活跃状态，则不允许登录
if admin and admin.status != '1':
return None
if admin and admin.check_password(password):
admin.last_login = datetime.now()
admin.update()
return admin
except:
if not all_admin:
admin = Admin('admin', email, password)
admin.save()
print '没有管理员，添加第一个管理员'
return True
print '找不到用户'
return None
def add_user():
default_admin = Admin(username='******',
email='*****@*****.**',
password='******',
phone_no='7777777777',
postal_address='1, Beverly Park Circle, California')
username = input("username: "******"email: ")
password = input("password: "******"phone no: ")
postal_address = input("postal address: ")
role = input("role: ")
res = default_admin.add_user(username=username,
email=email,
password=password,
phone_no=phone_no,
postal_address=postal_address,
role=role,
session_id=None)
print('User Added')
return res
def entrada(pieza):
forma = FormaEntrada()
muestraObj = db.session.query(Muestra).filter_by(sku=pieza).first()
if forma.validate_on_submit():
usuario = int(forma.empleado.data)
admin = int(forma.encargado.data)
usuarioObj = db.session.query(Usuario).filter_by(
numero=usuario).first()
if not usuarioObj:
usuarioObj = Usuario(numero=usuario)
adminObj = db.session.query(Admin).filter_by(numero=admin).first()
if not adminObj:
adminObj = Admin(numero=admin)
muestraObj.cantidad = 1
ordenEntrega = Orden(muestra=muestraObj,
usuario=usuarioObj,
admin=adminObj,
tipo='Entrada',
timestamp=datetime.now())
db.session.add(ordenEntrega)
db.session.commit()
flash('Se entrego la pieza: {}'.format(muestraObj.sku))
return redirect(url_for('main'))
return render_template('entrada.html', entrada=forma, pieza=muestraObj)
def test_get_by_id(self):
"""Get user by ID."""
user = Admin(username='******', password='******')
user.save()
retrieved = Admin.query.get_or_404(user.id)
assert retrieved == user
def addAdmin():
if request.method == 'POST':
if 'admin' in session:
# Request form data
adminName = request.form['Name']
adminPassword = request.form['Password']
# Query database for data
found_admin = Admin.query.filter_by(admin_name=adminName).first()
if found_admin != '':
flash('Admin already exists')
return redirect(url_for('adminController.addAdmin'))
else:
# Add data to database
admin = Admin(admin_name=adminName,
admin_password=adminPassword)
db.session.add(admin)
db.session.commit()
flash('New admin created successfully')
return redirect(url_for('adminController.admin'))
else:
flash('Please Login')
return redirect(url_for('adminController.adminLogin'))
else:
if 'admin' in session:
return render_template('add-admin.jinja')
else:
abort(403)
def index():
md5 = hashlib.md5()
md5.update('xxx')
pwd = md5.hexdigest()
admin = Admin('imagemlt', pwd)
db.session.add(admin)
db.session.commit()
return '<h1>this is index</h1>'
def create_admin(username, password, email):
# avatar = GithubAvatarGenerator()
# path = '..' + sep +'static'+ sep+ 'admin'+sep +'image'+ sep + email +'.png'
# avatar.save_avatar(filepath='.' + sep +'static'+ sep+ 'admin'+sep +'image'+ sep + email +'.png')
user = Admin(username=username, password=password, email=email)
db.session.add(user)
db.session.commit()
print('Admin添加成功！！！')
def create_room(message):
print("Added")
print(message['data']['user'])
user = User.query.filter_by(email=message['data']['user']).first()
room = Topic.query.filter_by(topicname=session.get('room')).first()
adm = Admin(user.uid, room.uid)
db.session.add(adm)
db.session.commit()
async def promote_user(message: types.Message):
user_id = message.reply_to_message.from_user.id
try:
db.add_admin(Admin(user_id))
await message.reply("Ok, user promoted to admins.")
except ValueError:
await message.reply("Already in admins.")
def create_creator_admin(self):
admin = Admin(creator_user_id=current_user.id,
active=True,
email=current_user.email,
role_id=100,
user_id=current_user.id,
corporation_id=self.corporation_id)
db.session.add(admin)
return admin
def createsuperuser(email):
check_email = Admin.query.filter(Admin.email == email).first()
if check_email is not None:
click.echo('Email already exists. Please try with another email.')
else:
admin = Admin(email=email, password=generate_password_hash('admin'))
db.session.add(admin)
db.session.commit()
click.echo('Superuser created successfully! & default password set to "admin"')
def initialize():
try:
user = input('请输入初始化用户名:').strip()
pwd = input('请输入初始化密码:').strip()
obj = Admin(user,pwd)
obj.save()
return True
except Exception as e:
print(e)
def fake_admin():
admin = Admin(username='******',
blog_title='江湖人称蛋总',
blog_subtitle='敬畏技术，谦卑而行',
name='王',
about='ssssss')
admin.set_password('password')
db.session.add(admin)
db.session.commit()
def initDbForE2e(wipe=True):
if wipe:
clearDB()
# TODO: Id of user a
admin_id = 123
db.session.add(
Character(
id=admin_id,
user_id=admin_id,
name='ADMIN NAME',
corporation_id=ascee_corp_id,
refresh_token='YOUR TOKEN HERE',
))
db.session.add(Admin(
id=admin_id,
user_id=admin_id,
name='Billy Admin',
))
# TODO: Id of recruiter user
recruiter_id = 345
db.session.add(
Character(id=recruiter_id,
user_id=recruiter_id,
name='RECRUITER NAME',
corporation_id=ascee_corp_id,
refresh_token='YOUR TOKEN HERE'))
db.session.add(Recruiter(
id=recruiter_id,
name='RECRUITER NAME',
))
# TODO: Id of applicant user
character_id = 234
db.session.add(
Character(id=character_id,
user_id=character_id,
name='APPLICANT NAME',
corporation_id=ascee_corp_id,
corporation=Corporation(id=ascee_corp_id, name='ASCEE'),
refresh_token='YOUR TOKEN HERE'))
db.session.add(User(
id=character_id,
name='APPLICANT NAME',
))
db.session.add(Question(text='How long have you been playing Eve?'))
db.session.add(Question(text='PVP or PVE? Why?'))
db.session.add(Application(user_id=character_id))
db.session.commit()
def create_admin(self):
admin = Admin(creator_user_id=current_user.id,
about=self.about,
email=self.email,
phone=self.phone,
role_id=self.role_id,
corporation_id=self.corporation_id,
active=False)
add_commit(admin)
return admin
def fake_admin():
admin = Admin(username='******',
password='******',
blog_title='PY-Blog',
blog_sub_title="人间正道是沧桑",
name='clear',
about='i am clear')
db.session.add(admin)
db.session.commit()
def create_user():
'''
Register as a new user.
'''
data = request.get_json()
hash_pwd = data['pwdhash']
new_user = Admin(data['username'], hash_pwd)
db.session.add(new_user)
db.session.commit()
return jsonify({'msg': 'new user created'})
def save_user(username, userdata):
admin = Admin.query.filter_by(uid=username).first()
if admin is None:
admin = Admin(username, userdata['name'].decode("utf-8"),
userdata['email'].decode("utf-8"))
db.session.add(admin)
else:
admin.name = userdata['name'].decode("utf-8")
admin.email = userdata['email'].decode("utf-8")
db.session.commit()
return admin
