更新 'main.py'

main.py

@@ -1,54 +1,56 @@
-import base64
+import base64
-import hashlib
+import hashlib
-import json
+import json
-import random
+import random
-import time
+import time
-from urllib import parse
+from urllib import parse
-from pathlib import Path
+from pathlib import Path
-from urllib.parse import parse_qsl, urlsplit
+from urllib.parse import parse_qsl, urlsplit
-from Crypto.Cipher import AES
+from Crypto.Cipher import AES
-from Crypto.Util.Padding import pad, unpad
+from Crypto.Util.Padding import pad, unpad
-from Cryptodome.Random import get_random_bytes
+from Cryptodome.Random import get_random_bytes
-from Cryptodome.Random import random
+from Cryptodome.Random import random
-from Cryptodome.Cipher import PKCS1_OAEP, AES
+from Cryptodome.Cipher import PKCS1_OAEP, AES
-from Cryptodome.Hash import CMAC, SHA256, HMAC, SHA1
+from Cryptodome.Hash import CMAC, SHA256, HMAC, SHA1
-from Cryptodome.PublicKey import RSA
+from Cryptodome.PublicKey import RSA
-from Cryptodome.Signature import pss
+from Cryptodome.Signature import pss
-from Cryptodome.Util import Padding
+from Cryptodome.Util import Padding
-from google.protobuf.message import DecodeError
+from google.protobuf.message import DecodeError
-from google.protobuf import text_format
+from google.protobuf import text_format
-import logging
+import logging
-import yaml
+import yaml
-from tabulate import tabulate
+from tabulate import tabulate
-from wasmer_compiler_cranelift import Compiler
+from wasmer_compiler_cranelift import Compiler
-from wasmer import Store, Type, Function, Memory, Module, ImportObject, engine, Instance, Table
+from wasmer import Store, Type, Function, Memory, Module, ImportObject, engine, Instance, Table
-from pywidevine.L3.cdm import deviceconfig
+from pywidevine.L3.cdm import deviceconfig
-from pywidevine.L3.decrypt.wvdecryptcustom import WvDecrypt
+from pywidevine.L3.decrypt.wvdecryptcustom import WvDecrypt
-import re, requests, time, json
+import re, requests, time, json
-from hashlib import md5
+from hashlib import md5
-import base64
+import base64
-from tools import rsa_dec, aes_decrypt, djb2Hash, b64decode, sha1withrsa, check_file, get_config
+from tools import rsa_dec, aes_decrypt, djb2Hash, b64decode, sha1withrsa, check_file, get_config
-from tx import TX
+
-from iqy import iqy
+from iqy import iqy
-from yk import YouKu
+from yk import YouKu
-
+
-if __name__ == '__main__':
+if __name__ == '__main__':
-    check_file()
+    check_file()
-    config = get_config()
+    config = get_config()
-    txck = config["txck"]
+  
-    yk = config["yk"]
+    yk = config["yk"]
-    aqy = config["aqy"]
+    aqy = config["aqy"]
-    tx = TX(txck)
+    try:
-    iq = iqy(aqy)
+        
-    youku = YouKu(yk)
+        iq = iqy(aqy)
-    while True:
+        youku = YouKu(yk)
-            url = input("请输入视频链接：")
+    except Exception as e:
-            if "v.qq.com" in url:
+        print("配置文件有误，请检查")
-                tx.run(url)
+        print(e)
-            elif "iqiyi.com" in url:
+    while True:
-                iq.run(url)
+            url = input("请输入视频链接：")
-            elif "youku.com" in url:
+            if "iqiyi.com" in url:
-                youku.start(url)
+                iq.run(url)
-            else:
+            elif "youku.com" in url:
-                print("暂不支持该链接")
+                youku.start(url)
-
+            else:
+                print("暂不支持该链接")
+