import sys
from PySide6.QtWidgets import QApplication
from ui.inspection_window import InspectionWindow
from ui.login_window import LoginWindow
from src.session_manager import SessionManager
from datetime import datetime

def get_mock_data():
    return [
        {
            "date": "2025-12-18",
            "inspector": "张三",
            "batch_no": "B20251218-001",
            "heat_no": "H88291",
            "material": "Q235B",
            "spec": "φ20",
            "weight": "1500kg"
        },
        {
            "date": "2025-12-18",
            "inspector": "李四",
            "batch_no": "B20251218-002",
            "heat_no": "H88292",
            "material": "HRB400",
            "spec": "φ25",
            "weight": "2300kg"
        },
        {
            "date": "2025-12-17",
            "inspector": "王五",
            "batch_no": "B20251217-005",
            "heat_no": "H88280",
            "material": "Q345",
            "spec": "10x10",
            "weight": "500kg"
        },
        {
            "date": "2025-12-16",
            "inspector": "赵六",
            "batch_no": "B20251216-010",
            "heat_no": "H88275",
            "material": "304",
            "spec": "2mm",
            "weight": "120kg"
        },
        {
            "date": "2025-12-16",
            "inspector": "赵六",
            "batch_no": "B20251216-010",
            "heat_no": "H88275",
            "material": "304",
            "spec": "2mm",
            "weight": "120kg"
        },
        {
            "date": "2025-12-16",
            "inspector": "赵六",
            "batch_no": "B20251216-010",
            "heat_no": "H88275",
            "material": "304",
            "spec": "2mm",
            "weight": "120kg"
        },
        {
            "date": "2025-12-16",
            "inspector": "赵六",
            "batch_no": "B20251216-010",
            "heat_no": "H88275",
            "material": "304",
            "spec": "2mm",
            "weight": "120kg"
        }
    ]

def main():
    app = QApplication(sys.argv)
    
    font = app.font()
    font.setFamily("Microsoft YaHei")
    font.setPointSize(10) # Base size
    app.setFont(font)

    session = SessionManager()
    
    login_window = LoginWindow()
    
    def on_login_success(user_info):
        session.set_current_user(user_info)

        window = InspectionWindow()

        # 使用新的分页加载方式
        window.load_initial_data()

        window.showMaximized()

    login_window.login_success.connect(on_login_success)
    login_window.showMaximized()
    
    sys.exit(app.exec())

if __name__ == "__main__":
    main()