84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
from database import DatabaseManager
|
||
import os
|
||
|
||
# 检查backend/data目录下的数据库
|
||
db_path = os.path.join(os.path.dirname(__file__), 'data', 'body_balance.db')
|
||
db = DatabaseManager(db_path)
|
||
db.init_database()
|
||
conn = db.get_connection()
|
||
cursor = conn.cursor()
|
||
|
||
# 检查患者总数
|
||
cursor.execute('SELECT COUNT(*) FROM patients')
|
||
count = cursor.fetchone()[0]
|
||
print(f'body_balance.db中的患者总数: {count}')
|
||
|
||
# 查看前5条患者数据
|
||
cursor.execute('SELECT * FROM patients LIMIT 5')
|
||
rows = cursor.fetchall()
|
||
print('前5条患者数据:')
|
||
for row in rows:
|
||
print(dict(row))
|
||
|
||
conn.close()
|
||
|
||
# 如果没有数据，添加测试数据
|
||
if count == 0:
|
||
print('\n数据库中没有患者数据，添加测试数据...')
|
||
test_patients = [
|
||
{
|
||
'name': '张三',
|
||
'gender': '男',
|
||
'age': 30,
|
||
'birth_date': '1994-01-15',
|
||
'nationality': '汉族',
|
||
'height': 175.0,
|
||
'weight': 70.0,
|
||
'phone': '13800138001',
|
||
'shoe_size': '42',
|
||
'medical_history': '无',
|
||
'notes': '测试患者1'
|
||
},
|
||
{
|
||
'name': '李四',
|
||
'gender': '女',
|
||
'age': 25,
|
||
'birth_date': '1999-03-22',
|
||
'nationality': '汉族',
|
||
'height': 165.0,
|
||
'weight': 55.0,
|
||
'phone': '13800138002',
|
||
'shoe_size': '37',
|
||
'medical_history': '高血压',
|
||
'notes': '测试患者2'
|
||
},
|
||
{
|
||
'name': '王五',
|
||
'gender': '男',
|
||
'age': 35,
|
||
'birth_date': '1989-07-08',
|
||
'nationality': '回族',
|
||
'height': 180.0,
|
||
'weight': 80.0,
|
||
'phone': '13800138003',
|
||
'shoe_size': '44',
|
||
'medical_history': '糖尿病',
|
||
'notes': '测试患者3'
|
||
}
|
||
]
|
||
|
||
for patient in test_patients:
|
||
patient_id = db.create_patient(patient)
|
||
print(f'添加患者: {patient["name"]}, ID: {patient_id}')
|
||
|
||
print('\n重新测试get_patients方法:')
|
||
patients = db.get_patients(page=1, size=10, keyword='')
|
||
print(f'查询结果: {len(patients)}条记录')
|
||
for p in patients:
|
||
print(f' - {p["name"]} ({p["gender"]}, {p["age"]}岁)')
|
||
else:
|
||
print('\n数据库中已有患者数据，测试get_patients方法:')
|
||
patients = db.get_patients(page=1, size=10, keyword='')
|
||
print(f'查询结果: {len(patients)}条记录')
|
||
for p in patients:
|
||
print(f' - {p["name"]} ({p["gender"]}, {p["age"]}岁)') |