98 lines
3.9 KiB
Python
98 lines
3.9 KiB
Python
|
import sqlite3
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
DB_PATH = Path(__file__).parent / "mcu.db"
|
|||
|
|
|
|||
|
|
def init_db():
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
cur.execute("""
|
|||
|
|
CREATE TABLE IF NOT EXISTS mcu (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT, -- ID
|
|||
|
|
name TEXT NOT NULL, -- 名称
|
|||
|
|
core_num TEXT NOT NULL, -- 核心数量
|
|||
|
|
core_type TEXT NOT NULL, -- 核心架构
|
|||
|
|
instruction_set TEXT, -- 指令集架构
|
|||
|
|
coremark TEXT, -- CoreMark 分数
|
|||
|
|
coremark_per_mhz REAL, -- CoreMark/MHz
|
|||
|
|
frequency TEXT NOT NULL, -- 主频 MHz
|
|||
|
|
flash TEXT, -- Flash KB
|
|||
|
|
rom TEXT, -- ROM KB
|
|||
|
|
ram TEXT, -- RAM KB
|
|||
|
|
sram TEXT, -- SRAM KB
|
|||
|
|
sram_in_rtc TEXT, -- RTC SRAM KB
|
|||
|
|
bus_width TEXT, -- 总线宽度 bit
|
|||
|
|
cache_type TEXT, -- Cache 类型 （L1/L2）
|
|||
|
|
l1_cache_size TEXT, -- L1 Cache 大小 KB
|
|||
|
|
l2_cache_size TEXT, -- L2 Cache 大小 KB
|
|||
|
|
l3_cache_size TEXT, -- L3 Cache 大小 KB
|
|||
|
|
pipeline_depth TEXT, -- 流水线深度
|
|||
|
|
simd_support BOOLEAN, -- SIMD 支持
|
|||
|
|
fpu TEXT, -- FPU 支持 （Single/Double Precision）
|
|||
|
|
package TEXT, -- 封装
|
|||
|
|
gpios TEXT, -- GPIO 引脚数量
|
|||
|
|
uarts TEXT, -- UART 数量
|
|||
|
|
i2cs TEXT, -- I2C 数量
|
|||
|
|
i2ses TEXT, -- I2S 数量
|
|||
|
|
spis TEXT, -- SPI 数量
|
|||
|
|
spi_protocols TEXT, -- SPI 协议
|
|||
|
|
systems TEXT, -- 系统
|
|||
|
|
manufacturer TEXT, -- 厂商
|
|||
|
|
link TEXT -- 链接
|
|||
|
|
)
|
|||
|
|
""")
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
# ==================== 查询所有 MCU（关键字段） ====================
|
|||
|
|
def get_all_mcus():
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
cur.execute("SELECT id, name, core_num, core_type, rom, ram, sram, flash, frequency, manufacturer FROM mcu")
|
|||
|
|
rows = cur.fetchall()
|
|||
|
|
conn.close()
|
|||
|
|
return [{"id": i, "name": n, "core_num": cn, "core_type": ct, "rom": ro, "ram": ra, "sram": sra, "flash":fl, "frequency": f, "manufacturer": m}
|
|||
|
|
for i, n, cn, ct, ro, ra, sra, fl, f, m in rows]
|
|||
|
|
|
|||
|
|
# ==================== 查询单个 MCU（完整字段） ====================
|
|||
|
|
def get_mcu(mcu_id):
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
conn.row_factory = sqlite3.Row # 允许返回 dict
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
cur.execute("SELECT * FROM mcu WHERE id=?", (mcu_id,))
|
|||
|
|
row = cur.fetchone()
|
|||
|
|
conn.close()
|
|||
|
|
return dict(row) if row else None
|
|||
|
|
|
|||
|
|
# ==================== 插入 MCU（支持所有字段） ====================
|
|||
|
|
def add_mcu(**fields):
|
|||
|
|
"""动态插入 MCU（支持所有字段）"""
|
|||
|
|
keys = ",".join(fields.keys())
|
|||
|
|
placeholders = ",".join("?" for _ in fields)
|
|||
|
|
values = tuple(fields.values())
|
|||
|
|
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
cur.execute(f"INSERT INTO mcu ({keys}) VALUES ({placeholders})", values)
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
# ==================== 更新 MCU（支持所有字段） ====================
|
|||
|
|
def update_mcu(mcu_id, **fields):
|
|||
|
|
"""动态更新 MCU（支持所有字段）"""
|
|||
|
|
set_clause = ",".join(f"{k}=?" for k in fields.keys())
|
|||
|
|
values = tuple(fields.values()) + (mcu_id,)
|
|||
|
|
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
cur.execute(f"UPDATE mcu SET {set_clause} WHERE id=?", values)
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
# ==================== 删除 MCU ====================
|
|||
|
|
def delete_mcu(mcu_id):
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
cur.execute("DELETE FROM mcu WHERE id=?", (mcu_id,))
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|