podflow 2025.1.26__py3-none-any.whl

Podflow/bilibili/login.py

@@ -0,0 +1,307 @@
+# Podflow/bilibili/login.py
+# coding: utf-8
+
+import binascii
+import json
+import os
+import re
+import time
+from datetime import datetime
+import requests
+from Cryptodome.Cipher import PKCS1_OAEP
+from Cryptodome.Hash import SHA256
+from Cryptodome.PublicKey import RSA
+from Podflow.basic.file_save import file_save
+from Podflow.basic.http_client import http_client
+from Podflow.basic.qr_code import qr_code
+from Podflow.basic.time_stamp import time_stamp
+from Podflow.basic.write_log import write_log
+from Podflow.netscape.bulid_netscape import bulid_netscape
+
+
+# 获取最新的img_key和sub_key模块
+def getWbiKeys(bilibili_cookie=None):
+    bilibili_url = "https://api.bilibili.com/x/web-interface/nav"
+    if resp := http_client(
+        bilibili_url, "获取最新的img_key和sub_key", 10, 4, True, bilibili_cookie
+    ):
+        resp.raise_for_status()
+        json_content = resp.json()
+        img_url: str = json_content["data"]["wbi_img"]["img_url"]
+        sub_url: str = json_content["data"]["wbi_img"]["sub_url"]
+        img_key = img_url.rsplit("/", 1)[1].split(".")[0]
+        sub_key = sub_url.rsplit("/", 1)[1].split(".")[0]
+        return img_key, sub_key
+    else:
+        return "", ""
+
+
+# 申请哔哩哔哩二维码并获取token和URL模块
+def bilibili_request_qr_code():
+    # 实际申请二维码的API请求
+    response = http_client(
+        "https://passport.bilibili.com/x/passport-login/web/qrcode/generate",
+        "申请BiliBili二维码",
+        3,
+        5,
+        True,
+    )
+    data = response.json()
+    return data["data"]["qrcode_key"], data["data"]["url"]
+
+
+# 扫码登录哔哩哔哩并返回状态和cookie模块
+def bilibili_scan_login(token):
+    # 发送GET请求
+    response = http_client(
+        "https://passport.bilibili.com/x/passport-login/web/qrcode/poll",
+        "",
+        1,
+        1,
+        True,
+        None,
+        {"qrcode_key": token},
+    )
+    if not response:
+        return None, None, None
+    data = response.json()
+    return data["data"]["code"], response.cookies, data["data"]["refresh_token"]
+
+
+# 登陆哔哩哔哩模块
+def bilibili_login():
+    buvid3_and_bnut = http_client(
+        "https://www.bilibili.com", "哔哩哔哩主页", 10, 4, True
+    ).cookies.get_dict()
+    token, url = bilibili_request_qr_code()
+    print(f"{datetime.now().strftime('%H:%M:%S')}|请用BiliBili App扫描登录:")
+    upward = qr_code(url)
+    login_status_change = ""
+    time_print = f"{datetime.now().strftime('%H:%M:%S')}|BiliBili "
+    while True:
+        status, cookie, refresh_token = bilibili_scan_login(token)
+        if status == 86101:
+            login_status = "\033[0m未扫描\033[0m"
+        elif status == 86038:
+            login_status = "\033[31m二维码超时, 请重试\033[0m"
+        elif status == 86090:
+            login_status = "\033[32m扫描成功\033[0m"
+        elif status == 0:
+            login_status = "\033[32m登陆成功\033[0m"
+        else:
+            login_status = "\033[31m错误\033[0m"
+        if login_status_change != login_status:
+            if login_status == "":
+                print(f"{time_print}{login_status}".ljust(42), end="")
+            else:
+                print(f"\r{time_print}{login_status}".ljust(42), end="")
+        login_status_change = login_status
+        if status == 86038:
+            print("")
+            return status, refresh_token, upward
+        elif status == 0:
+            print("")
+            cookie["buvid3"] = buvid3_and_bnut.get("buvid3", "")
+            cookie["b_nut"] = buvid3_and_bnut.get("b_nut", "")
+            return cookie, refresh_token, upward
+        time.sleep(1)
+
+
+# 保存哔哩哔哩登陆成功后的cookies模块
+def save_bilibili_cookies():
+    bilibili_cookie, refresh_token, upward = bilibili_login()
+    if bilibili_cookie == 86038:
+        return {"cookie": None}, upward
+    bilibili_cookie = requests.utils.dict_from_cookiejar(bilibili_cookie)
+    bilibili_data = {"cookie": bilibili_cookie, "refresh_token": refresh_token}
+    bulid_netscape("yt_dlp_bilibili", bilibili_cookie)
+    return bilibili_data, upward
+
+
+# 检查哔哩哔哩是否需要刷新模块
+def judgment_bilibili_update(cookies):
+    url = "https://passport.bilibili.com/x/passport-login/web/cookie/info"
+    response = http_client(url, "BiliBili刷新判断", 3, 5, True, cookies)
+    response = response.json()
+    if response["code"] == 0:
+        return response["code"], response["data"]["refresh"]
+    else:
+        return response["code"], None
+
+
+# 生成CorrespondPath模块
+def getCorrespondPath(ts, key):
+    cipher = PKCS1_OAEP.new(key, SHA256)
+    encrypted = cipher.encrypt(f"refresh_{ts}".encode())
+    return binascii.b2a_hex(encrypted).decode()
+
+
+def bilibili_cookie_update_data(
+    bilibili_cookie, new_bilibili_cookie, bilibili_data, new_refresh_token
+):
+    new_bilibili_cookie["buvid3"] = bilibili_cookie["buvid3"]
+    new_bilibili_cookie["b_nut"] = bilibili_cookie["b_nut"]
+    bilibili_data["cookie"] = new_bilibili_cookie
+    bilibili_data["refresh_token"] = new_refresh_token
+    bulid_netscape("yt_dlp_bilibili", new_bilibili_cookie)
+    return bilibili_data
+
+
+# 哔哩哔哩cookie刷新模块
+def bilibili_cookie_update(bilibili_data):
+    bilibili_cookie = bilibili_data["cookie"]
+    # 获取refresh_csrf
+    key = RSA.importKey("""\
+-----BEGIN PUBLIC KEY-----
+MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDLgd2OAkcGVtoE3ThUREbio0Eg
+Uc/prcajMKXvkCKFCWhJYJcLkcM2DKKcSeFpD/j6Boy538YXnR6VhcuUJOhH2x71
+nzPjfdTcqMz7djHum0qSZA0AyCBDABUqCrfNgCiJ00Ra7GmRj+YCK1NJEuewlb40
+JNrRuoEUXpabUzGB8QIDAQAB
+-----END PUBLIC KEY-----""")
+
+    # 获取当前时间戳
+    ts = time_stamp()
+    # 获取refresh_csrf
+    refresh_csrf_response = http_client(
+        f"https://www.bilibili.com/correspond/1/{getCorrespondPath(ts, key)}",
+        "获取refresh_csrf",
+        3,
+        5,
+        True,
+        bilibili_cookie,
+    )
+    if refresh_csrf_match := re.search(
+        r'<div id="1-name">(.+?)</div>', refresh_csrf_response.text
+    ):
+        refresh_csrf_value = refresh_csrf_match[1]
+    else:
+        return {"cookie": None}
+    # 更新bilibili_cookie
+    update_cookie_url = (
+        "https://passport.bilibili.com/x/passport-login/web/cookie/refresh"
+    )
+    update_cookie_data = {
+        "csrf": bilibili_cookie["bili_jct"],
+        "refresh_csrf": refresh_csrf_value,
+        "source": "main_web",
+        "refresh_token": bilibili_data["refresh_token"],
+    }
+    update_cookie_response = http_client(
+        update_cookie_url,
+        "更新BiliBili_cookie",
+        3,
+        5,
+        True,
+        bilibili_cookie,
+        update_cookie_data,
+        "post",
+    )
+    if update_cookie_response.json()["code"] != 0:
+        return {"cookie": None}
+    new_bilibili_cookie = requests.utils.dict_from_cookiejar(
+        update_cookie_response.cookies
+    )
+    new_refresh_token = update_cookie_response.json()["data"]["refresh_token"]
+    # 确认更新bilibili_cookie
+    confirm_cookie_url = (
+        "https://passport.bilibili.com/x/passport-login/web/confirm/refresh"
+    )
+    confirm_cookie_data = {
+        "csrf": new_bilibili_cookie["bili_jct"],
+        "refresh_token": bilibili_data["refresh_token"],
+    }
+    confirm_cookie_response = http_client(
+        confirm_cookie_url,
+        "确认更新BiliBili_cookie",
+        3,
+        5,
+        True,
+        new_bilibili_cookie,
+        confirm_cookie_data,
+        "post",
+    )
+    if confirm_cookie_response.json()["code"] == 0:
+        return bilibili_cookie_update_data(
+            bilibili_cookie,
+            new_bilibili_cookie,
+            bilibili_data,
+            new_refresh_token,
+        )
+    else:
+        return {"cookie": None}
+
+
+def get_bilibili_data_success(bilibili_data, channelid_bilibili_ids):
+    print(
+        f"{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[32m获取cookie成功\033[0m"
+    )
+    img_key, sub_key = getWbiKeys()
+    bilibili_data["img_key"] = img_key
+    bilibili_data["sub_key"] = sub_key
+    bilibili_data["timestamp"] = time.time()
+    file_save(bilibili_data, "bilibili_data.json", "channel_data")
+    if not os.path.isfile("channel_data/yt_dlp_bilibili.txt"):
+        bulid_netscape("yt_dlp_bilibili", bilibili_data["cookie"])
+    return channelid_bilibili_ids, bilibili_data
+
+
+def get_bilibili_data_state(bilibili_data, channelid_bilibili_ids):
+    bilibili_login_code, bilibili_login_refresh_token = judgment_bilibili_update(
+        bilibili_data["cookie"]
+    )
+    upward = 0
+    try_num = 0
+    while try_num < 2 and (
+        bilibili_login_code != 0 or bilibili_login_refresh_token is not False
+    ):
+        if bilibili_login_code != 0:
+            if try_num == 0:
+                print(
+                    f"{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[31m未登陆\033[0m"
+                )
+            else:
+                print(
+                    f"\033[{upward + 3}F\033[{upward + 3}K{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[31m未登陆, 重试第\033[0m{try_num}\033[31m次\033[0m"
+                )
+            bilibili_data, upward = save_bilibili_cookies()
+            try_num += 1
+        else:
+            print(
+                f"{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[33m需刷新\033[0m"
+            )
+            bilibili_data = bilibili_cookie_update(bilibili_data)
+            if bilibili_data["cookie"]:
+                print(
+                    f"{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[32m刷新成功\033[0m"
+                )
+            else:
+                print(
+                    f"{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[31m刷新失败, 重新登陆\033[0m"
+                )
+        bilibili_login_code, bilibili_login_refresh_token = judgment_bilibili_update(
+            bilibili_data["cookie"]
+        )
+    if bilibili_login_code == 0 and bilibili_login_refresh_token is False:
+        return get_bilibili_data_success(bilibili_data, channelid_bilibili_ids)
+    write_log("BiliBili \033[31m获取cookie失败\033[0m")
+    return {}, {"cookie": None, "timestamp": 0.0}
+
+
+# 登陆刷新哔哩哔哩并获取data
+def get_bilibili_data(channelid_bilibili_ids):
+    if not channelid_bilibili_ids:
+        return {}, {"cookie": None, "timestamp": 0.0}
+    try:
+        with open("channel_data/bilibili_data.json", "r") as file:
+            bilibili_data = file.read()
+        bilibili_data = json.loads(bilibili_data)
+    except Exception:
+        bilibili_data = {"cookie": None, "timestamp": 0.0}
+    if time.time() - bilibili_data["timestamp"] - 60 * 60 > 0:
+        return get_bilibili_data_state(bilibili_data, channelid_bilibili_ids)
+    print(
+        f"{datetime.now().strftime('%H:%M:%S')}|BiliBili \033[33m获取cookie成功\033[0m"
+    )
+    if not os.path.isfile("channel_data/yt_dlp_bilibili.txt"):
+        bulid_netscape("yt_dlp_bilibili", bilibili_data["cookie"])
+    return channelid_bilibili_ids, bilibili_data

