141 lines
4.5 KiB
Python
141 lines
4.5 KiB
Python
|
from database import init_db, get_all_mcus, get_mcu, add_mcu, update_mcu, delete_mcu
|
|||
|
|
|
|||
|
|
# 定义所有 MCU 字段（除了 id）
|
|||
|
|
MCU_FIELDS = [
|
|||
|
|
"name", "core_num", "core_type", "instruction_set", "coremark", "coremark_per_mhz",
|
|||
|
|
"frequency", "flash", "rom", "ram", "sram", "sram_in_rtc", "bus_width",
|
|||
|
|
"cache_type", "l1_cache_size", "l2_cache_size", "l3_cache_size",
|
|||
|
|
"pipeline_depth", "simd_support", "fpu", "package", "gpios", "uarts",
|
|||
|
|
"i2cs", "i2ses", "spis", "spi_protocols", "systems", "manufacturer", "link"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
FIELD_LABELS = {
|
|||
|
|
"name": "Name",
|
|||
|
|
"core_num": "# of Cores",
|
|||
|
|
"core_type": "Core Architecture",
|
|||
|
|
"instruction_set": "Instruction Set",
|
|||
|
|
"coremark": "CoreMark Score",
|
|||
|
|
"coremark_per_mhz": "CoreMark/MHz",
|
|||
|
|
"frequency": "Frequency (MHz)",
|
|||
|
|
"flash": "Flash (KB)",
|
|||
|
|
"rom": "ROM (KB)",
|
|||
|
|
"ram": "RAM (KB)",
|
|||
|
|
"sram": "SRAM (KB)",
|
|||
|
|
"sram_in_rtc": "RTC SRAM (KB)",
|
|||
|
|
"bus_width": "Bus Width (bit)",
|
|||
|
|
"cache_type": "Cache Type",
|
|||
|
|
"l1_cache_size": "L1 Cache (KB)",
|
|||
|
|
"l2_cache_size": "L2 Cache (KB)",
|
|||
|
|
"l3_cache_size": "L3 Cache (KB)",
|
|||
|
|
"pipeline_depth": "Pipeline Depth",
|
|||
|
|
"simd_support": "SIMD support (y=1, n=0)",
|
|||
|
|
"fpu": "FPU support (Single/Double Precision)",
|
|||
|
|
"package": "Package",
|
|||
|
|
"gpios": "Number of GPIO ",
|
|||
|
|
"uarts": "Number of UARTs",
|
|||
|
|
"i2cs": "Number of I2C",
|
|||
|
|
"i2ses": "Number of I2S",
|
|||
|
|
"spis": "Number of SPI",
|
|||
|
|
"spi_protocols": "SPI Protocols",
|
|||
|
|
"systems": "Systems",
|
|||
|
|
"manufacturer": "Manufacturer",
|
|||
|
|
"link": "Link"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def list_mcus():
|
|||
|
|
mcus = get_all_mcus()
|
|||
|
|
if not mcus:
|
|||
|
|
print("⚠️ MCU 数据库为空。")
|
|||
|
|
else:
|
|||
|
|
print("\n=== MCU 列表 ===")
|
|||
|
|
for m in mcus:
|
|||
|
|
print(f"ID: {m['id']} | 名称: {m['name']} | 核心: {m['core_num']} x {m['core_type']} | "
|
|||
|
|
f"频率: {m['frequency']} MHz | 厂商: {m.get('manufacturer','')}")
|
|||
|
|
print("=================\n")
|
|||
|
|
|
|||
|
|
def input_fields(existing=None):
|
|||
|
|
"""动态输入所有字段（友好名称提示），existing 用于编辑时提供默认值"""
|
|||
|
|
data = {}
|
|||
|
|
for field in MCU_FIELDS:
|
|||
|
|
old_val = existing.get(field) if existing else ""
|
|||
|
|
display_name = FIELD_LABELS.get(field, field) # ✅ 使用中文提示
|
|||
|
|
prompt = f"{display_name} ({old_val if old_val is not None else ''}): "
|
|||
|
|
val = input(prompt).strip()
|
|||
|
|
if val == "" and existing: # 编辑时保留原值
|
|||
|
|
continue
|
|||
|
|
if val == "": # 新增时空值存 None
|
|||
|
|
data[field] = None
|
|||
|
|
else:
|
|||
|
|
# ✅ 类型转换
|
|||
|
|
if field in ["pipeline_depth"]:
|
|||
|
|
try:
|
|||
|
|
data[field] = int(val)
|
|||
|
|
except ValueError:
|
|||
|
|
data[field] = None
|
|||
|
|
elif field in ["coremark_per_mhz"]:
|
|||
|
|
try:
|
|||
|
|
data[field] = float(val)
|
|||
|
|
except ValueError:
|
|||
|
|
data[field] = None
|
|||
|
|
elif field == "simd_support":
|
|||
|
|
data[field] = val.lower() in ["1", "true", "yes", "y"]
|
|||
|
|
else:
|
|||
|
|
data[field] = val
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def create_mcu():
|
|||
|
|
print("\n=== 添加 MCU ===")
|
|||
|
|
fields = input_fields()
|
|||
|
|
add_mcu(**fields)
|
|||
|
|
print("✅ MCU 添加成功！")
|
|||
|
|
|
|||
|
|
def edit_mcu():
|
|||
|
|
mcu_id = int(input("输入要修改的 MCU ID: "))
|
|||
|
|
existing = get_mcu(mcu_id)
|
|||
|
|
if not existing:
|
|||
|
|
print("❌ MCU ID 不存在")
|
|||
|
|
return
|
|||
|
|
print("\n=== 编辑 MCU（直接回车保留原值） ===")
|
|||
|
|
fields = input_fields(existing)
|
|||
|
|
if fields:
|
|||
|
|
update_mcu(mcu_id, **fields)
|
|||
|
|
print("✅ MCU 更新成功！")
|
|||
|
|
else:
|
|||
|
|
print("⚠️ 未修改任何字段。")
|
|||
|
|
|
|||
|
|
def remove_mcu():
|
|||
|
|
mcu_id = int(input("输入要删除的 MCU ID: "))
|
|||
|
|
if not get_mcu(mcu_id):
|
|||
|
|
print("❌ MCU ID 不存在")
|
|||
|
|
return
|
|||
|
|
delete_mcu(mcu_id)
|
|||
|
|
print("✅ MCU 删除成功！")
|
|||
|
|
|
|||
|
|
def menu():
|
|||
|
|
while True:
|
|||
|
|
print("\n=== MCU 管理菜单 ===")
|
|||
|
|
print("1. 查看所有 MCU")
|
|||
|
|
print("2. 添加 MCU")
|
|||
|
|
print("3. 修改 MCU")
|
|||
|
|
print("4. 删除 MCU")
|
|||
|
|
print("5. 退出")
|
|||
|
|
choice = input("选择操作 (1-5): ")
|
|||
|
|
|
|||
|
|
if choice == "1":
|
|||
|
|
list_mcus()
|
|||
|
|
elif choice == "2":
|
|||
|
|
create_mcu()
|
|||
|
|
elif choice == "3":
|
|||
|
|
edit_mcu()
|
|||
|
|
elif choice == "4":
|
|||
|
|
remove_mcu()
|
|||
|
|
elif choice == "5":
|
|||
|
|
print("👋 退出 MCU 管理工具")
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
print("❌ 无效选择，请重新输入。")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
init_db()
|
|||
|
|
menu()
|