sql-assistant 1.0.0__py3-none-any.whl

sql_assistant/web/static/css/theme.css

@@ -0,0 +1,200 @@
+/* === Theme Variables === */
+:root,
+[data-theme="dark"] {
+    --bg-primary: #0d1117;
+    --bg-secondary: #161b22;
+    --bg-tertiary: #21262d;
+    --bg-hover: #30363d;
+    --border-color: #30363d;
+    --text-primary: #e6edf3;
+    --text-secondary: #8b949e;
+    --text-muted: #6e7681;
+    --accent: #58a6ff;
+    --accent-hover: #79c0ff;
+    --success: #3fb950;
+    --warning: #d29922;
+    --danger: #f85149;
+    --font-mono: '宋体', 'SimSun', 'Times New Roman', 'Fira Code', 'Consolas', monospace;
+    --font-sans: '宋体', 'SimSun', 'Times New Roman', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+    --radius: 8px;
+    --radius-sm: 6px;
+    --transition: 150ms ease;
+    --shadow: 0 4px 12px rgba(0,0,0,0.3);
+
+    --hl-keyword: #ff7b72;
+    --hl-string: #a5d6ff;
+    --hl-number: #79c0ff;
+    --hl-comment: #8b949e;
+    --hl-function: #d2a8ff;
+    --hl-operator: #ff7b72;
+    --hl-built_in: #ffa657;
+    --hl-type: #79c0ff;
+    --hl-params: #e6edf3;
+    --hl-literal: #79c0ff;
+}
+
+[data-theme="light"] {
+    --bg-primary: #ffffff;
+    --bg-secondary: #f6f8fa;
+    --bg-tertiary: #eaeef2;
+    --bg-hover: #d0d7de;
+    --border-color: #d0d7de;
+    --text-primary: #1f2328;
+    --text-secondary: #656d76;
+    --text-muted: #8b949e;
+    --accent: #0969da;
+    --accent-hover: #0550ae;
+    --success: #1a7f37;
+    --warning: #9a6700;
+    --danger: #cf222e;
+    --font-mono: '宋体', 'SimSun', 'Times New Roman', 'Fira Code', 'Consolas', monospace;
+    --font-sans: '宋体', 'SimSun', 'Times New Roman', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+    --shadow: 0 4px 12px rgba(0,0,0,0.1);
+
+    --hl-keyword: #cf222e;
+    --hl-string: #0a3069;
+    --hl-number: #0550ae;
+    --hl-comment: #6e7781;
+    --hl-function: #8250df;
+    --hl-operator: #cf222e;
+    --hl-built_in: #953800;
+    --hl-type: #0550ae;
+    --hl-params: #1f2328;
+    --hl-literal: #0550ae;
+}
+
+/* === Header Actions === */
+.header-actions {
+    display: flex;
+    gap: 8px;
+}
+
+/* === Theme Section === */
+.theme-section {
+    margin-top: 20px;
+    padding: 16px;
+    background: var(--bg-tertiary);
+    border-radius: var(--radius);
+}
+
+.theme-section h4 {
+    font-size: 14px;
+    margin-bottom: 12px;
+    color: var(--text-primary);
+}
+
+.theme-toggle {
+    display: flex;
+    align-items: center;
+    gap: 12px;
+    margin-bottom: 16px;
+}
+
+.theme-toggle span {
+    font-size: 13px;
+    color: var(--text-secondary);
+}
+
+.switch {
+    position: relative;
+    display: inline-block;
+    width: 44px;
+    height: 24px;
+}
+
+.switch input {
+    opacity: 0;
+    width: 0;
+    height: 0;
+}
+
+.slider {
+    position: absolute;
+    cursor: pointer;
+    top: 0;
+    left: 0;
+    right: 0;
+    bottom: 0;
+    background-color: var(--bg-hover);
+    transition: 0.3s;
+    border-radius: 24px;
+}
+
+.slider:before {
+    position: absolute;
+    content: "";
+    height: 18px;
+    width: 18px;
+    left: 3px;
+    bottom: 3px;
+    background-color: white;
+    transition: 0.3s;
+    border-radius: 50%;
+}
+
+input:checked + .slider {
+    background-color: var(--accent);
+}
+
+input:checked + .slider:before {
+    transform: translateX(20px);
+}
+
+/* === Color Picker === */
+.color-picker {
+    display: flex;
+    align-items: center;
+    gap: 12px;
+    flex-wrap: wrap;
+}
+
+.color-picker > span {
+    font-size: 13px;
+    color: var(--text-secondary);
+}
+
+.color-options {
+    display: flex;
+    gap: 8px;
+    flex-wrap: wrap;
+}
+
+.color-btn {
+    width: 28px;
+    height: 28px;
+    border-radius: 50%;
+    border: 2px solid transparent;
+    cursor: pointer;
+    transition: all var(--transition);
+    position: relative;
+}
+
+.color-btn:hover {
+    transform: scale(1.1);
+}
+
+.color-btn.active {
+    border-color: var(--text-primary);
+    box-shadow: 0 0 0 2px var(--bg-primary);
+}
+
+.color-btn[data-color="blue"] { background: #58a6ff; }
+.color-btn[data-color="purple"] { background: #a371f7; }
+.color-btn[data-color="pink"] { background: #f778ba; }
+.color-btn[data-color="red"] { background: #f85149; }
+.color-btn[data-color="orange"] { background: #d29922; }
+.color-btn[data-color="yellow"] { background: #d4a72c; }
+.color-btn[data-color="green"] { background: #3fb950; }
+.color-btn[data-color="cyan"] { background: #39c5cf; }
+
+.color-btn.active::after {
+    content: '✓';
+    position: absolute;
+    top: 50%;
+    left: 50%;
+    transform: translate(-50%, -50%);
+    color: white;
+    font-size: 14px;
+    font-weight: bold;
+    text-shadow: 0 1px 2px rgba(0,0,0,0.3);
+}

sql_assistant/web/static/js/api.js

@@ -0,0 +1,38 @@
+/**
+ * API Helper Module
+ */
+const API = {
+    async post(url, data) {
+        const res = await fetch(url, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(data),
+        });
+        if (!res.ok) {
+            const err = await res.json().catch(() => ({ detail: res.statusText }));
+            throw new Error(err.detail || '请求失败');
+        }
+        return res.json();
+    },
+    async get(url) {
+        const res = await fetch(url);
+        if (!res.ok) throw new Error('请求失败');
+        return res.json();
+    },
+    async del(url) {
+        const res = await fetch(url, { method: 'DELETE' });
+        if (!res.ok) throw new Error('请求失败');
+        return res.json();
+    },
+    async put(url, data) {
+        const res = await fetch(url, {
+            method: 'PUT',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(data || {}),
+        });
+        if (!res.ok) throw new Error('请求失败');
+        return res.json();
+    },
+};
+
+window.API = API;

sql_assistant/web/static/js/app.js

@@ -0,0 +1,161 @@
+/**
+ * SQL 智能助手 - 主入口文件
+ * 负责初始化和事件绑定
+ */
+
+// ============================================================
+// Event Listeners
+// ============================================================
+document.addEventListener('DOMContentLoaded', () => {
+    // 初始化主题管理器
+    ThemeManager.init();
+    ColorThemeManager.init();
+
+    // 加载流式响应设置
+    state.streamingEnabled = false;
+
+    // 加载设置和对话列表
+    loadSettings().then(() => {
+        if (state.activeDB) refreshSchema();
+    });
+    loadConversations();
+
+    // Query form
+    dom.queryForm.addEventListener('submit', (e) => {
+        e.preventDefault();
+        sendQuery(dom.queryInput.value);
+    });
+
+    dom.queryInput.addEventListener('keydown', (e) => {
+        if (e.key === 'Enter' && !e.shiftKey) {
+            e.preventDefault();
+            sendQuery(dom.queryInput.value);
+        }
+    });
+
+    // Auto-resize textarea
+    dom.queryInput.addEventListener('input', () => {
+        dom.queryInput.style.height = 'auto';
+        dom.queryInput.style.height = Math.min(dom.queryInput.scrollHeight, 120) + 'px';
+    });
+
+    // Example tags
+    document.querySelectorAll('.example-tag').forEach(tag => {
+        tag.addEventListener('click', () => {
+            const question = tag.dataset.question;
+            dom.queryInput.value = question;
+            sendQuery(question);
+        });
+    });
+
+    // Settings
+    $('btn-settings').addEventListener('click', openSettings);
+    $('btn-close-modal').addEventListener('click', closeSettings);
+    dom.settingsModal.addEventListener('click', (e) => {
+        if (e.target === dom.settingsModal) closeSettings();
+    });
+
+    // Tabs
+    document.querySelectorAll('.tab-btn').forEach(btn => {
+        btn.addEventListener('click', () => switchTab(btn.dataset.tab));
+    });
+
+    // LLM form
+    $('btn-add-llm').addEventListener('click', () => {
+        state.editingLLM = null;
+        $('llm-form-title').textContent = '添加 LLM 提供商';
+        $('llm-api-key').placeholder = 'sk-...';
+        $('llm-name').value = '';
+        $('llm-provider').value = 'deepseek';
+        $('llm-api-key').value = '';
+        $('llm-base-url').value = '';
+        updateModelSelect('deepseek');
+        updateLLMFormByProvider('deepseek');
+        dom.llmForm.style.display = 'block';
+        dom.llmForm.scrollIntoView({ behavior: 'smooth' });
+    });
+    
+    // Provider change - update model list and form fields
+    $('llm-provider').addEventListener('change', async (e) => {
+        const provider = e.target.value;
+        if (!state.llmModels[provider] || state.llmModels[provider].length === 0) {
+            await loadLLMModels(provider);
+        }
+        updateModelSelect(provider);
+        updateLLMFormByProvider(provider);
+    });
+
+    // LLM form events
+    $('btn-save-llm').addEventListener('click', saveLLM);
+    $('btn-test-llm').addEventListener('click', testNewLLM);
+
+    // DB form
+    $('btn-add-db').addEventListener('click', () => {
+        state.editingDB = null;
+        $('db-form-title').textContent = '添加数据库连接';
+        $('db-password').placeholder = '数据库密码';
+        dom.dbForm.style.display = 'block';
+        dom.dbForm.scrollIntoView({ behavior: 'smooth' });
+    });
+    $('btn-save-db').addEventListener('click', saveDB);
+    $('btn-cancel-db').addEventListener('click', cancelDBForm);
+    $('btn-test-db').addEventListener('click', testNewDB);
+
+    // New conversation
+    $('btn-new-chat').addEventListener('click', createNewConversation);
+
+    // Refresh schema button
+    const refreshBtn = $('btn-refresh-schema');
+    if (refreshBtn) {
+        refreshBtn.addEventListener('click', refreshSchema);
+    }
+
+    // Backup
+    const backupBtn = $('btn-create-backup');
+    if (backupBtn) {
+        backupBtn.addEventListener('click', createBackup);
+    }
+
+    const confirmRestoreBtn = $('btn-confirm-restore');
+    if (confirmRestoreBtn) {
+        confirmRestoreBtn.addEventListener('click', confirmRestore);
+    }
+
+    // Load backup list when backup tab is shown
+    document.querySelectorAll('.tab-btn').forEach(btn => {
+        btn.addEventListener('click', () => {
+            if (btn.dataset.tab === 'backup') {
+                loadBackupList();
+            }
+        });
+    });
+
+    // SQL 确认对话框
+    $('btn-close-sql-confirm').addEventListener('click', hideSQLConfirmDialog);
+    $('btn-cancel-sql').addEventListener('click', hideSQLConfirmDialog);
+    $('btn-confirm-sql').addEventListener('click', confirmAndExecuteSQL);
+    
+    $('sql-confirm-modal').addEventListener('click', (e) => {
+        if (e.target === $('sql-confirm-modal')) hideSQLConfirmDialog();
+    });
+    
+    $('btn-copy-confirm-sql').addEventListener('click', () => {
+        const code = document.getElementById('sql-confirm-code').textContent;
+        navigator.clipboard.writeText(code).then(() => {
+            showToast('已复制到剪贴板', 'success');
+        });
+    });
+
+    // Keyboard shortcut: Escape to close modal
+    document.addEventListener('keydown', (e) => {
+        if (e.key === 'Escape' && dom.settingsModal.classList.contains('active')) {
+            closeSettings();
+        }
+        if (e.key === 'Escape' && $('restore-modal').classList.contains('active')) {
+            closeRestoreModal();
+        }
+        if (e.key === 'Escape' && $('sql-confirm-modal').classList.contains('active')) {
+            hideSQLConfirmDialog();
+        }
+    });
+});

sql_assistant/web/static/js/backup.js

@@ -0,0 +1,216 @@
+/**
+ * Backup Module
+ */
+
+// ============================================================
+// Backup Functions
+// ============================================================
+function toggleTableSelection() {
+    const scope = $('backup-scope').value;
+    const group = $('table-selection-group');
+    if (scope === 'selected') {
+        group.style.display = 'block';
+        loadTablesForBackup();
+    } else {
+        group.style.display = 'none';
+    }
+}
+
+async function loadTablesForBackup() {
+    try {
+        const schema = await API.get('/api/schema');
+        const tables = schema.tables || [];
+        const select = $('backup-tables');
+        select.innerHTML = tables.map(t => `
+            <option value="${escapeHtml(t.name)}">${escapeHtml(t.name)}</option>
+        `).join('');
+    } catch (err) {
+        showToast('加载表列表失败: ' + err.message, 'error');
+    }
+}
+
+async function createBackup() {
+    const backupType = $('backup-type').value;
+    const scope = $('backup-scope').value;
+    const includeSchema = $('backup-include-schema').checked;
+    const includeData = $('backup-include-data').checked;
+
+    let tables = null;
+    if (scope === 'selected') {
+        const select = $('backup-tables');
+        tables = Array.from(select.selectedOptions).map(opt => opt.value);
+        if (tables.length === 0) {
+            return showToast('请至少选择一个表', 'error');
+        }
+    }
+
+    const btn = $('btn-create-backup');
+    const originalText = btn.textContent;
+    btn.textContent = '备份中...';
+    btn.disabled = true;
+
+    try {
+        showToast('正在创建备份...', 'info');
+        const result = await API.post('/api/backup', {
+            backup_type: backupType,
+            tables,
+            include_schema: includeSchema,
+            include_data: includeData
+        });
+
+        if (result.success) {
+            showToast(result.message, 'success');
+            await loadBackupList();
+        } else {
+            showToast(result.message, 'error');
+        }
+    } catch (err) {
+        showToast('备份失败: ' + err.message, 'error');
+    } finally {
+        btn.textContent = originalText;
+        btn.disabled = false;
+    }
+}
+
+async function loadBackupList() {
+    try {
+        const result = await API.get('/api/backup/list');
+        const backups = result.backups || [];
+        const list = $('backup-list');
+
+        if (backups.length === 0) {
+            list.innerHTML = '<div class="empty-state">暂无备份记录</div>';
+            return;
+        }
+
+        list.innerHTML = backups.map(b => `
+            <div class="backup-item">
+                <div class="backup-item-header">
+                    <span class="backup-item-id">${escapeHtml(b.backup_id)}</span>
+                    <span class="backup-item-type ${b.backup_type}">${b.backup_type === 'full' ? '全量备份' : '增量备份'}</span>
+                </div>
+                <div class="backup-item-info">数据库: ${escapeHtml(b.db_name || '未知')}</div>
+                <div class="backup-item-meta">
+                    <span>表数: ${b.tables.length}</span>
+                    <span>记录数: ${b.record_count}</span>
+                    <span>大小: ${formatFileSize(b.file_size)}</span>
+                    <span>${b.backup_time ? new Date(b.backup_time).toLocaleString('zh-CN') : ''}</span>
+                </div>
+                <div class="backup-item-actions">
+                    <button onclick="viewBackupDetail('${escapeHtml(b.backup_id)}')">详情</button>
+                    <button onclick="openRestoreModal('${escapeHtml(b.backup_id)}')">恢复</button>
+                    <button class="danger" onclick="deleteBackup('${escapeHtml(b.backup_id)}')">删除</button>
+                </div>
+            </div>
+        `).join('');
+    } catch (err) {
+        showToast('加载备份列表失败: ' + err.message, 'error');
+    }
+}
+
+function formatFileSize(bytes) {
+    if (bytes === 0) return '0 B';
+    const k = 1024;
+    const sizes = ['B', 'KB', 'MB', 'GB'];
+    const i = Math.floor(Math.log(bytes) / Math.log(k));
+    return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
+}
+
+async function viewBackupDetail(backupId) {
+    try {
+        const result = await API.get(`/api/backup/${encodeURIComponent(backupId)}`);
+        const info = [
+            `备份ID: ${result.backup_id}`,
+            `类型: ${result.backup_type === 'full' ? '全量备份' : '增量备份'}`,
+            `数据库: ${result.db_name}`,
+            `数据库类型: ${result.db_type}`,
+            `表数量: ${result.tables.length}`,
+            `记录数: ${result.record_count}`,
+            `大小: ${formatFileSize(result.file_size)}`,
+            `时间: ${result.backup_time ? new Date(result.backup_time).toLocaleString('zh-CN') : ''}`
+        ];
+        alert(info.join('\n'));
+    } catch (err) {
+        showToast('获取备份详情失败: ' + err.message, 'error');
+    }
+}
+
+async function deleteBackup(backupId) {
+    if (!confirm(`确定删除备份 "${backupId}" 吗？`)) return;
+
+    try {
+        await API.del(`/api/backup/${encodeURIComponent(backupId)}`);
+        showToast('备份已删除', 'success');
+        await loadBackupList();
+    } catch (err) {
+        showToast('删除备份失败: ' + err.message, 'error');
+    }
+}
+
+let currentRestoreBackupId = null;
+
+function openRestoreModal(backupId) {
+    currentRestoreBackupId = backupId;
+    $('restore-backup-info').innerHTML = `
+        <div style="font-weight: 600; margin-bottom: 8px;">备份ID: ${escapeHtml(backupId)}</div>
+        <div style="font-size: 12px; color: var(--text-secondary);">
+            恢复操作会将备份数据写入当前激活的数据库。
+        </div>
+    `;
+    $('restore-schema').checked = false;
+    $('restore-data').checked = true;
+    $('restore-warning').style.display = 'block';
+    $('restore-modal').classList.add('active');
+}
+
+function closeRestoreModal() {
+    $('restore-modal').classList.remove('active');
+    currentRestoreBackupId = null;
+}
+
+async function confirmRestore() {
+    if (!currentRestoreBackupId) return;
+
+    const restoreSchema = $('restore-schema').checked;
+    const restoreData = $('restore-data').checked;
+
+    if (!restoreData && !restoreSchema) {
+        return showToast('请至少选择一个恢复选项', 'error');
+    }
+
+    const btn = $('btn-confirm-restore');
+    const originalText = btn.textContent;
+    btn.textContent = '恢复中...';
+    btn.disabled = true;
+
+    try {
+        showToast('正在恢复备份...', 'info');
+        const result = await API.post('/api/backup/restore', {
+            backup_id: currentRestoreBackupId,
+            restore_schema: restoreSchema,
+            restore_data: restoreData
+        });
+
+        if (result.success) {
+            showToast(result.message, 'success');
+            closeRestoreModal();
+        } else {
+            showToast(result.message, 'error');
+        }
+    } catch (err) {
+        showToast('恢复失败: ' + err.message, 'error');
+    } finally {
+        btn.textContent = originalText;
+        btn.disabled = false;
+    }
+}
+
+// Expose functions
+window.toggleTableSelection = toggleTableSelection;
+window.createBackup = createBackup;
+window.loadBackupList = loadBackupList;
+window.viewBackupDetail = viewBackupDetail;
+window.deleteBackup = deleteBackup;
+window.openRestoreModal = openRestoreModal;
+window.closeRestoreModal = closeRestoreModal;
+window.confirmRestore = confirmRestore;