Podflow/config/__init__.py

@@ -0,0 +1,2 @@
+# Podflow/config/__init__.py
+# coding: utf-8

Podflow/config/build_original.py

@@ -0,0 +1,41 @@
+# Podflow/config/build_original.py
+# coding: utf-8
+
+from Podflow import gVar, parse
+from Podflow.config.get_config import get_config
+from Podflow.basic.folder_build import folder_build
+from Podflow.config.get_channelid import get_channelid
+from Podflow.config.correct_config import correct_config
+from Podflow.config.correct_channelid import correct_channelid
+from Podflow.config.get_channelid_id import get_channelid_id
+
+
+def build_original():
+    # 获取配置文件config
+    gVar.config = get_config(parse.config)
+    # 纠正配置信息config
+    correct_config()
+    # 从配置文件中获取YouTube的频道
+    gVar.channelid_youtube = get_channelid("youtube")
+    # 从配置文件中获取哔哩哔哩的频道
+    gVar.channelid_bilibili = get_channelid("bilibili")
+    # 构建文件夹channel_id
+    folder_build("channel_id")
+    # 构建文件夹channel_audiovisual
+    folder_build("channel_audiovisual")
+    # 构建文件夹channel_rss
+    folder_build("channel_rss")
+    # 构建文件夹channel_data
+    folder_build("channel_data")
+    # 修正channelid_youtube
+    gVar.channelid_youtube = correct_channelid(gVar.channelid_youtube, "youtube")
+    # 修正channelid_bilibili
+    gVar.channelid_bilibili = correct_channelid(gVar.channelid_bilibili, "bilibili")
+    # 读取youtube频道的id
+    gVar.channelid_youtube_ids = get_channelid_id(gVar.channelid_youtube, "youtube")
+    # 复制youtube频道id用于删除已抛弃的媒体文件夹
+    gVar.channelid_youtube_ids_original = gVar.channelid_youtube_ids.copy()
+    # 读取bilibili频道的id
+    gVar.channelid_bilibili_ids = get_channelid_id(gVar.channelid_bilibili, "bilibili")
+    # 复制bilibili频道id用于删除已抛弃的媒体文件夹
+    gVar.channelid_bilibili_ids_original = gVar.channelid_bilibili_ids.copy()

