def init_route(self):
handler_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), "handlers")
load_handlers(handler_path=handler_path, pkg_prefix="toughadmin.console.handlers")
conn = self.db()
oprs = conn.query(models.TraOperator)
for opr in oprs:
if opr.operator_type > 0:
for rule in self.db.query(models.TraOperatorRule).filter_by(operator_name=opr.operator_name):
permit.bind_opr(rule.operator_name, rule.rule_path)
elif opr.operator_type == 0: # 超级管理员授权所有
permit.bind_super(opr.operator_name)
def post(self):
form = operator_form.operator_update_form()
if not form.validates(source=self.get_params()):
rules = self.db.query(models.TraOperatorRule.rule_path).filter_by(operator_name=form.d.operator_name)
rules = [r[0] for r in rules]
return self.render("base_form.html", form=form,rules=rules)
operator = self.db.query(models.TraOperator).get(form.d.id)
if form.d.operator_pass:
operator.operator_pass = md5(form.d.operator_pass.encode()).hexdigest()
operator.operator_desc = form.d.operator_desc
operator.operator_status = form.d.operator_status
ops_log = models.TraOperateLog()
ops_log.operator_name = self.get_secure_cookie("tra_user")
ops_log.operate_ip = self.get_secure_cookie("tra_login_ip")
ops_log.operate_time = utils.get_currtime()
ops_log.operate_desc = u'修改操作员%s信息' % operator.operator_name
self.db.add(ops_log)
# update rules
self.db.query(models.TraOperatorRule).filter_by(operator_name=operator.operator_name).delete()
for path in self.get_arguments("rule_item"):
item = permit.get_route(path)
if not item: continue
rule = models.TraOperatorRule()
rule.operator_name = operator.operator_name
rule.rule_name = item['name']
rule.rule_path = item['path']
rule.rule_category = item['category']
self.db.add(rule)
permit.unbind_opr(operator.operator_name)
self.db.commit()
for rule in self.db.query(models.TraOperatorRule).filter_by(operator_name=operator.operator_name):
permit.bind_opr(rule.operator_name, rule.rule_path)
self.redirect("/operator",permanent=False)
def post(self):
uname = self.get_argument("username")
upass = self.get_argument("password")
if not uname:
return self.render_json(code=1, msg=u"请填写用户名")
if not upass:
return self.render_json(code=1, msg=u"请填写密码")
enpasswd = md5(upass.encode()).hexdigest()
opr = self.db.query(models.TraOperator).filter_by(
operator_name=uname,
operator_pass=enpasswd
).first()
if not opr:
return self.render_json(code=1, msg=u"用户名密码不符")
if opr.operator_status == 1:
return self.render_json(code=1, msg=u"该操作员账号已被停用")
self.set_secure_cookie("tra_user", uname, expires_days=None)
self.set_secure_cookie("tra_login_time", utils.get_currtime(), expires_days=None)
self.set_secure_cookie("tra_login_ip", self.request.remote_ip, expires_days=None)
self.set_secure_cookie("tra_opr_type", str(opr.operator_type), expires_days=None)
if opr.operator_type == 1:
for rule in self.db.query(models.TraOperatorRule).filter_by(operator_name=uname):
permit.bind_opr(rule.operator_name, rule.rule_path)
ops_log = models.TraOperateLog()
ops_log.operator_name = uname
ops_log.operate_ip = self.request.remote_ip
ops_log.operate_time = utils.get_currtime()
ops_log.operate_desc = u'操作员(%s)登陆' % (uname,)
self.db.add(ops_log)
self.db.commit()
self.render_json(code=0, msg="ok")
def post(self):
form = operator_form.operator_add_form()
if not form.validates(source=self.get_params()):
return self.render("base_form.html", form=form)
if self.db.query(models.TraOperator.id).filter_by(operator_name=form.d.operator_name).count() > 0:
return self.render("base_form.html", form=form, msg=u"操作员已经存在")
operator = models.TraOperator()
operator.operator_name = form.d.operator_name
operator.operator_pass = md5(form.d.operator_pass.encode()).hexdigest()
operator.operator_type = 1
operator.operator_desc = form.d.operator_desc
operator.operator_status = form.d.operator_status
self.db.add(operator)
ops_log = models.TraOperateLog()
ops_log.operator_name = self.current_user.username
ops_log.operate_ip = self.current_user.ipaddr
ops_log.operate_time = utils.get_currtime()
ops_log.operate_desc = u'新增操作员信息:%s' % operator.operator_name
self.db.add(ops_log)
for path in self.get_arguments("rule_item"):
item = permit.get_route(path)
if not item: continue
rule = models.TraOperatorRule()
rule.operator_name = operator.operator_name
rule.rule_name = item['name']
rule.rule_path = item['path']
rule.rule_category = item['category']
self.db.add(rule)
self.db.commit()
for rule in self.db.query(models.TraOperatorRule).filter_by(operator_name=operator.operator_name):
permit.bind_opr(rule.operator_name, rule.rule_path)
self.redirect("/operator",permanent=False)
