try:
    # 导入库
    from outputs import *

    # 程序初始化
    def init():
        config(pinkshell = True)
        clean.screen()

    # 登录
    def login():
        log.warn('请先登录！（密码为112233）')
        output.horizon()
        while True:
            password = keyboard('密码')
            if password == '112233':
                return None
            elif password == '':
                pass
            else:
                log.error('密码错误！')

    # 关于
    def readme():
        clean.screen()
        cursor.hide()
        about()
        keyboard(f'按{color.yellow}Enter{style.rst}退出...')
        cursor.show()
        main()

    # 计算
    def calc():
        clean.screen()
        try:
            all = int(keyboard('晚自习所有人数'))
            to = int(keyboard('晚自习实到人数'))
        except Exception:
            log.error('请输入整数！')
            keyboard(f'按{color.yellow}Enter{style.rst}重来...')
            calc()
        results = all - to
        cursor.hide()
        output.bar('正在计算', '计算完成', 5)
        clean.screen()
        output.horizon()
        output.li([
            f'应到：{color.yellow}{all}{style.rst}人',
            f'实到：{color.yellow}{to}{style.rst}人'
        ])
        output.horizon()
        output.echo(f'有{color.yellow}{results}{style.rst}人未到！')
        output.horizon()
        keyboard(f'按{color.yellow}Enter{style.rst}退出...')
        cursor.show()
        main()

    # 主菜单
    def main():
        clean.screen()
        output.horizon()
        output.echo(f'{color.yellow}欢迎使用晚自习人数计算系统！{style.rst}\n')
        output.li([
            '这是Output库的一个示例程序。',
            '里面涵盖了一些功能。',
            '对学习和体验有帮助。'
        ])
        output.horizon()
        output.echo(f'{color.yellow}W{style.rst}和{color.yellow}S{style.rst}选择，{color.yellow}Enter{style.rst}确定\n')
        output.menu(
            [' 计算 ', ' 关于 ', ' 退出 '],
            [calc, readme, lambda: log.fatal('已退出！')]
        )

    # 主程序
    if __name__ == '__main__':
        init()
        login()
        main()
except Exception as err:
    traceback(err)