增加爱奇艺链接方式解析

pfunc/cfunc.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3.7
+# coding=utf-8
+'''
+# 作者: weimo
+# 创建日期: 2020-01-05 12:45:18
+# 上次编辑时间: 2020-01-05 14:44:42
+# 一个人的命运啊,当然要靠自我奋斗,但是...
+'''
+from urllib.parse import urlparse
+
+
+def check_url_site(url):
+    return urlparse(url).netloc.split(".")[-2]
+
+def check_url_locale(url):
+    flag = {
+        "cn":"zh_cn",
+        "tw":"zh_tw",
+        "intl":"intl"
+    }
+    if urlparse(url).netloc.split(".")[0] == "tw":
+        return flag["tw"]
+    else:
+        return flag["cn"] 

pfunc/dump_to_ass.py

@@ -1,14 +1,27 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3.7
 # coding=utf-8
 '''
 # 作者: weimo
 # 创建日期: 2020-01-04 19:17:44
-# 上次编辑时间       : 2020-01-04 19:30:24
+# 上次编辑时间: 2020-01-05 14:45:03
 # 一个人的命运啊,当然要靠自我奋斗,但是...
 '''
-
 import os
 
+from basic.ass import get_ass_head, check_font
+from methods.assbase import ASS
+from methods.sameheight import SameHeight
+
+def write_one_video_subtitles(file_path, comments, args):
+    # 对于合集则每次都都得检查一次 也可以放在上一级 放在这里 考虑后面可能特殊指定字体的情况
+    font_path, font_style_name = check_font(args.font)
+    ass_head = get_ass_head(font_style_name, args.font_size)
+    get_xy_obj = SameHeight("那就写这一句作为初始化测试吧！", font_path=font_path, font_size=args.font_size)
+    subtitle = ASS(file_path, get_xy_obj, font=font_style_name)
+    for comment in comments:
+        subtitle.create_new_line(comment)
+    write_lines_to_file(ass_head, subtitle.lines, file_path)
+
 def write_lines_to_file(ass_head, lines, file_path):
     with open(file_path, "a+", encoding="utf-8") as f:
         f.write(ass_head + "\n")

pfunc/request_info.py

@@ -1,17 +1,18 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3.7
 # coding=utf-8
 '''
 # 作者: weimo
-# 创建日期: 2020-01-04 13:15:25
-# 上次编辑时间       : 2020-01-04 17:47:16
+# 创建日期: 2020-01-04 19:14:43
+# 上次编辑时间: 2020-01-05 14:47:16
 # 一个人的命运啊,当然要靠自我奋斗,但是...
 '''
-
 import re
 import json
 import requests
 
-from basic.vars import qqlive, iqiyiplayer
+from time import localtime
+from pfunc.cfunc import check_url_locale
+from basic.vars import qqlive, iqiyiplayer, chrome
 
 # 放一些仅通过某个id获取另一个/多个id的方法
 
@@ -26,7 +27,7 @@ def get_danmu_target_id_by_vid(vid: str):
     try:
         r = requests.get(api_url, params=params, headers=qqlive).content.decode("utf-8")
     except Exception as e:
-        print("error info -->", e)
+        print("target_id requests error info -->", e)
         return None
     data = json.loads(r.lstrip("QZOutputJson=").rstrip(";"))
     target_id = None
@@ -53,7 +54,7 @@ def get_all_vids_by_cid(cid):
     try:
         nomal_ids = json.loads(data["results"][0]["fields"]["nomal_ids"])
     except Exception as e:
-        print("error info -->", e)
+        print("load nomal_ids error info -->", e)
         return None
     # F 2是免费 7是会员 0是最新正片之前的预告 4是正片之后的预告
     vids = [item["V"] for item in nomal_ids if item["F"] in [2, 7]]
@@ -63,23 +64,132 @@ def get_all_vids_by_cid(cid):
 
 #-------------------------------------------iqiyi--------------------------------------------
 
-def get_vinfos(aid):
+def get_vinfos(aid, locale="zh_cn"):
     api_url = "http://cache.video.iqiyi.com/avlist/{}/0/".format(aid)
+    if locale != "zh_cn":
+        api_url += "?locale=" + locale
     try:
-        r = requests.get(api_url, headers=iqiyiplayer).content.decode("utf-8")
+        r = requests.get(api_url, headers=chrome, timeout=5).content.decode("utf-8")
     except Exception as e:
-        print("error info -->", e)
+        print("get_vinfos requests error info -->", e)
         return None
     data = json.loads(r[len("var videoListC="):])
     try:
         vlist = data["data"]["vlist"]
     except Exception as e:
-        print("error info -->", e)
+        print("get_vinfos load vlist error info -->", e)
         return None
     vinfos = [[v["shortTitle"] + "_" + str(v["timeLength"]), v["timeLength"], ["id"]] for v in vlist]
     return vinfos
 
