mirror of https://github.com/tp4a/teleport
111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import json
|
|
from app.const import *
|
|
from app.base.db import get_db, SQL
|
|
from app.base.logger import log
|
|
from app.base.utils import tp_timestamp_utc_now
|
|
from . import syslog
|
|
|
|
|
|
def save_config(handler, msg, name, value):
|
|
db = get_db()
|
|
|
|
str_val = json.dumps(value, separators=(',', ':'))
|
|
|
|
sql = 'SELECT name FROM `{dbtp}config` WHERE name="{name}";'.format(dbtp=db.table_prefix, name=name)
|
|
db_ret = db.query(sql)
|
|
if db_ret is not None and len(db_ret) > 0:
|
|
sql = 'UPDATE `{dbtp}config` SET value={dbph} WHERE name="{name}";'.format(dbtp=db.table_prefix, dbph=db.place_holder, name=name)
|
|
db_ret = db.exec(sql, (str_val,))
|
|
else:
|
|
sql = 'INSERT INTO `{dbtp}config` (name, value) VALUES ({dbph}, {dbph});'.format(dbtp=db.table_prefix, dbph=db.place_holder)
|
|
db_ret = db.exec(sql, (name, str_val))
|
|
|
|
if not db_ret:
|
|
return TPE_DATABASE
|
|
|
|
operator = handler.get_current_user()
|
|
syslog.sys_log(operator, handler.request.remote_ip, TPE_OK, msg)
|
|
|
|
return TPE_OK
|
|
|
|
|
|
def add_role(handler, role_name, privilege):
|
|
db = get_db()
|
|
_time_now = tp_timestamp_utc_now()
|
|
operator = handler.get_current_user()
|
|
|
|
# 1. 判断是否已经存在了
|
|
sql = 'SELECT id FROM {}role WHERE name="{name}";'.format(db.table_prefix, name=role_name)
|
|
db_ret = db.query(sql)
|
|
if db_ret is not None and len(db_ret) > 0:
|
|
return TPE_EXISTS, 0
|
|
|
|
sql = 'INSERT INTO `{}role` (name, privilege, creator_id, create_time) VALUES ' \
|
|
'("{name}", {privilege}, {creator_id}, {create_time});' \
|
|
''.format(db.table_prefix, name=role_name, privilege=privilege, creator_id=operator['id'], create_time=_time_now)
|
|
db_ret = db.exec(sql)
|
|
if not db_ret:
|
|
return TPE_DATABASE, 0
|
|
|
|
_id = db.last_insert_id()
|
|
|
|
syslog.sys_log(operator, handler.request.remote_ip, TPE_OK, "创建角色：{}".format(role_name))
|
|
|
|
return TPE_OK, _id
|
|
|
|
|
|
def update_role(handler, role_id, role_name, privilege):
|
|
"""
|
|
更新一个远程账号
|
|
"""
|
|
db = get_db()
|
|
|
|
# 1. 判断是否存在
|
|
sql = 'SELECT id FROM {}role WHERE id={rid};'.format(db.table_prefix, rid=role_id)
|
|
db_ret = db.query(sql)
|
|
if db_ret is None or len(db_ret) == 0:
|
|
return TPE_NOT_EXISTS
|
|
|
|
sql = 'UPDATE `{}role` SET name="{name}", privilege={p} WHERE id={rid};' \
|
|
''.format(db.table_prefix, name=role_name, p=privilege, rid=role_id)
|
|
|
|
db_ret = db.exec(sql)
|
|
if not db_ret:
|
|
return TPE_DATABASE
|
|
|
|
return TPE_OK
|
|
|
|
|
|
def remove_role(handler, role_id):
|
|
db = get_db()
|
|
|
|
s = SQL(db)
|
|
# 1. 判断是否存在
|
|
s.select_from('role', ['name'], alt_name='r')
|
|
s.where('r.id={rid}'.format(rid=role_id))
|
|
err = s.query()
|
|
if err != TPE_OK:
|
|
return err
|
|
if len(s.recorder) == 0:
|
|
return TPE_NOT_EXISTS
|
|
|
|
role_name = s.recorder[0].name
|
|
|
|
sql_list = []
|
|
|
|
sql = 'DELETE FROM `{}role` WHERE id={};'.format(db.table_prefix, role_id)
|
|
sql_list.append(sql)
|
|
|
|
# 更新此角色相关的用户信息
|
|
sql = 'UPDATE `{}user` SET role_id=0 WHERE role_id={rid};'.format(db.table_prefix, rid=role_id)
|
|
sql_list.append(sql)
|
|
|
|
if not db.transaction(sql_list):
|
|
return TPE_DATABASE
|
|
|
|
syslog.sys_log(handler.get_current_user(), handler.request.remote_ip, TPE_OK, "删除角色：{}".format(role_name))
|
|
|
|
return TPE_OK
|