Squashed commit of the following:

commit 948d94b6fe7472624f664d7a4cb8e3f0899af77f
Author: Julian Freeman <freemanjulian372@gmail.com>
Date:   Tue Apr 2 10:21:08 2024 -0400

    v1.3.0

.gitignore

@@ -157,4 +157,4 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
+.idea/

README.md

@@ -1 +1,9 @@
-# IdQuery
+# IdQuery
+
+## v1.3.0
+
+基本功能：
+1. ID 查小组
+2. 小组查 ID
+3. 插入数据
+4. 插入数据后显示不重复数据

idquery.qrc

@@ -0,0 +1,6 @@
+<RCC>
+    <qresource prefix="/">
+        <file>idquery.ico</file>
+        <file>idquery_16.png</file>
+    </qresource>
+</RCC>

iqh_func.py

@@ -0,0 +1,62 @@
+# coding: utf8
+import os
+import sqlite3
+from pathlib import Path
+
+
+__all__ = ["iqh_init", "iqh_create_conn", "iqh_close_conn",
+           "iqh_create_table", "iqh_select_id2group",
+           "iqh_select_group2id", "iqh_insert_data"]
+
+
+def iqh_init() -> Path:
+    base_path = Path(os.path.expandvars("%appdata%"), "JnPrograms", "IdQuery")
+    if not base_path.exists():
+        base_path.mkdir(parents=True, exist_ok=True)
+
+    return base_path
+
+
+def iqh_create_conn(base_path: Path, db_name: str) -> sqlite3.Connection:
+    db_path = Path(base_path, db_name)
+    conn = sqlite3.connect(db_path)
+    return conn
+
+
+def iqh_close_conn(conn: sqlite3.Connection):
+    conn.close()
+
+
+def iqh_create_table(conn: sqlite3.Connection):
+    cursor = conn.cursor()
+    cursor.execute("""CREATE TABLE IF NOT EXISTS idrepo (
+        id_index INTEGER PRIMARY KEY NOT NULL,
+        id_num INTEGER NOT NULL,
+        group_name TEXT
+    )""")
+    cursor.close()
+
+
+def iqh_select_id2group(conn: sqlite3.Connection, id_list: list[int]) -> list[tuple[int, str]]:
+    cursor = conn.cursor()
+    ids = ",".join(map(str, id_list))
+    cursor.execute(f"SELECT id_num, group_name FROM idrepo WHERE id_num IN ({ids})")
+    result = cursor.fetchall()
+    cursor.close()
+    return result
+
+
+def iqh_select_group2id(conn: sqlite3.Connection, group_name: str) -> list[tuple[int, str]]:
+    cursor = conn.cursor()
+    cursor.execute(f"SELECT id_num, group_name FROM idrepo WHERE group_name = '{group_name}'")
+    result = cursor.fetchall()
+    cursor.close()
+    return result
+
+
+def iqh_insert_data(conn: sqlite3.Connection, data: list[tuple[int, str]]):
+    cursor = conn.cursor()
+    values = ",".join(map(str, data))
+    cursor.execute(f"INSERT INTO idrepo (id_num, group_name) VALUES {values}")
+    conn.commit()
+    cursor.close()

mw_idq.py

