run.py

from bookshelf import *
from instance import *
import HbookerAPI
import datetime
import msg
import sys
import re
import token_parser
import cache

default_current_app_version = "2.9.290"


def refresh_bookshelf_list():
    response = HbookerAPI.BookShelf.get_shelf_list()
    if response.get('code') == '100000':
        BookShelfList.clear()
        for shelf in response['data']['shelf_list']:
            BookShelfList.append(BookShelf(shelf))
    else:
        print(msg.m('error_response') + str(response))
    for shelf in BookShelfList:
        shelf.show_info()
    if len(BookShelfList) == 1:
        shell_bookshelf(['', '1'])


def shell_login(inputs):
    print('！！！此登入法已失效！！！保留測試用 20220609 ')
    if len(inputs) == 1 and Vars.cfg.data.get('user_account') is not None and \
            Vars.cfg.data.get('user_password') is not None:
        pass
    elif len(inputs) >= 3:
        Vars.cfg.data['user_account'] = inputs[1]
        Vars.cfg.data['user_password'] = inputs[2]
    else:
        print(msg.m('input_correct_var'))
        return False
    response = HbookerAPI.SignUp.login(Vars.cfg.data['user_account'], Vars.cfg.data['user_password'])
    if response.get('code') == '100000':
        Vars.cfg.data['reader_name'] = response['data']['reader_info']['reader_name']
        Vars.cfg.data['user_code'] = response['data']['user_code']
        Vars.cfg.data['common_params'] = {'login_token': response['data']['login_token'],
                                          'account': response['data']['reader_info']['account']}
        Vars.cfg.save()
        HbookerAPI.set_common_params(Vars.cfg.data['common_params'])
        print(msg.m('login_success_user') + Vars.cfg.data['reader_name'])
        return True
    else:
        # print("response logon: " + str(response))
        print(response)
        return False


def shell_bookshelf(inputs):
    if len(inputs) >= 2:
        if not BookShelfList:
            refresh_bookshelf_list()
        Vars.current_bookshelf = get_bookshelf_by_index(inputs[1])
        if Vars.current_bookshelf is None:
            print(msg.m('input_correct_var'))
        else:
            print(msg.m('picked_shelf_s') + Vars.current_bookshelf.shelf_name + msg.m('picked_shelf_e'))
            Vars.current_bookshelf.get_book_list()
            Vars.current_bookshelf.show_book_list()
    else:
        refresh_bookshelf_list()


def shell_select_books(inputs):
    if len(inputs) >= 2:
        Vars.current_book = None
        if Vars.current_bookshelf is not None and not re.match('^[0-9]{9,}$', inputs[1]):
            Vars.current_book = Vars.current_bookshelf.get_book(inputs[1])
            if Vars.current_book is None:
                print(msg.m('failed_get_book_info_index'), inputs[1])
                return
        elif re.match('^[0-9]{9,}$', inputs[1]):
            Vars.current_book = HbookerAPI.Book.get_info_by_id(inputs[1])
            if Vars.current_book.get('code') == '100000':
                Vars.current_book = Book(None, Vars.current_book['data']['book_info'])
            else:
                # test local cache and init a book.Book
                if not cache.test_cache_and_init_object(inputs[1]):
                    return
        else:
            print('input', inputs[1], 'not a book ID, ID should be a 9 digit number')
            return
        # update book info
        cache.save_cache(f"{Vars.current_book.book_id}.json", Vars.current_book.book_info)

        print('《' + Vars.current_book.book_name + '》')
        Vars.current_book.get_division_list()
        Vars.current_book.get_chapter_catalog()
        # with open('test/book info.json', 'w') as f:
        #     json.dump({'book_info': response['data']['book_info'], 'division_list': Book.division_list}, f, indent=4)
        if len(inputs) < 3:
            Vars.current_book.show_division_list()
            Vars.current_book.show_latest_chapter()
        server_time = (datetime.datetime.now(tz=datetime.timezone.utc) +
                       datetime.timedelta(hours=8)).replace(tzinfo=None)
        last_update_time = \
            datetime.datetime.strptime(Vars.current_book.last_chapter_info['uptime'], '%Y-%m-%d %H:%M:%S')
        up_ago_time = server_time - last_update_time
        print('  last update ' + str(up_ago_time.days) + ' days ago')
    else:
        if Vars.current_book is not None:
            Vars.current_book.show_chapter_list_order_division()
        elif Vars.current_bookshelf is None:
            print(msg.m('not_picked_shelf'))
        else:
            Vars.current_bookshelf.get_book_list()
            Vars.current_bookshelf.show_book_list()


