biliDanmuAuto.py

# (C) 2019-2020 lifegpc
# This file is part of bili.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
import biliDanmu
import json
import biliLogin
import time
import biliTime
from JSONParser import loadset
import sys
from command import gopt
from lang import getdict, getlan
from inspect import currentframe


lan = None
se = loadset()
if se == -1 or se == -2:
    se = {}
ip = {}
if len(sys.argv) > 1:
    ip = gopt(sys.argv[1:])
lan = getdict('biliDanmu', getlan(se, ip))


def getMembers(filen, r, da, pos, mri, ip):
    "获取弹幕条数"
    log = False
    logg = None
    if 'logg' in ip:
        log = True
        logg = ip['logg']
    ns = True
    if 's' in ip:
        ns = False
    if log:
        logg.write(f"ns = {ns}", currentframe(), "biliDanmuAuto Var")
    bs = True
    ts = 300
    rec = 0
    m2 = mri
    if ns:
        print(lan['OUTPUT7'].replace('<date>', biliTime.tostr(biliTime.getDate(da))))  # 正在抓取<date>的弹幕……
    while bs:
        if log:
            logg.write(f"ts = {ts}", currentframe(), "biliDanmuAuto Var2")
        read = biliDanmu.downloadh(filen, r, pos, da, logg)
        if read == -1:
            if log:
                logg.write(f"read = {read}", currentframe(), "biliDanmuAuto Var3")
            return -1
        elif read == -3:
            rec = rec + 1
            if log:
                logg.write(f"read = {read}\nrec = {rec}", currentframe(), "biliDanmuAuto Var4")
            if rec % 5 != 0:
                time.sleep(5)
                print(lan['OUTPUT8'].replace('<number>', str(rec)))  # 正在进行第<number>次重连
            else:
                bss = True
                while bss:
                    inn = input(f"{lan['INPUT4'].replace('<number>',str(rec))}(y/n)")  # 是否重连？（已经失败<number>次）
                    if len(inn) > 0 and inn[0].lower() == 'y':
                        time.sleep(5)
                        print(lan['OUTPUT8'].replace('<number>', str(rec)))  # 正在进行第<number>次重连
                        bss = False
                    elif len(inn) > 0 and inn[0].lower() == 'n':
                        return -1
        elif 'status' in read and read['status'] == -2:
            if log:
                logg.write(f"read = {read}", currentframe(), "biliDanmuAuto Var5")
            obj = json.loads(read['d'])
            if obj['code'] == -101:
                if obj['message'] == '账户未登录':
                    ud = {}
                    read = biliLogin.login(r, ud, ip, logg)
                    if read > 1:
                        return -1
                else:
                    print(obj)
                    print(lan['OUTPUT9'].replace('<number>', str(ts)))  # 休眠<number>s
                    time.sleep(ts)
                    ts = ts + 300
            else:
                print(obj)
                print(lan['OUTPUT9'].replace('<number>', str(ts)))  # 休眠<number>s
                time.sleep(ts)
                ts = ts + 300
        else:
            bs = False
    d = read
    l = 0  # noqa: E741
    li = []
    z = len(d['list'])
    if ns:
        print(lan['OUTPUT12'])  # 正在处理……
    for i in d['list']:
        if int(m2) < int(i['ri']):
            m2 = i['ri']
        if int(mri) < int(i['ri']):
            l = l + 1  # noqa: E741
            li.append(i)
    d['list'] = li
    if log:
        logg.write(f"m2 = {m2}", currentframe(), "biliDanmuAuto Var6")
    return {'z': z, 'l': l, 'm': m2, 'd': d}


def reload(d, mri, ns):
    l = 0  # noqa: E741
    li = []
    if ns:
        print(lan['OUTPUT12'])  # 正在处理……
    for i in d['d']['list']:
        if int(mri) < int(i['ri']):
            l = l + 1  # noqa: E741
            li.append(i)
    d['d']['list'] = li
    return {'z': d['z'], 'l': l, 'm': d['m'], 'd': d['d']}


def getnownumber(d, mri):
    l = 0  # noqa: E741
    m = len(d['list'])
    for i in d['list']:
        if int(mri) < int(i['ri']):
            l = l + 1  # noqa: E741
    return {'l': l, 'm': m}