@@ -0,0 +1,246 @@
+# coding: utf8
+import sys
+from PySide6 import QtWidgets, QtCore, QtGui
+from iqh_func import *
+
+import idquery_rc
+
+
+version = [1, 3, 0, 20240402]
+
+
+def accept_warning(widget: QtWidgets.QWidget, condition: bool,
+                   caption: str = "警告", text: str = "确定要继续吗？") -> bool:
+    if condition:
+        b = QtWidgets.QMessageBox.question(widget, caption, text)
+        if b == QtWidgets.QMessageBox.StandardButton.No:
+            return True
+    return False
+
+
+class UiMainWindow(object):
+
+    def __init__(self, window: QtWidgets.QMainWindow):
+        self.sb = window.statusBar()
+        self.mb = window.menuBar()
+        window.setWindowTitle("ID 查询库")
+        window.setWindowIcon(QtGui.QIcon(":/idquery_16.png"))
+        window.resize(600, 700)
+        self.cw = QtWidgets.QWidget(window)
+        window.setCentralWidget(self.cw)
+        self.vly_cw = QtWidgets.QVBoxLayout()
+        self.cw.setLayout(self.vly_cw)
+
+        self.hly_buttons = QtWidgets.QHBoxLayout()
+        self.pbn_insert = QtWidgets.QPushButton("插入", self.cw)
+        self.pbn_select_id2group = QtWidgets.QPushButton("ID查小组", self.cw)
+        self.pbn_select_group2id = QtWidgets.QPushButton("小组查ID", self.cw)
+        self.hly_buttons.addWidget(self.pbn_insert)
+        self.hly_buttons.addWidget(self.pbn_select_id2group)
+        self.hly_buttons.addWidget(self.pbn_select_group2id)
+        self.hly_buttons.addStretch(1)
+        self.pbn_clear = QtWidgets.QPushButton("清空两栏", self.cw)
+        self.hly_buttons.addWidget(self.pbn_clear)
+
+        self.hly_quantity = QtWidgets.QHBoxLayout()
+        self.lne_id_qu = QtWidgets.QLineEdit(self.cw)
+        self.lne_group_qu = QtWidgets.QLineEdit(self.cw)
+        self.lne_id_qu.setDisabled(True)
+        self.lne_group_qu.setDisabled(True)
+        self.hly_quantity.addWidget(self.lne_id_qu)
+        self.hly_quantity.addWidget(self.lne_group_qu)
+
+        self.hly_body = QtWidgets.QHBoxLayout()
+        self.pte_ids = QtWidgets.QPlainTextEdit(self.cw)
+        self.pte_groups = QtWidgets.QPlainTextEdit(self.cw)
+        self.hly_body.addWidget(self.pte_ids)
+        self.hly_body.addWidget(self.pte_groups)
+
+        self.vly_cw.addLayout(self.hly_buttons)
+        self.vly_cw.addLayout(self.hly_quantity)
+        self.vly_cw.addLayout(self.hly_body)
+
+        self.menu_file = self.mb.addMenu("文件")
+        self.act_read_csv = QtGui.QAction("读取 CSV 文件", self.cw)
+        self.menu_file.addAction(self.act_read_csv)
+
+        self.menu_about = self.mb.addMenu("关于")
+        self.act_about = QtGui.QAction("关于 ID 查询库", self.cw)
+        self.menu_about.addAction(self.act_about)
+
+
+class MainWindow(QtWidgets.QMainWindow):
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.ui = UiMainWindow(self)
+
+        self.base_path = iqh_init()
+        self.db_name = "idquery.db"
+        self.db_conn = iqh_create_conn(self.base_path, self.db_name)
+        iqh_create_table(self.db_conn)
+
+        self.ui.pte_ids.textChanged.connect(self.on_pte_ids_text_changed)
+        self.ui.pte_groups.textChanged.connect(self.on_pte_groups_text_changed)
+        self.ui.pbn_clear.clicked.connect(self.on_pbn_clear_clicked)
+        self.ui.act_about.triggered.connect(self.on_act_about_triggered)
+
+        self.ui.pbn_select_id2group.clicked.connect(self.on_pbn_select_id2group_clicked)
+        self.ui.pbn_select_group2id.clicked.connect(self.on_pbn_select_group2id_clicked)
+        self.ui.pbn_insert.clicked.connect(self.on_pbn_insert_clicked)
+        self.ui.act_read_csv.triggered.connect(self.on_act_read_csv_triggered)
+
+    def __del__(self):
+        iqh_close_conn(self.db_conn)
+
+    def on_pte_ids_text_changed(self):
+        ids = self.ui.pte_ids.toPlainText().strip()
+        id_ls = [i.strip() for i in ids.split("\n") if len(i.strip()) != 0]
+        self.ui.lne_id_qu.setText(f"ID 数量：{len(id_ls)}")
+
+    def on_pte_groups_text_changed(self):
+        groups = self.ui.pte_groups.toPlainText().strip()
+        group_ls = [g.strip() for g in groups.split("\n") if len(g.strip()) != 0]
+        self.ui.lne_group_qu.setText(f"小组数量：{len(group_ls)}")
+
+    def on_pbn_clear_clicked(self):
+        self.ui.pte_ids.clear()
+        self.ui.pte_groups.clear()
+
+    def on_act_about_triggered(self):
+        ver = f"{version[0]}.{version[1]}.{version[2]}，于 {version[-1]}"
+        QtWidgets.QMessageBox.about(
+            self, "关于",
+            f"版本：{ver}\n数据库路径：\n{str(self.base_path / self.db_name)}"
+        )
+
+    def on_pbn_select_id2group_clicked(self):
+        raw_ids = self.ui.pte_ids.toPlainText().strip()
+        if len(raw_ids) == 0:
+            msg = "没有找到有效的 ID"
+            self.ui.sb.showMessage(msg)
+            QtWidgets.QMessageBox.critical(self, "错误", msg)
+            return
+        raw_groups = self.ui.pte_groups.toPlainText().strip()
+        if accept_warning(self, len(raw_groups) != 0, text="小组栏不为空，确定要覆盖吗？"):
+            self.ui.sb.showMessage("请先清空小组栏。")
+            return
+
+        id_list = raw_ids.split("\n")
+        q_ids = [int(x.strip()) for x in id_list if len(x) != 0 and x.strip().isdigit()]
+        result = iqh_select_id2group(self.db_conn, q_ids)
+        result_d = {str(id_num): group_name for id_num, group_name in result}
+        res_groups = []
+        for r in id_list:
+            r = r.strip()
+            if r in result_d:
+                res_groups.append(result_d[r])
+            else:
+                res_groups.append("")
+
+        self.ui.pte_groups.setPlainText("\n".join(res_groups))
+
+        msg = f"共找到 {len(result)} 条记录。"
+        self.ui.sb.showMessage(msg)
+
+    def on_pbn_select_group2id_clicked(self):
+        raw_groups = self.ui.pte_groups.toPlainText().strip()
+        if len(raw_groups) == 0:
+            msg = "没有找到有效的小组"
+            self.ui.sb.showMessage(msg)
+            QtWidgets.QMessageBox.critical(self, "错误", msg)
+            return
+        raw_ids = self.ui.pte_ids.toPlainText().strip()
+        if accept_warning(self, len(raw_ids) != 0, text="ID 栏不为空，确定要覆盖吗？"):
+            self.ui.sb.showMessage("请先清空 ID 栏。")
+            return
+
+        group_list = raw_groups.split("\n")
+        group = group_list[0].strip()  # 只找第一个小组，后面的忽略
+        result = iqh_select_group2id(self.db_conn, group)
+        res_ids = [str(id_num) for id_num, _ in result]
+
+        self.ui.pte_ids.setPlainText("\n".join(res_ids))
+
+        msg = f"共找到 {len(result)} 条记录。"
+        self.ui.sb.showMessage(msg)
+
+    def on_pbn_insert_clicked(self):
+        raw_ids = self.ui.pte_ids.toPlainText().strip()
+        raw_groups = self.ui.pte_groups.toPlainText().strip()
+        id_list = raw_ids.split("\n")
+        group_list = raw_groups.split("\n")
+        if accept_warning(self, len(id_list) != len(group_list), text="ID 数量和小组数量不一致，要自动填充小组吗？"):
+            self.ui.sb.showMessage("ID 数量和小组数量不一致，未进行自动填充。")
+            return
+
+        # 填充并制作数据
+        last_group = ""
+        len_group_list = len(group_list)
+        data = []
+        new_ids = []
+        for i in range(len(id_list)):
+            ids = id_list[i].strip()
+            if not (len(ids) != 0 and ids.isdigit()):
+                continue
+            idi = int(ids)
+            if i < len_group_list:
+                group = group_list[i]
+                last_group = group
+            else:
+                group = last_group
+
+            data.append((idi, group))
+            new_ids.append(idi)
+
+        # 排重
+        result = iqh_select_id2group(self.db_conn, new_ids)
+        res_ids = [i for i, _ in result]
+        new_data = [(i, g) for i, g in data if i not in res_ids]
+
+        self.ui.pte_ids.setPlainText("\n".join([str(i) for i, _ in new_data]))
+        self.ui.pte_groups.setPlainText("\n".join([g for _, g in new_data]))
+
+        if len(new_data) != 0:
+            iqh_insert_data(self.db_conn, new_data)
+
+        msg = f"一共找到 {len(data)} 个 ID，成功插入 {len(new_data)} 个。"
+        self.ui.sb.showMessage(msg)
+        QtWidgets.QMessageBox.information(self, "信息", msg)
+
+    def on_act_read_csv_triggered(self):
+        if accept_warning(self, len(self.ui.pte_ids.toPlainText().strip()) != 0, text="ID 栏还有内容，确定继续覆盖吗？"):
+            self.ui.sb.showMessage("请先清空 ID 栏。")
+            return
+
+        filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "打开 CSV 文件", "..",
+                                                            filter="CSV 文件 (*.csv);;所有文件(*)")
+        if len(filename) == 0:
+            return
+
+        with open(filename, "r", encoding="utf8") as f:
+            lines = f.read()
+
+        id_lines = []
+        group_lines = []
+        for line in lines.strip().split("\n"):
+            ids, group = line.split(",", 1)  # type: str, str
+            ids = ids.strip()
+            if not (len(ids) != 0 and ids.isdigit()):
+                continue
+            id_lines.append(ids)
+            group_lines.append(group)
+
+        self.ui.pte_ids.setPlainText("\n".join(id_lines))
+        self.ui.pte_groups.setPlainText("\n".join(group_lines))
+
+
+def main():
+    app = QtWidgets.QApplication(sys.argv)
+    win = MainWindow()
+    win.show()
+    return app.exec()
+
+
+if __name__ == '__main__':
+    sys.exit(main())