+def matchit(patterns, text):
+    ret = None
+    for pattern in patterns:
+        match = re.match(pattern, text)
+        if match:
+            ret = match.group(1)
+            break
+    return ret
+
+def duration_to_sec(duration):
+    return sum(x * int(t) for x, t in zip([3600, 60, 1][2 - duration.count(":"):], duration.split(":")))
+
+def get_year_range(aid, locale="zh_cn"):
+    # 获取第一个和最新一个视频的年份，生成列表返回，遇到任何错误则返回当前年份
+    year_start = year_end = localtime().tm_year
+    api_url = "http://pcw-api.iqiyi.com/album/album/baseinfo/{}".format(aid)
+    if locale != "zh_cn":
+        api_url += "?locale=" + locale
+    try:
+        r = requests.get(api_url, headers=chrome, timeout=5).content.decode("utf-8")
+    except Exception as e:
+        print("error info -->", e)
+        return list(range(year_start, year_end + 1))
+    data = json.loads(r)["data"]
+    if data.get("firstVideo"):
+        year_start = int(data["firstVideo"]["period"][:4])
+    if data.get("latestVideo"):
+        year_end = int(data["latestVideo"]["period"][:4])
+    return list(range(year_start, year_end + 1))
+
+def get_vinfo_by_tvid(tvid, locale="zh_cn"):
+    api_url = "https://pcw-api.iqiyi.com/video/video/baseinfo/{}".format(tvid)
+    if locale != "zh_cn":
+        api_url += "?locale=" + locale
+    try:
+        r = requests.get(api_url, headers=chrome, timeout=5).content.decode("utf-8")
+    except Exception as e:
+        print("error info -->", e)
+        return
+    data = json.loads(r)["data"]
+    if data.__class__ != dict:
+        return None
+    name = data["name"]
+    duration = data["durationSec"]
+    return [name + "_" + str(duration), duration, tvid]
+
+def get_vinfos_by_year(aid, years: list, cid=6, locale="zh_cn"):
+    api_url = "https://pcw-api.iqiyi.com/album/source/svlistinfo?cid={}&sourceid={}&timelist={}".format(cid, aid, ",".join([str(_) for _ in years.copy()]))
+    if locale != "zh_cn":
+        api_url += "&locale=" + locale
+    try:
+        r = requests.get(api_url, headers=chrome, timeout=5).content.decode("utf-8")
+    except Exception as e:
+        print("get_vinfos_by_year error info -->", e)
+        return None
+    data = json.loads(r)["data"]
+    vinfos = []
+    for year in years:
+        if year.__class__ != str:
+            year = str(year)
+        if data.get(year) is None:
+            continue
+        for ep in data[year]:
+            sec = duration_to_sec(ep["duration"])
+            vinfos.append([ep["shortTitle"] + "_" + str(sec), sec, ep["tvId"]])
+    return vinfos
+
 def get_vinfos_by_url(url):
-    pass
+    locale = check_url_locale(url)
+    patterns = [".+?/w_(\w+?).html", ".+?/v_(\w+?).html", ".+?/a_(\w+?).html", ".+?/lib/m_(\w+?).html"]
+    isw, isep, isas, isms = [re.match(pattern, url) for pattern in patterns]
+    if isw is None and isep is None and isas is None and isms is None:
+        return None
+    try:
+        r = requests.get(url, headers=chrome, timeout=5).content.decode("utf-8")
+    except Exception as e:
+        print("get_vinfos_by_url error info -->", e)
+        return None
+    cid_patterns = ["[\s\S]+?\.cid.+?(\d+)", "[\s\S]+?cid: \"(\d+)\"", "[\s\S]+?channelID.+?\"(\d+)\""]
+    cid = matchit(cid_patterns, r)
+    aid_patterns = ["[\s\S]+?aid:'(\d+)'", "[\s\S]+?albumid=\"(\d+)\"", "[\s\S]+?movlibalbumaid=\"(\d+)\"", "[\s\S]+?data-score-tvid=\"(\d+)\""]
+    aid = matchit(aid_patterns, r)
+    tvid_patterns = ["[\s\S]+?\"tvid\":\"(\d+)\"", "[\s\S]+?\['tvid'\].+?\"(\d+)\""]
+    tvid = matchit(tvid_patterns, r)
+    if cid is None:
+        cid = ""
+    elif cid == "6" and isas or isms:#对于综艺合集需要获取年份
+        # year_patterns = ["[\s\S]+?datePublished.+?(\d\d\d\d)-\d\d-\d\d", "[\s\S]+?data-year=\"(\d+)\""]
+        # year = matchit(year_patterns, r)
+        # if year is None:
+        #     years = [localtime().tm_year]
+        # else:
+        #     years = [year]
+        years = get_year_range(aid, locale=locale)
+    else:
+        pass#暂时没有其他的情况计划特别处理
+
+    if isep or isw:
+        if tvid is None:
+            return
+        return get_vinfo_by_tvid(tvid, locale=locale)
+
+    if isas or isms:
+        if aid is None:
+            return
+        if cid == "6":
+            return get_vinfos_by_year(aid, years, locale=locale)
+        else:
+            return get_vinfos(aid, locale=locale)
 
 #-------------------------------------------iqiyi--------------------------------------------