fix: replace requests with httpx

This commit is contained in:
abc1763613206
2025-02-14 00:23:35 +08:00
parent 506e30af24
commit af3e85e11b

25
main.py
View File

@@ -4,7 +4,7 @@ import json
import os
import re
import traceback
import requests
import httpx
import subprocess
import time
import shutil
@@ -14,7 +14,7 @@ from sys import stdout
from termcolor import colored, RESET
from datetime import datetime
from func_timeout import func_set_timeout, FunctionTimedOut
from requests.adapters import HTTPAdapter
dt = datetime.now()
# Channel Group Source Link Description
# Description 应当对该源的已知参数进行标注如码率HDR
@@ -47,19 +47,16 @@ def get_stream(num, clist, uri):
def check_channel(clist, num):
# clist 为一行 csv
uri = clist[3]
requests.adapters.DEFAULT_RETRIES = 3
try:
r = requests.get(clist[3], timeout=1) # 先测能不能正常访问
if (r.status_code == requests.codes.ok):
# ffprobe = FFprobe(inputs={uri: '-v warning'})
# errors = tuple(filter(
# lambda line: not (line in ('', RESET) or any(regex.search(line) for regex in SKIP_FFPROBE_MESSAGES)),
# ffprobe.run(stderr=PIPE)[1].decode('utf-8').split('\n')
# ))
# if errors: # https://github.com/Jamim/iptv-checker/blob/master/iptv-checker.py#L26
# print('[{}] {}({}) Error:{}'.format(str(num), clist[0], clist[2], str(errors)))
# return False
# else: # 查视频信息
with httpx.Client(timeout=0.5) as client:
ReqStatus = False
try:
r = client.get(clist[3], follow_redirects=True)
if r.status_code == 200:
ReqStatus = True
except httpx.UnsupportedProtocol:
ReqStatus = True
if ReqStatus:
cdata = get_stream(num, clist, uri)
if cdata:
flagAudio = 0