diff --git a/test_database.py b/test_database.py new file mode 100644 index 0000000..969244a --- /dev/null +++ b/test_database.py @@ -0,0 +1,32 @@ +import logging +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker +from app.models import Domain, Mailbox # 根据你的项目结构调整导入 + +# 配置数据库连接 +DATABASE_URL = "sqlite:///email_system.db" # 使用默认的 SQLite 数据库 +engine = create_engine(DATABASE_URL) +Session = sessionmaker(bind=engine) + +def test_database_queries(): + session = Session() + try: + # 查询所有域名 + domains = session.query(Domain).all() + print("所有域名:") + for domain in domains: + print(f"域名: {domain.name}, 描述: {domain.description}") + + # 查询所有邮箱 + mailboxes = session.query(Mailbox).all() + print("\n所有邮箱:") + for mailbox in mailboxes: + print(f"邮箱: {mailbox.address}, 域名ID: {mailbox.domain_id}, 描述: {mailbox.description}") + + except Exception as e: + logging.error(f"查询数据库时出错: {str(e)}") + finally: + session.close() + +if __name__ == "__main__": + test_database_queries() \ No newline at end of file