def create_superuser():
admin = Admin()
admin.manager = 'admin'
admin.password = generate_password_hash('123')
db.session.add(admin)
db.session.commit()
return redirect(url_for('admin.login'))
def fake_admin():
admin = Admin(
username = "******",
)
admin.set_password('123456')
db.session.add(admin)
db.session.commit()
def init(username, password):
click.echo('初始化数据库...')
db.create_all()
admin = Admin.query.first()
if admin is not None:
click.echo('管理员已经存在，正在更新...')
admin.username = username
admin.set_password(password)
else:
click.echo('正在创建临时管理员帐户...')
admin = Admin(username=username,
blog_title='AmeBlog',
blog_sub_title="Hello world",
name='Admin',
about='Hello world')
admin.set_password(password)
db.session.add(admin)
category = Category.query.first()
if category is None:
click.echo('正在创建默认类别...')
category = Category(name='Default')
db.session.add(category)
db.session.commit()
click.echo('完成.')
def init(username, password):
"""Building Bluelog, just for you."""
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
)
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 initadmin(username, password):
"""Initialize the administrator user and default category"""
from app.models import Admin, Category
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,
email='*****@*****.**',
blog_title='Blog',
blog_sub_title='blog sub title, something.',
name='Mr.Wang',
about="um, The man was lazy and didn't leave a profile")
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 test_admin_model(self):
"""Test number of admin."""
ADMIN = list()
ADMIN.append(Admin(1, 'Sam', '*****@*****.**', 'hardtoguess'))
ADMIN.append(Admin(2, 'Joan', '*****@*****.**', 'hardtoguess'))
self.assertEqual(len(ADMIN), 2)
self.assertEqual(ADMIN[0].name, 'Sam')
def test_admin_expired_confirmation_token(self):
u = Admin(password='******')
db.session.add(u)
db.session.commit()
token = u.generate_auth_token(1)
time.sleep(2)
self.assertFalse(u.verify_auth_token(token) == u)
def init(username, password):
"""
Init Admin account
"""
click.echo('Initializing Database...')
db.create_all()
admin = Admin.query.first()
with db.auto_commit():
if admin:
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='临时博客名',
blog_sub_title='临时副标题',
name='临时昵称',
about='临时关于'
)
admin.set_password(password)
db.session.add(admin)
category = Category.query.first()
if category is None:
click.echo('Create the default category...')
category = Category(name='默认')
db.session.add(category)
click.echo('Done.')
def create_admin(client, username="******", password="******"):
with client.application.test_request_context():
admin = Admin(username=username)
admin.set_password(password)
db.session.add(admin)
db.session.commit()
return admin
def editPost():
data = request.get_json() or {}
Admin.edit(UnibCombos, data)
db.session.commit()
response = jsonify({})
response.status_code = 201
return response
def fake_admin():
admin = Admin(username='******',
email='*****@*****.**',
blog_title='Blog',
blog_sub_title='blog sub title, something.',
name='Mr.Wang',
about="um, The man was lazy and didn't leave a profile")
admin.set_password('123456')
db.session.add(admin)
db.session.commit()
def fake_admin():
admin = Admin(
username='******',
blog_title='Bluelog',
blog_sub_title="No, I'm the real thing.",
name='Mima Kirigoe',
about='Um, l, Mima Kirigoe, had a fun time as a member of CHAM...')
admin.set_password('helloflask')
db.session.add(admin)
db.session.commit()
def admin(username, password, email):
"""设置管理员用户名、密码以及邮箱"""
# 处理 MySQL 错误
try:
admin = Admin.query.first()
except Exception as e:
if '1146' in str(e.orig):
click.echo('数据表不存在，请执行 `flask initdb` 创建数据表')
else:
print(e)
click.echo('请检查错误信息')
return
with db.auto_commit():
if admin:
click.echo('更新管理员账户信息...')
admin.username = username
admin.password = password
admin.email = email
else:
click.echo('创建管理员账户中...')
admin = Admin()
admin.username = username
admin.password = password
admin.email = email
admin.blog_title = '临时博客名'
admin.blog_subtitle = '临时博客副标题'
admin.blog_about = '临时博客关于'
admin.nickname = '临时昵称'
db.session.add(admin)
click.echo('Done.')
def post(self):
'''
添加用户
'''
args_register = parse_register.parse_args()
password = args_register.get('password')
username = args_register.get('username').lower()
name = args_register.get('name')
email = args_register.get('email')
phone = args_register.get('phone')
has_admin = Admin.query.filter_by(username = username,is_del='0').first()
if has_admin:
abort(RET.Forbidden,msg='管理员已存在')
admin = Admin()
admin.username = username
admin.password = password
admin.name = name
admin.email = email
admin.phone = phone
admin.last_editor = g.admin.username
if admin.add():
data = {
'status':RET.Created,
'msg':'新增管理员成功',
'data':admin
}
return marshal(data,sing_user_fields)
abort(RET.BadRequest,msg='新增失败')
def register():
form = RegistrationForm()
if form.validate_on_submit():
user = Admin(username=form.username.data, email=form.email.data, phone=form.phone.data,department=form.department.data, usertype=int(form.usertype.data))
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Congratulations, you registered a new user for the system!')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def admin_add():
form = AdminForm()
if form.validate_on_submit():
data = form.data
admin = Admin(name=data['name'], role_id=data['role_id'], is_super=1)
admin.set_password(data['password'])
db.session.add(admin)
db.session.commit()
flash("管理员添加成功", "success")
return redirect(url_for("admin.admin_add"))
return render_template("admin/admin_add.html", form=form)
def fake_admin():
admin = Admin(
username = '******',
blog_title = 'nblog',
blog_sub_title = 'blog blog',
name = 'Li Ning',
about = '个人主页'
)
admin.set_password('20201111')
db.session.add(admin)
db.session.commit()
def run(self):
username = input("Username:"******"Admin {} created".format(username), fg="green")
else:
click.secho(
"Admin with username {} already exists".format(username),
fg="red")
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
admin = Admin(email=form.email.data)
admin.set_password(form.password.data)
db.session.add(admin)
db.session.commit()
flash('Congratulations, you are now an Admin', 'success')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def register():
form = AdminRegistrationForm()
if form.validate_on_submit():
user = Admin(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash(
'Congratulations, you are now a registered user! Login to contunue.'
)
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def register():
try:
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = Admin(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
except Exception as e:
raise e
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegistrationForm()
if form.validate_on_submit():
admin = Admin(email=form.email.data,
first_name=form.first_name.data,
last_name=form.last_name.data)
admin.setPassword(form.password.data)
db.session.add(admin)
db.session.commit()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('auth.login'))
return render_template('auth/register.html', title='Register', form=form)
def admin_add():
"""添加管理员"""
form = AdminForm()
if form.validate_on_submit():
data = form.data
admin_count = Admin.query.filter_by(name=data["name"]).count()
if admin_count == 1:
flash("管理员名称已存在", "err")
return redirect(url_for('admin.admin_add'))
admin = Admin(name=data["name"],
pwd=generate_password_hash(data["pwd"]),
is_super=1,
role_id=data["role_id"])
db.session.add(admin)
db.session.commit()
flash("添加管理员成功", "ok")
# 记录操作日志
adminoplog = AdminOpLog(
admin_id=session["admin_id"],
ip=request.remote_addr, # 获取登陆ip,
reason="添加了一个管理员：%s→%s" % (session["admin"], data["name"]))
db.session.add(adminoplog)
db.session.commit()
return redirect(url_for("admin.admin_add"))
return render_template('admin/admin_add.html', form=form)
def admin_add():
form = AdminAddForm()
if form.validate_on_submit():
data = form.data
admin = Admin.query.filter_by(name=session["admin"]).first()
check_admin = Admin.query.filter_by(name=data['adminname']).first()
role = Role.query.filter_by(name=data['roletype']).first()
if admin.role == None:
flash('您没有添加权限')
return redirect(url_for("Admin.admin_add"))
if admin.role.name != '超级管理员':
flash('您没有添加权限')
return redirect(url_for("Admin.admin_add"))
elif check_admin is None and admin.role.name == '超级管理员':
add_admin = Admin(name=data['adminname'],
password=data['password'],
role=role)
adminOplog = Oplog(reason='添加管理员',
ip=request.environ['REMOTE_ADDR'],
admin=admin)
db.session.add(adminOplog)
db.session.add(add_admin)
flash('管理员已添加')
return redirect(url_for("Admin.admin_add"))
else:
flash('管理员昵称已存在')
return redirect(url_for("Admin.admin_add"))
return render_template('admin/admin_add.html', form=form)
def register_admin():
if current_user.is_authenticated:
return redirect(url_for('admin.dashboard_account'))
form = AdminRegistrationForm()
if form.validate_on_submit():
admin = Admin(admin_full_name=form.admin_full_name.data,
admin_email=form.admin_email.data,
admin_phone=form.admin_phone.data)
admin.set_password(form.admin_password.data)
db.session.add(admin)
db.session.commit()
flash('Admin successfully registered. Login to continue!')
return redirect(url_for('auth.login_admin'))
return render_template('auth/register_user.html',
title='Admin Registration',
form=form)
def admin_add():
username = get_username()
form=RegistAdminForm()
if form.validate_on_submit():
data=form.data
admin_count=Admin.query.filter_by(name=data['name']).count()
if admin_count!=0:
flash('管理员姓名重复','err')
return redirect(url_for('admin.admin_add'))
admin = Admin(
name=data['name'],
pwd=generate_password_hash(data['pwd']),
# is_super=1为普通管理员,=0为超级
is_super=1,
role_id=int(data['role_id']),
)
db.session.add(admin)
oplog = Oplog(
ip=request.remote_addr,
admin_id=session["admin_id"],
reson='添加管理员' + '<' + admin.name + '>',
)
db.session.add(oplog)
db.session.commit()
context = {
'username': username
}
return render_template("admin/admin_add.html",context=context,form=form)
def admin_add():
form = AdminForm()
if form.validate_on_submit():
data = form.data
if Admin.query.filter_by(name=data["name"]).count() == 1:
flash('管理员"{0}"已经存在！'.format(data["name"]), "err")
return redirect(url_for("admin.admin_add"))
from werkzeug.security import generate_password_hash
admin = Admin(
name=data["name"],
pwd=generate_password_hash(data["pwd"]),
is_super=1,
role_id=data["role_id"],
)
db.session.add(admin)
oplog = OpLog(
admin_id=session["admin_id"],
ip=request.remote_addr,
reason='添加管理员"{0}"'.format(data["name"]),
)
db.session.add(oplog)
db.session.commit()
flash('管理员"{0}"添加成功！'.format(data["name"]), "ok")
return redirect(url_for("admin.admin_add"))
return render_template("admin/admin_add.html", form=form)
def admin_add():
form = AdminForm()
from werkzeug.security import generate_password_hash
if form.validate_on_submit():
data = form.data
admin = Admin.query.filter_by(name=data['name']).count()
if admin != 0:
flash('管理员名称已经存在', 'err')
return redirect(url_for('admin.auth_add'))
else:
admin = Admin(name=data['name'],
pwd=generate_password_hash(data['pwd']),
role_id=data['role_id'],
is_super=1)
db.session.add(admin)
db.session.commit()
flash('权限管理员成功', 'ok')
oplog = Oplog(admin_id=session['admin_id'],
ip=request.remote_addr,
reason='添加管理员:%s' % data['name'])
db.session.add(oplog)
db.session.commit()
return redirect(url_for('admin.admin_add'))
return render_template('admin/admin_add.html', form=form)
def admin_add():
form = AdminForm()
if form.validate_on_submit():
data = form.data
admin_count = Admin.query.filter_by(name=data["name"]).count()
if admin_count:
flash("管理员已存在", "Error")
return redirect(url_for("admin.admin_list", page=1))
admin = Admin(name=data["name"],
pwd=generate_password_hash(data["pwd"]),
role_id=data["role_id"],
is_super="1")
db.session.add(admin)
db.session.commit()
flash("添加管理员成功", "OK")
# 添加操作日志
operatelog = OperateLog(admin_id=session["admin_id"],
ip=request.remote_addr,
reason="添加管理员：" + data["name"])
db.session.add(operatelog)
db.session.commit()
return redirect(url_for("admin.admin_list", page=1))
return render_template("admin/admin_add.html", form=form)
def admin_add():
form = AdminForm()
# 需要实时查询角色信息.zhanghai.
form.role_id.choices = [(0, "未选择")] + [(v.id, v.name)
for v in Role.query.all()]
if form.validate_on_submit():
data = form.data
from werkzeug.security import generate_password_hash
admin = Admin(name=data["name"],
pwd=generate_password_hash(data["pwd"]),
role_id=data["role_id"],
is_super=1)
db.session.add(admin)
db.session.commit()
flash("添加管理员成功！", "ok")
oplog = Oplog(admin_id=session["admin_id"],
ip=request.remote_addr,
reason="添加新管理员：%s" % data["name"])
db.session.add(oplog)
db.session.commit()
return redirect(url_for("admin.admin_add"))
# role_list = Role.query.order_by(Role.addtime.desc()).all()
return render_template("admin/admin_add.html",
form=form,
role_list=role_list)
from app.models import Source, Admin, Settings
from config import Config
db.create_all()
# Add term sources in the db
existing_sources = Source.query.all()
if len(existing_sources) < len(Config.USER_SOURCES + Config.EVENT_SOURCES):
for source in Config.EVENT_SOURCES:
s = Source(type=source)
db.session.add(s)
for source in Config.USER_SOURCES:
s = Source(type=source)
db.session.add(s)
# add administrator in the db
existing_admins = Admin.query.all()
if len(existing_admins) < 1:
a = Admin()
a.password = Config.ADMIN_PASSWORD
db.session.add(a)
# add settings in the db
existing_settings = Settings.query.all()
if len(existing_settings) < 1:
s = Settings(penalized_correlations_on=True)
db.session.add(s)
db.session.commit()
