54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import requests
|
|
import urllib3
|
|
import yaml
|
|
from hikvision_openapi_signer import HikvisionOpenAPISigner
|
|
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
|
|
# 初始化签名器
|
|
signer = HikvisionOpenAPISigner(
|
|
"https://29.2.19.19:443",
|
|
"28031410",
|
|
"Y79rv6LqBahxoEH6adLv",
|
|
headers={'tagId': '0'} # 根据平台要求设置
|
|
)
|
|
|
|
def get_camera_preview_url(camera_index_code):
|
|
# 签名一个请求
|
|
request = signer.sign(
|
|
'POST',
|
|
'/api/video/v1/cameras/previewURLs',
|
|
jsons={'cameraIndexCode': camera_index_code, 'protocol': 'rtsp', 'expand': 'streamform=rtp'},
|
|
accept='application/json'
|
|
)
|
|
method, url, headers, body = request
|
|
# 发送请求
|
|
response = requests.request(method, url, headers=headers, data=body, verify=False)
|
|
return response.json()
|
|
|
|
# 读取 config.yaml
|
|
with open('config.yaml', 'r', encoding='utf-8') as f:
|
|
config = yaml.safe_load(f)
|
|
|
|
# 遍历所有摄像头
|
|
for camera in config['cameras']:
|
|
if 'index' in camera:
|
|
index = camera['index']
|
|
print(f"正在获取摄像头 {camera['name']} (index: {index}) 的预览地址...")
|
|
result = get_camera_preview_url(index)
|
|
print(f"API返回结果: {result}")
|
|
|
|
# 提取 url 并更新到 config
|
|
if 'data' in result and 'url' in result['data']:
|
|
rtsp_url = result['data']['url']
|
|
camera['rtsp_url'] = rtsp_url
|
|
print(f"更新 rtsp_url: {rtsp_url}")
|
|
else:
|
|
print(f"未找到 url 在返回结果中")
|
|
|
|
# 保存更新后的 config.yaml
|
|
with open('config.yaml', 'w', encoding='utf-8') as f:
|
|
yaml.dump(config, f, default_flow_style=False, allow_unicode=True)
|
|
|
|
print("config.yaml 已更新")
|