Podflow/config/channge_icon.py

@@ -0,0 +1,163 @@
+# Podflow/config/channge_icon.py
+# coding: utf-8
+
+from datetime import datetime, timedelta, timezone
+from astral import LocationInfo
+from astral.sun import sun
+from Podflow.basic.http_client import http_client
+from Podflow.basic.write_log import write_log
+from Podflow import gVar, default_config
+
+
+# 获取日出日落并判断昼夜模块
+def http_day_and_night(latitude, longitude):
+    sun_url = "https://api.sunrise-sunset.org/json"
+    sun_data = {
+        "lat": latitude,
+        "lng": longitude,
+        "date": "today",
+    }
+    sunrise_sunset = http_client(sun_url, "获取日出日落", 3, 5, True, None, sun_data)
+    if not sunrise_sunset:
+        return None
+    try:
+        time_dict = sunrise_sunset.json()["results"]
+        sunrise = time_dict["sunrise"]
+        sunset = time_dict["sunset"]
+    except KeyError:
+        return None
+    # 获取当前时间, 并去除时区
+    now = datetime.now()
+    # 将日出和日落时间转换为datetime对象
+    today = now.date()
+    sunrise_time = datetime.strptime(sunrise, "%I:%M:%S %p")
+    sunrise_time = sunrise_time.replace(
+        year=today.year, month=today.month, day=today.day, tzinfo=timezone.utc
+    )
+    sunset_time = datetime.strptime(sunset, "%I:%M:%S %p")
+    sunset_time = sunset_time.replace(
+        year=today.year, month=today.month, day=today.day, tzinfo=timezone.utc
+    )
+    # 转换日出和日落时间为时间戳
+    sunrise_now = sunrise_time.timestamp()
+    sunset_now = sunset_time.timestamp()
+    today = now.timestamp()
+    # 计算昨天及明天日出和日落时间戳
+    sunrise_yesterday = sunrise_now - 3600 * 24
+    sunset_yesterday = sunset_now - 3600 * 24
+    sunrise_tommorrow = sunrise_now + 3600 * 24
+    sunset_tommorrow = sunset_now + 3600 * 24
+    if sunrise_now < sunset_now:
+        return (
+            "light"
+            if (
+                sunrise_now < today < sunset_now
+                or sunrise_yesterday < today < sunset_yesterday
+                or sunrise_tommorrow < today < sunset_tommorrow
+            )
+            else "dark"
+        )
+    if (
+        sunrise_now > today > sunset_now
+        or sunrise_yesterday > today > sunset_yesterday
+        or sunrise_tommorrow > today > sunset_tommorrow
+    ):
+        return "dark"
+    else:
+        return "light"
+
+
+# 根据经纬度判断昼夜模块
+def judging_day_and_night(latitude, longitude):
+    # 创建一个 LocationInfo 对象, 只提供经纬度信息
+    location = LocationInfo("", "", "", latitude=latitude, longitude=longitude)
+    # 获取当前日期和时间, 并为其添加时区信息
+    now = datetime.now(timezone.utc)
+    yesterday = now - timedelta(days=1)
+    tommorrow = now + timedelta(days=1)
+
+    def sunrise_sunset(time):
+        # 创建一个 Sun 对象
+        sun_time = sun(location.observer, date=time)
+        # 计算日出和日落时间, 以及日落前和日出后的一小时
+        sunrise = sun_time["sunrise"]
+        sunset = sun_time["sunset"]
+        sunrise_minus_one_hour = sunrise  # - timedelta(hours=1)
+        sunset_plus_one_hour = sunset  # + timedelta(hours=1)
+        return sunrise_minus_one_hour, sunset_plus_one_hour
+
+    sunrise_now, sunset_now = sunrise_sunset(now)
+    sunrise_yesterday, sunset_yesterday = sunrise_sunset(yesterday)
+    sunrise_tommorrow, sunset_tommorrow = sunrise_sunset(tommorrow)
+    if sunrise_now < sunset_now:
+        return (
+            "light"
+            if (
+                sunrise_now < now < sunset_now
+                or sunrise_yesterday < now < sunset_yesterday
+                or sunrise_tommorrow < now < sunset_tommorrow
+            )
+            else "dark"
+        )
+    if (
+        sunrise_now > now > sunset_now
+        or sunrise_yesterday > now > sunset_yesterday
+        or sunrise_tommorrow > now > sunset_tommorrow
+    ):
+        return "dark"
+    else:
+        return "light"
+
+
+def ipinfo():
+    if response := http_client("https://ipinfo.io/json/", "", 1, 0):
+        data = response.json()
+        # 提取经度和纬度
+        coordinates = data["loc"].split(",")
+        return True, coordinates[0], coordinates[1]
+    else:
+        return False, None, None
+
+
+def ipapi():
+    if response := http_client("http://ip-api.com/json/", "", 1, 0):
+        data = response.json()
+        # 提取经度和纬度
+        return True, data["lat"], data["lon"]
+    else:
+        return False, None, None
+
+
+def freegeoip():
+    if response := http_client("https://freegeoip.app/json/", "", 1, 0):
+        data = response.json()
+        # 提取经度和纬度
+        return True, data["latitude"], data["longitude"]
+    else:
+        return False, None, None
+
+
+# 根据日出日落修改封面(只适用原封面)模块
+def channge_icon():
+    config = gVar.config
+    if config["icon"] != default_config["icon"]:
+        return
+    label = False
+    # 公网获取经纬度
+    label, latitude, longitude = ipinfo()
+    if label is False:
+        write_log("获取经纬度信息重试中...\033[97m1\033[0m")
+        label, latitude, longitude = ipapi()
+    if label is False:
+        write_log("获取经纬度信息重试中...\033[97m2\033[0m")
+        label, latitude, longitude = freegeoip()
+    if label is False:
+        write_log("获取经纬度信息失败")
+    if label:
+        picture_name = http_day_and_night(latitude, longitude)
+        if not picture_name:
+            write_log("获取日出日落失败, 将计算昼夜")
+            picture_name = judging_day_and_night(latitude, longitude)
+        config["icon"] = (
+            f"https://raw.githubusercontent.com/gruel-zxz/podflow/main/Podflow_{picture_name}.png"
+        )