☰

LeetCodeCopyTestcase2

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

Version vom 05.11.2024. Aktuellste Version

// ==UserScript==
// @name         LeetCodeCopyTestcase2
// @namespace    https://leetcode.cn/
// @version      0.1
// @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';

    // 获取表的对象集合信息
    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`;
        });
    }

    // 添加到页面上
    $("#details-summary").append(`
        <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>
    `);

    $("#sql").val(dropTableSQL + createTableSQL + insertSQL);
    $("#pandas").val(pandasCode);

    // 添加滑动及其处理
    $("#toggle").click(function() {
        const isHidden = $("#sql").css("display") === "none";
        $("#sql").css("display", isHidden ? "block" : "none");
        $("#pandas").css("display", isHidden ? "none" : "block");
        $(this).val(isHidden ? "隐藏" : "查看");
    });

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

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