main.py

import sys
from PySide2.QtWidgets import QApplication
from forms import Setup, Query, paperChooser
from forms.models import (
    dbSession,
    Paper,
    Question,
    Record,
    V_Question,
    Q_Record,
    Q_Property,
    Q_Operation,
    OP,
    fill_db_with_questions,
)

from sqlalchemy import and_


class Main:
    def __init__(self):
        super().__init__()
        # 显示：启动界面
        self.setup = Setup()
        self.setup.show()
        self.session = dbSession()
        self.setup.ui.pushButtonQuery.clicked.connect(self.open_paper)

    def open_paper(self):
        def get_last_record_id():
            """从数据库的True_Paper的最后一行读取数据
            如果表中没有行，或者最后一行已经finished，则返回None"""
            # and_:使用两个条件，即未完成的真题测试
            last_record = (
                self.session.query(Record)
                .filter_by(finished=False, is_practice=True)
                .one_or_none()
            )
            if last_record:
                return last_record.id
            else:
                return None

        last_record_id = get_last_record_id()
        if last_record_id:  # 如果存在 上次试卷，则
            # 打开: 答题界面
            self.open_query(last_record_id)
        else:  # 如果不存在 上次试卷，则
            # 打开：选择试卷对话框
            self.open_paperChooser()

    def open_paperChooser(self):
        """打开试卷选择对话框，并设置信号"""
        # 显示：选试卷对话框
        self.paper_chooser = paperChooser()
        self.paper_chooser.show()
        # 点击：确定按钮 -> 新建：选择的试卷为record
        self.paper_chooser.ui.buttonBox.accepted.connect(self.add_record)

    def add_record(self):
        """使用试卷选择对话框的数据，使用选定的试卷，新建record"""
        record_id = self.paper_chooser.new_record()
        self.open_query(record_id)

    def open_query(self, record_id):
        """打开答题界面
        
        Arguments:
            paper {str tuple} -- (科目、地区、试卷名)
        """
        # 启动显示：答题界面
        self.query = Query(record_id)
        self.query.ui.show()
        # 启动：定时器(每秒触发)
        self.query.ui.timer.start(1000)
        # 启动：答题总时间计时器
        self.query.ui.elapsed_time.start()
        # 记录：打开试卷时间
        self.query.add_operation_time(OP.OPEN_QUERY)
        # 完全打开答题界面后，再
        # 关闭：启动界面
        self.setup.close()


if __name__ == "__main__":
    # fill_db_with_questions()
    DEBUG = 0
    if DEBUG:
        session = dbSession()
        session.query(Record).delete()
        session.query(V_Question).delete()
        session.query(Q_Record).delete()
        session.query(Q_Property).delete()
        session.query(Q_Operation).delete()
        session.commit()
    else:
        fill_db_with_questions()
    app = QApplication()
    w = Main()
    sys.exit(app.exec_())