def shell_download_book(inputs):
    if len(inputs) > 1:
        shell_select_books(inputs)
    if Vars.current_book is None:
        print(msg.m('not_picked_book'))
        return
    print(msg.m('start_book_dl'))
    Vars.current_book.download_book_multi_thread()


def shell_download_list(inputs):
    if len(inputs) >= 2:
        list_file = inputs[1]
    else:
        list_file = 'list.txt'
    try:
        list_file_input = open(list_file, 'r', encoding='utf-8')
    except OSError:
        print(OSError)
        return
    list_lines = list_file_input.readlines()
    for line in list_lines:
        if re.match("^\\s*([0-9]{9}).*$", line):
            book_id = re.sub("^\\s*([0-9]{9}).*$\\n?", "\\1", line)
            print("Book ID: " + book_id + " ", end='')
            shell_download_book(['', book_id, ''])


def shell_download_shelf(inputs):
    if len(inputs) >= 2:
        shell_bookshelf(inputs)
    if Vars.current_bookshelf is not None:
        for bookshelf_index, book in enumerate(Vars.current_bookshelf.BookList, start=1):
            # shell_download_book(['', book.book_id])
            shell_download_book(['', str(bookshelf_index)])  # check if bookshelf_index is correct and download
    else:
        print(msg.m('not_picked_shelf'))


def check_in_today():
    # if Vars.cfg.data.get('user_account') is None or Vars.cfg.data.get('user_account') == "" \
    #         or Vars.cfg.data.get('user_password') is None or Vars.cfg.data.get('user_password') == "":
    #     print(msg.m('not_login_pl_login'))
    #     return False
    check_in_records = HbookerAPI.CheckIn.get_check_in_records()
    if check_in_records.get('code') == '100000':
        if check_in_today_do(check_in_records):
            return True
        else:
            return False
    elif check_in_records.get('code') == '200001':
        # {'code': '200001', 'tip': '缺少登录必需参数'}
        # {'code': '310002', 'tip': '此账户未实名认证，请先绑定手机'}
        # {'code': '240001', 'tip': '注册超过24小时的用户才能签到哦~'}
        print(check_in_records)
        print(msg.m('not_login_pl_login'))
        return False
    # elif check_in_records.get('code') == '200100':
    #     # {'code': '200100', 'tip': '登录状态过期，请重新登录'}
    #     print(msg.m('check_in_token_failed'))
    #     if shell_login(['']):
    #         print(msg.m('check_in_re_login_retry_check_in'))
    #         check_in_records = HbookerAPI.CheckIn.get_check_in_records()
    #         if check_in_records.get('code') == '100000':
    #             if check_in_today_do(check_in_records):
    #                 return True
    #             else:
    #                 return False
    #         else:
    #             print(msg.m('check_in_error_1') + str(check_in_records) + '\n')
    #             return False
    #     else:
    #         print(msg.m('check_in_re_login_failed'))
    #         return False
    else:
        print(msg.m('check_in_error_2') + str(check_in_records) + '\n')
        return False


def check_in_today_do(check_in_records):
    # UTC+8
    server_time = datetime.datetime.now(tz=datetime.timezone.utc) + datetime.timedelta(hours=8)
    print(str(server_time.date()) + " " + str(server_time.hour).rjust(2, '0') + ":"
          + str(server_time.minute).rjust(2, '0'))
    today = str(server_time.date())
    for record in check_in_records['data']['sign_record_list']:
        if record['date'] == today:
            if record['is_signed'] == '0':
                check_in = HbookerAPI.CheckIn.do_check_in()
                if check_in.get('code') == '100000':
                    check_in_exp = check_in.get('data').get('bonus').get('exp')
                    check_in_hlb = check_in.get('data').get('bonus').get('hlb')
                    check_in_recommend = check_in.get('data').get('bonus').get('recommend')
                    print(msg.m('check_in_success_got') + str(check_in_exp) + msg.m('check_in_xp') + str(check_in_hlb)
                          + msg.m('check_in_token') + str(check_in_recommend) + msg.m('check_in_recommend'))
                    if check_in_exp is None or check_in_hlb is None or check_in_recommend is None:
                        print('debug : check\n' + str(check_in))  # debug
                    return True
                elif check_in.get('code') == '340001':
                    print(msg.m('check_in_no_redo'))
                    return True
                elif check_in.get('code') == '310002':
                    print(msg.m('check_in_failed') + msg.m('check_in_no_certification'))
                    return True
                else:
                    print(msg.m('check_in_failed') + str(check_in) + '\n')
                    return False
            else:
                print(msg.m('check_in_already'))
                return True
    # 日期異常，未找本日對應簽到記錄，不進行簽到嘗試
    print(msg.m('check_in_error_day_not_found') + str(check_in_records))
    return False


