☰

LeetCode Copy Testcase

自动将 LeetCode 测试用例转化为 SQL 和 Pandas 语句

// ==UserScript==
// @name         LeetCode Copy Testcase
// @namespace    https://leetcode.cn/
// @version      0.2
// @description  自动将 LeetCode 测试用例转化为 SQL 和 Pandas 语句
// @author       wangxm
// @match        https://leetcode.cn/problems/*
// @match        https://leetcode-cn.com/submissions/detail/*
// @grant        none
// @license      MIT
// ==/UserScript==

(function() {
    'use strict';

    // 使用 MutationObserver 确保在页面加载完成后执行
    const observer = new MutationObserver((mutations, obs) => {
        const targetElement = document.querySelector("#details-summary");
        if (targetElement) {
            obs.disconnect();
            addButtons(targetElement);
        }
    });

    observer.observe(document, { childList: true, subtree: true });

    function addButtons(targetElement) {
        // 获取表的对象集合信息
        let createTableSQL = "";
        let insertSQL = "";
        let dropTableSQL = "";
        let pandasCode = "";

        const tableInfos = JSON.parse(pageData.submissionData.input).headers;
        const tableRows = JSON.parse(pageData.submissionData.input).rows;

        // 生成表创建和删除语句
        for (const name in tableInfos) {
            const columns = tableInfos[name].map(col => `${col} VARCHAR(200)`);
            createTableSQL += `CREATE TABLE IF NOT EXISTS ${name} (${columns.join(", ")});\n`;
            dropTableSQL += `DROP TABLE IF EXISTS ${name};\n`;

            // 生成 Pandas 代码
            const pandasColumns = tableInfos[name].map(col => `"${col}"`);
            const pandasData = tableRows[name].map(row => `[${row.map(value => value === null ? "None" : `"${value}"`).join(", ")}]`);
            pandasCode += `import pandas as pd\n${name}_df = pd.DataFrame(columns=[${pandasColumns.join(", ")}], data=[${pandasData.join(", ")}])\n\n`;
        }

        // 处理数据并生成插入语句
        for (const tableName in tableRows) {
            tableRows[tableName].forEach(row => {
                const values = row.map(value => value === null ? "NULL" : `"${value}"`);
                insertSQL += `INSERT INTO ${tableName} VALUES (${values.join(", ")});\n`;
            });
        }

        // 添加到页面上
        targetElement.insertAdjacentHTML('beforeend', `
            <input type="button" value="复制 SQL" id="copy-sql" class="btn btn-primary">
            <input type="button" value="复制 Pandas" id="copy-pandas" class="btn btn-primary" style="margin-left:10px">
            <input type="button" value="隐藏" id="toggle" style="margin-left:10px" class="btn btn-primary">
            <textarea style="margin-top:10px;height:200px;display:block" id="sql" class="form-control"></textarea>
            <textarea style="margin-top:10px;height:200px;display:none" id="pandas" class="form-control"></textarea>
        `);

        document.getElementById("sql").value = dropTableSQL + createTableSQL + insertSQL;
        document.getElementById("pandas").value = pandasCode;

        // 添加滑动及其处理
        document.getElementById("toggle").addEventListener('click', function() {
            const sqlTextarea = document.getElementById("sql");
            const pandasTextarea = document.getElementById("pandas");
            const isHidden = sqlTextarea.style.display === "none";
            sqlTextarea.style.display = isHidden ? "block" : "none";
            pandasTextarea.style.display = isHidden ? "none" : "block";
            this.value = isHidden ? "隐藏" : "查看";
        });

        document.getElementById("copy-sql").addEventListener('click', function() {
            const sqlTextarea = document.getElementById("sql");
            sqlTextarea.select();
            document.execCommand("Copy");
            alert("SQL 已复制好，可贴粘。");
        });

        document.getElementById("copy-pandas").addEventListener('click', function() {
            const pandasTextarea = document.getElementById("pandas");
            pandasTextarea.select();
            document.execCommand("Copy");
            alert("Pandas 代码已复制好，可贴粘。");
        });
    }
})();