mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/sky22333/ansible-ui
synced 2026-09-20 08:03:57 +08:00
更新其他功能适配密钥认证
This commit is contained in:
user123456
466
database.py
466
database.py
@@ -1,235 +1,233 @@
|
||||
import sqlite3
|
||||
from contextlib import contextmanager
|
||||
import os
|
||||
from crypto_utils import CryptoUtils
|
||||
|
||||
class Database:
|
||||
def __init__(self, db_path="db/ansible.db"):
|
||||
self.db_path = db_path
|
||||
self.crypto = CryptoUtils()
|
||||
|
||||
# 判断父目录是否存在，不存在则创建
|
||||
db_dir = os.path.dirname(self.db_path)
|
||||
if not os.path.exists(db_dir):
|
||||
os.makedirs(db_dir, exist_ok=True)
|
||||
|
||||
self.init_database()
|
||||
|
||||
def init_database(self):
|
||||
"""初始化数据库表"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS hosts (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
comment TEXT NOT NULL,
|
||||
address TEXT NOT NULL,
|
||||
username TEXT NOT NULL,
|
||||
port INTEGER NOT NULL,
|
||||
password TEXT NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
""")
|
||||
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS command_logs (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
host_id INTEGER,
|
||||
command TEXT NOT NULL,
|
||||
output TEXT,
|
||||
status TEXT NOT NULL,
|
||||
executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
FOREIGN KEY (host_id) REFERENCES hosts (id)
|
||||
)
|
||||
""")
|
||||
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS access_logs (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
ip_address TEXT NOT NULL,
|
||||
path TEXT NOT NULL,
|
||||
status TEXT NOT NULL,
|
||||
status_code INTEGER NOT NULL,
|
||||
access_time TIMESTAMP DEFAULT (datetime('now', '+8 hours'))
|
||||
)
|
||||
""")
|
||||
|
||||
def init_users_table(self):
|
||||
"""初始化用户表"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
username TEXT NOT NULL,
|
||||
password TEXT NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
""")
|
||||
|
||||
@contextmanager
|
||||
def get_connection(self):
|
||||
"""获取数据库连接的上下文管理器"""
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
conn.row_factory = sqlite3.Row
|
||||
try:
|
||||
yield conn
|
||||
conn.commit()
|
||||
except Exception as e:
|
||||
conn.rollback()
|
||||
raise e
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def add_host(self, host_data):
|
||||
"""添加单个主机"""
|
||||
with self.get_connection() as conn:
|
||||
# 加密密码后存储
|
||||
encrypted_password = self.crypto.encrypt(host_data['password'])
|
||||
cursor = conn.execute("""
|
||||
INSERT INTO hosts (comment, address, username, port, password)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
""", (
|
||||
host_data['comment'],
|
||||
host_data['address'],
|
||||
host_data['username'],
|
||||
host_data['port'],
|
||||
encrypted_password
|
||||
))
|
||||
return cursor.lastrowid
|
||||
|
||||
def add_hosts_batch(self, hosts_data):
|
||||
"""批量添加主机"""
|
||||
with self.get_connection() as conn:
|
||||
# 加密每个主机的密码
|
||||
encrypted_hosts = []
|
||||
for host in hosts_data:
|
||||
encrypted_hosts.append((
|
||||
host['comment'],
|
||||
host['address'],
|
||||
host['username'],
|
||||
host['port'],
|
||||
self.crypto.encrypt(host['password'])
|
||||
))
|
||||
|
||||
cursor = conn.executemany("""
|
||||
INSERT INTO hosts (comment, address, username, port, password)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
""", encrypted_hosts)
|
||||
return cursor.rowcount
|
||||
|
||||
def get_hosts(self):
|
||||
"""获取所有主机"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("SELECT * FROM hosts ORDER BY created_at DESC")
|
||||
hosts = [dict(row) for row in cursor.fetchall()]
|
||||
|
||||
# 处理返回的主机数据，解密密码
|
||||
for host in hosts:
|
||||
# 保留一个加密版本的密码，供前端识别是否需要重新输入
|
||||
host['encrypted_password'] = host['password']
|
||||
# 解密密码供后端使用
|
||||
host['password'] = self.crypto.decrypt(host['password'])
|
||||
return hosts
|
||||
|
||||
def get_host(self, host_id):
|
||||
"""获取单个主机信息"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("SELECT * FROM hosts WHERE id = ?", (host_id,))
|
||||
row = cursor.fetchone()
|
||||
if row:
|
||||
host = dict(row)
|
||||
# 保留一个加密版本的密码，供前端识别是否需要重新输入
|
||||
host['encrypted_password'] = host['password']
|
||||
# 解密密码供后端使用
|
||||
host['password'] = self.crypto.decrypt(host['password'])
|
||||
return host
|
||||
return None
|
||||
|
||||
def update_host(self, host_id, host_data):
|
||||
"""更新主机信息"""
|
||||
with self.get_connection() as conn:
|
||||
if host_data.get('password'):
|
||||
# 确保更新时加密新密码
|
||||
encrypted_password = self.crypto.encrypt(host_data['password'])
|
||||
conn.execute("""
|
||||
UPDATE hosts
|
||||
SET comment = ?, address = ?, username = ?, port = ?, password = ?
|
||||
WHERE id = ?
|
||||
""", (
|
||||
host_data['comment'],
|
||||
host_data['address'],
|
||||
host_data['username'],
|
||||
host_data['port'],
|
||||
encrypted_password,
|
||||
host_id
|
||||
))
|
||||
else:
|
||||
conn.execute("""
|
||||
UPDATE hosts
|
||||
SET comment = ?, address = ?, username = ?, port = ?
|
||||
WHERE id = ?
|
||||
""", (
|
||||
host_data['comment'],
|
||||
host_data['address'],
|
||||
host_data['username'],
|
||||
host_data['port'],
|
||||
host_id
|
||||
))
|
||||
|
||||
def delete_host(self, host_id):
|
||||
"""删除主机"""
|
||||
with self.get_connection() as conn:
|
||||
# 首先删除与主机相关的命令日志
|
||||
conn.execute("DELETE FROM command_logs WHERE host_id = ?", (host_id,))
|
||||
# 然后删除主机记录
|
||||
conn.execute("DELETE FROM hosts WHERE id = ?", (host_id,))
|
||||
|
||||
def log_command(self, host_id, command, output, status):
|
||||
"""记录命令执行日志"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
INSERT INTO command_logs (host_id, command, output, status)
|
||||
VALUES (?, ?, ?, ?)
|
||||
""", (host_id, command, output, status))
|
||||
|
||||
def get_command_logs(self, limit=100):
|
||||
"""获取命令执行日志"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("""
|
||||
SELECT cl.*, h.comment, h.address
|
||||
FROM command_logs cl
|
||||
LEFT JOIN hosts h ON cl.host_id = h.id
|
||||
ORDER BY cl.executed_at DESC
|
||||
LIMIT ?
|
||||
""", (limit,))
|
||||
return [dict(row) for row in cursor.fetchall()]
|
||||
|
||||
def add_access_log(self, ip_address, path, status, status_code):
|
||||
"""添加访问日志"""
|
||||
# 注释掉内网IP过滤，确保记录所有来源IP，包括通过代理转发的
|
||||
# if (ip_address.startswith(('10.', '172.', '192.168.')) or
|
||||
# ip_address in ['127.0.0.1', 'localhost']):
|
||||
# return
|
||||
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
INSERT INTO access_logs (ip_address, path, status, status_code)
|
||||
VALUES (?, ?, ?, ?)
|
||||
""", (ip_address, path, status, status_code))
|
||||
|
||||
def get_access_logs(self, limit=100):
|
||||
"""获取访问日志"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("""
|
||||
SELECT * FROM access_logs
|
||||
ORDER BY access_time DESC
|
||||
LIMIT ?
|
||||
""", (limit,))
|
||||
return [dict(row) for row in cursor.fetchall()]
|
||||
|
||||
def cleanup_old_logs(self):
|
||||
"""清理7天前的日志（使用北京时间）"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
DELETE FROM access_logs
|
||||
WHERE access_time < datetime('now', '+8 hours', '-7 days')
|
||||
import sqlite3
|
||||
from contextlib import contextmanager
|
||||
import os
|
||||
from crypto_utils import CryptoUtils
|
||||
|
||||
class Database:
|
||||
def __init__(self, db_path="db/ansible.db"):
|
||||
self.db_path = db_path
|
||||
self.crypto = CryptoUtils()
|
||||
|
||||
# 判断父目录是否存在，不存在则创建
|
||||
db_dir = os.path.dirname(self.db_path)
|
||||
if not os.path.exists(db_dir):
|
||||
os.makedirs(db_dir, exist_ok=True)
|
||||
|
||||
self.init_database()
|
||||
|
||||
def init_database(self):
|
||||
"""初始化数据库表"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS hosts (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
comment TEXT NOT NULL,
|
||||
address TEXT NOT NULL,
|
||||
username TEXT NOT NULL,
|
||||
port INTEGER NOT NULL,
|
||||
password TEXT,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
auth_method TEXT NOT NULL DEFAULT 'password'
|
||||
)
|
||||
""")
|
||||
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS command_logs (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
host_id INTEGER,
|
||||
command TEXT NOT NULL,
|
||||
output TEXT,
|
||||
status TEXT NOT NULL,
|
||||
executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
FOREIGN KEY (host_id) REFERENCES hosts (id)
|
||||
)
|
||||
""")
|
||||
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS access_logs (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
ip_address TEXT NOT NULL,
|
||||
path TEXT NOT NULL,
|
||||
status TEXT NOT NULL,
|
||||
status_code INTEGER NOT NULL,
|
||||
access_time TIMESTAMP DEFAULT (datetime('now', '+8 hours'))
|
||||
)
|
||||
""")
|
||||
|
||||
def init_users_table(self):
|
||||
"""初始化用户表"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
username TEXT NOT NULL,
|
||||
password TEXT NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
""")
|
||||
|
||||
@contextmanager
|
||||
def get_connection(self):
|
||||
"""获取数据库连接的上下文管理器"""
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
conn.row_factory = sqlite3.Row
|
||||
try:
|
||||
yield conn
|
||||
conn.commit()
|
||||
except Exception as e:
|
||||
conn.rollback()
|
||||
raise e
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def add_host(self, host_data):
|
||||
"""添加单个主机"""
|
||||
with self.get_connection() as conn:
|
||||
encrypted_password = None
|
||||
auth_method = host_data.get('auth_method', 'password')
|
||||
|
||||
if auth_method == 'password' and host_data.get('password'):
|
||||
encrypted_password = self.crypto.encrypt(host_data['password'])
|
||||
|
||||
cursor = conn.execute("""
|
||||
INSERT INTO hosts (comment, address, username, port, password, auth_method)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
host_data['comment'],
|
||||
host_data['address'],
|
||||
host_data['username'],
|
||||
host_data['port'],
|
||||
encrypted_password,
|
||||
auth_method
|
||||
))
|
||||
return cursor.lastrowid
|
||||
|
||||
def add_hosts_batch(self, hosts_data):
|
||||
"""批量添加主机"""
|
||||
with self.get_connection() as conn:
|
||||
processed_hosts = []
|
||||
for host in hosts_data:
|
||||
encrypted_password = None
|
||||
auth_method = host.get('auth_method', 'password')
|
||||
|
||||
if auth_method == 'password' and host.get('password'):
|
||||
encrypted_password = self.crypto.encrypt(host['password'])
|
||||
|
||||
processed_hosts.append((
|
||||
host['comment'],
|
||||
host['address'],
|
||||
host['username'],
|
||||
host['port'],
|
||||
encrypted_password,
|
||||
auth_method
|
||||
))
|
||||
|
||||
cursor = conn.executemany("""
|
||||
INSERT INTO hosts (comment, address, username, port, password, auth_method)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
""", processed_hosts)
|
||||
return cursor.rowcount
|
||||
|
||||
def get_hosts(self):
|
||||
"""获取所有主机"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("SELECT * FROM hosts ORDER BY created_at DESC")
|
||||
hosts = [dict(row) for row in cursor.fetchall()]
|
||||
|
||||
for host in hosts:
|
||||
host['encrypted_password'] = host['password']
|
||||
if host['auth_method'] == 'password' and host['password']:
|
||||
host['password'] = self.crypto.decrypt(host['password'])
|
||||
else:
|
||||
host['password'] = None
|
||||
return hosts
|
||||
|
||||
def get_host(self, host_id):
|
||||
"""获取单个主机信息"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("SELECT * FROM hosts WHERE id = ?", (host_id,))
|
||||
row = cursor.fetchone()
|
||||
if row:
|
||||
host = dict(row)
|
||||
host['encrypted_password'] = host['password']
|
||||
if host['auth_method'] == 'password' and host['password']:
|
||||
host['password'] = self.crypto.decrypt(host['password'])
|
||||
else:
|
||||
host['password'] = None
|
||||
return host
|
||||
return None
|
||||
|
||||
def update_host(self, host_id, host_data):
|
||||
"""更新主机信息"""
|
||||
with self.get_connection() as conn:
|
||||
auth_method = host_data.get('auth_method', 'password')
|
||||
encrypted_password = None
|
||||
|
||||
if auth_method == 'password' and host_data.get('password'):
|
||||
encrypted_password = self.crypto.encrypt(host_data['password'])
|
||||
|
||||
conn.execute("""
|
||||
UPDATE hosts
|
||||
SET comment = ?, address = ?, username = ?, port = ?, password = ?, auth_method = ?
|
||||
WHERE id = ?
|
||||
""", (
|
||||
host_data['comment'],
|
||||
host_data['address'],
|
||||
host_data['username'],
|
||||
host_data['port'],
|
||||
encrypted_password,
|
||||
auth_method,
|
||||
host_id
|
||||
))
|
||||
|
||||
def delete_host(self, host_id):
|
||||
"""删除主机"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("DELETE FROM command_logs WHERE host_id = ?", (host_id,))
|
||||
conn.execute("DELETE FROM hosts WHERE id = ?", (host_id,))
|
||||
|
||||
def log_command(self, host_id, command, output, status):
|
||||
"""记录命令执行日志"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
INSERT INTO command_logs (host_id, command, output, status)
|
||||
VALUES (?, ?, ?, ?)
|
||||
""", (host_id, command, output, status))
|
||||
|
||||
def get_command_logs(self, limit=100):
|
||||
"""获取命令执行日志"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("""
|
||||
SELECT cl.*, h.comment, h.address
|
||||
FROM command_logs cl
|
||||
LEFT JOIN hosts h ON cl.host_id = h.id
|
||||
ORDER BY cl.executed_at DESC
|
||||
LIMIT ?
|
||||
""", (limit,))
|
||||
return [dict(row) for row in cursor.fetchall()]
|
||||
|
||||
def add_access_log(self, ip_address, path, status, status_code):
|
||||
"""添加访问日志"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
INSERT INTO access_logs (ip_address, path, status, status_code)
|
||||
VALUES (?, ?, ?, ?)
|
||||
""", (ip_address, path, status, status_code))
|
||||
|
||||
def get_access_logs(self, limit=100):
|
||||
"""获取访问日志"""
|
||||
with self.get_connection() as conn:
|
||||
cursor = conn.execute("""
|
||||
SELECT * FROM access_logs
|
||||
ORDER BY access_time DESC
|
||||
LIMIT ?
|
||||
""", (limit,))
|
||||
return [dict(row) for row in cursor.fetchall()]
|
||||
|
||||
def cleanup_old_logs(self):
|
||||
"""清理7天前的日志（使用北京时间）"""
|
||||
with self.get_connection() as conn:
|
||||
conn.execute("""
|
||||
DELETE FROM access_logs
|
||||
WHERE access_time < datetime('now', '+8 hours', '-7 days')
|
||||
""")
|
||||
Reference in New Issue
Block a user