def agreed_read_readme():
    if Vars.cfg.data.get('agreed_to_readme') != 'yes':
        print(msg.m('read_readme'))
        print(msg.m('agree_terms'))

        confirm = get('>').strip()
        if confirm == 'yes':
            Vars.cfg.data['agreed_to_readme'] = 'yes'
            Vars.cfg.save()
        else:
            sys.exit()


def shell_switch_message_charter_set():
    if Vars.cfg.data['interface_traditional_chinese']:
        Vars.cfg.data['interface_traditional_chinese'] = False
        pass
    else:
        Vars.cfg.data['interface_traditional_chinese'] = True
    Vars.cfg.save()
    msg.set_message_lang(Vars.cfg.data['interface_traditional_chinese'])
    print(msg.m('lang'))


def setup_config():
    Vars.cfg.load()
    config_change = False
    if type(Vars.cfg.data.get('interface_traditional_chinese')) is not bool:
        Vars.cfg.data['interface_traditional_chinese'] = False
        msg.set_message_lang()
        config_change = True
    msg.set_message_lang(Vars.cfg.data['interface_traditional_chinese'])

    if type(Vars.cfg.data.get('cache_dir')) is not str or Vars.cfg.data.get('cache_dir') == "":
        Vars.cfg.data['cache_dir'] = "./Cache/"
        config_change = True

    if type(Vars.cfg.data.get('output_dir')) is not str or Vars.cfg.data.get('output_dir') == "":
        Vars.cfg.data['output_dir'] = "./Hbooker/"
        config_change = True

    if type(Vars.cfg.data.get('local_cache_dir')) is not str or Vars.cfg.data.get('local_cache_dir') == "":
        Vars.cfg.data['local_cache_dir'] = "./LocalCache/"
        config_change = True

    if not isinstance(Vars.cfg.data.get('backups_local_cache'), bool):
        Vars.cfg.data['backups_local_cache'] = True
        config_change = True

    if type(Vars.cfg.data.get('do_backup')) is not bool:
        Vars.cfg.data['do_backup'] = True
        config_change = True

    if type(Vars.cfg.data.get('backup_dir')) is not str or Vars.cfg.data.get('backup_dir') == "":
        Vars.cfg.data['backup_dir'] = "./Hbooker/"
        config_change = True

    if not isinstance(Vars.cfg.data.get('force_book_description'), bool):
        Vars.cfg.data['force_book_description'] = True
        config_change = True

    if type(Vars.cfg.data.get('max_concurrent_downloads')) is not int or \
            Vars.cfg.data.get('max_concurrent_downloads') < 1:
        Vars.cfg.data['max_concurrent_downloads'] = 8
        config_change = True

    if type(Vars.cfg.data.get('current_app_version')) is not str:
        Vars.cfg.data['current_app_version'] = default_current_app_version
        config_change = True
    HbookerAPI.common_params['app_version'] = Vars.cfg.data['current_app_version']

    # if type(Vars.cfg.data.get('export_epub')) is not bool:
    #     Vars.cfg.data['export_txt'] = True
    #     config_change = True
    #
    # if type(Vars.cfg.data.get('export_txt')) is not bool:
    #     Vars.cfg.data['export_txt'] = True
    #     config_change = True

    if config_change:
        Vars.cfg.save()


def get_app_update_version_info():
    response = (HbookerAPI.CheckAppVersion.get_version())
    if response.get('code') == '100000':
        android_version = response.get('data').get('android_version')
        print(msg.m('app_update_info') + str(response))
        print(msg.m('current_version_var') + HbookerAPI.common_params['app_version'])
        print(msg.m('get_app_version_var') + android_version)

        print(msg.m('confirm_change_version_var'))
        confirm = get('>').strip()
        if confirm == 'yes':
            print(msg.m('confirm_msg'))
            HbookerAPI.common_params['app_version'] = android_version
            Vars.cfg.data['current_app_version'] = android_version
            Vars.cfg.save()
        else:
            print(msg.m('cancel_msg'))
        print(msg.m('current_version_var') + HbookerAPI.common_params['app_version'])
    else:
        print("error response: " + str(response))


def import_token():
    print(msg.m('login_method_change_message'))
    import_method = input('1/2: ')
    if import_method == '1':
        user_token = token_parser.token_from_novel_preferences_xml()
        Vars.cfg.data['reader_name'] = user_token.get('reader_name')
        Vars.cfg.data['user_code'] = user_token.get('user_code')
        Vars.cfg.data['common_params'] = {'login_token': user_token.get('login_token'),
                                          'account': user_token.get('account')}
        Vars.cfg.save()
        HbookerAPI.set_common_params(Vars.cfg.data['common_params'])
        if HbookerAPI.common_params.get('device_token') is not None:
            HbookerAPI.common_params.pop('device_token')
        print(user_token)
    else:
        user_token = token_parser.token_from_input()
        Vars.cfg.data['common_params'] = {'login_token': user_token.get('login_token'),
                                          'account': user_token.get('account')}
        Vars.cfg.save()
        HbookerAPI.set_common_params(Vars.cfg.data['common_params'])
        print(user_token)
    print(msg.m('import_token_complete'))
    token_test()


def toggle_token_device():
    if type(Vars.cfg.data.get('common_params')) is dict:
        if 'device_token' in Vars.cfg.data.get('common_params'):
            Vars.cfg.data['common_params'].pop('device_token')
            HbookerAPI.common_params.pop('device_token')
            print('remove device_token\': \'ciweimao_')
        else:
            Vars.cfg.data['common_params'].update({'device_token': 'ciweimao_'})
            HbookerAPI.common_params.update({'device_token': 'ciweimao_'})
            print('add device_token\': \'ciweimao_')
        Vars.cfg.save()


def token_test():
    login_in_test = HbookerAPI.CheckIn.get_check_in_records()
    if login_in_test.get('code') == '100000':
        print('login success')
        return
    elif login_in_test.get('code') == '200100':
        toggle_token_device()
    else:
        return

    login_in_test = HbookerAPI.CheckIn.get_check_in_records()
    if login_in_test.get('code') != '100000':
        print(msg.m('not_login_pl_login'))


def shell():
    if Vars.cfg.data.get('common_params') is not None:
        HbookerAPI.set_common_params(Vars.cfg.data['common_params'])
        if len(sys.argv) > 1:
            if str(sys.argv[1]).startswith('t'):
                if check_in_today():
                    sys.exit()
                else:
                    sys.exit(5)
                    # loop = True
                    # inputs = ['']
            else:
                check_in_today()
                loop = False
                inputs = sys.argv[1:]
        else:
            check_in_today()
            loop = True
            print(msg.m('help_msg'))
            refresh_bookshelf_list()
            inputs = re.split('\\s+', get('>').strip())
    else:
        loop = True
        save = False
        if Vars.cfg.data.get('user_account') is None or Vars.cfg.data.get('user_account') is not str:
            Vars.cfg.data['user_account'] = ""
            save = True
        if Vars.cfg.data.get('user_password') is None or Vars.cfg.data.get('user_password') is not str:
            Vars.cfg.data['user_password'] = ""
            save = True
        if save:
            Vars.cfg.save()
        print(msg.m('help_msg'))
        print(msg.m('not_login_pl_login'))
        if len(sys.argv) > 1:
            inputs = sys.argv[1:]
        else:
            inputs = re.split('\\s+', get('>').strip())
    while True:
        if inputs[0].startswith('q'):
            sys.exit()
        elif inputs[0].startswith('l'):
            shell_login(inputs)
            check_in_today()
        elif inputs[0].startswith('i'):
            import_token()
            check_in_today()
        elif inputs[0].startswith('s'):
            shell_bookshelf(inputs)
        elif inputs[0].startswith('b'):
            shell_select_books(inputs)
        elif inputs[0].startswith('ds') or inputs[0].startswith('downloads'):
            shell_download_shelf(inputs)
        elif inputs[0].startswith('d'):
            shell_download_book(inputs)
        elif inputs[0].startswith('u'):
            shell_download_list(inputs)
        elif inputs[0].startswith('t'):
            check_in_today()
        elif inputs[0].startswith('m'):
            shell_switch_message_charter_set()
        elif inputs[0].startswith('version'):
            get_app_update_version_info()
        # elif inputs[0].startswith('p'):
        #     toggle_token_device()
        else:
            print(msg.m('help_msg'))
        if loop is False:
            break
        inputs = re.split('\\s+', get('>').strip())


if __name__ == "__main__":
    setup_config()

    agreed_read_readme()

    shell()