299 lines
12 KiB
Python
299 lines
12 KiB
Python
import os
|
||
import time
|
||
import json
|
||
from datetime import datetime
|
||
import sys
|
||
from framework.settings import BASE_DIR, PROJECT_UA, PROJECT_BUILT, PROJECT_VERSION, PROJECT_FLAG, PROJECT_ADMIN_START_TIMESTAMP
|
||
from app.utils.ZLMediaKitApi import ZLMediaKitApi
|
||
from app.utils.Config import Config
|
||
from app.utils.Logger import CreateLogger
|
||
from app.utils.Database import Database
|
||
from app.utils.GB28181SipServer import GB28181SipServer
|
||
|
||
# ========== 应用名常量 ==========
|
||
APP_NAME_LIVE = "live"
|
||
APP_NAME_RTP = "rtp"
|
||
from app.utils.LanguageUtils import LANG_UI_DICT, LANG_VIEWS_USE_LANG_T
|
||
from app.models import *
|
||
|
||
# BASE_DIR 是项目根目录（扁平化后 settings.py 的父父目录即根目录）
|
||
# BASE_PARENT_DIR = str(BASE_DIR) # 扁平化后 BASE_PARENT_DIR 与 BASE_DIR 相同，均在根目录
|
||
g_filepath_settings_json = os.path.join(BASE_DIR, "settings.json")
|
||
g_filepath_config_json = os.path.join(BASE_DIR, "config.json")
|
||
|
||
g_config = Config(filepath=g_filepath_config_json)
|
||
|
||
__log_dir = os.path.join(BASE_DIR, "log")
|
||
if not os.path.exists(__log_dir):
|
||
os.makedirs(__log_dir)
|
||
|
||
__log_name = "%s%s.log" % ("monitor", datetime.now().strftime("%Y%m%d-%H%M%S"))
|
||
g_logger = CreateLogger(filepath=os.path.join(__log_dir, __log_name),
|
||
is_show_console=False,
|
||
log_debug=g_config.logDebug)
|
||
|
||
g_logger.info("%s v%s,%s" % (PROJECT_UA, PROJECT_VERSION, PROJECT_FLAG))
|
||
g_logger.info(PROJECT_BUILT)
|
||
g_logger.info("g_filepath_config_json=%s" % g_filepath_config_json)
|
||
g_logger.info("config.json:%s" % g_config.getStr())
|
||
g_logger.info("logDebug=%d" % g_config.logDebug)
|
||
|
||
__argv_extend = sys.argv[1] if len(sys.argv) >= 2 else None
|
||
g_logger.info("argv_extend=%s" % str(__argv_extend))
|
||
|
||
g_zlm = ZLMediaKitApi(logger=g_logger, config=g_config)
|
||
g_database = Database(logger=g_logger)
|
||
|
||
__config_sip_server = g_config.sipServer
|
||
# SIP 对象仅在此构造；绑定端口由显式 ServiceManager 负责。
|
||
g_gb28181SipServer = GB28181SipServer(
|
||
server_ip=__config_sip_server.get("sipServerIp"),
|
||
server_port=__config_sip_server.get("sipServerPort"),
|
||
server_id=__config_sip_server.get("sipServerId"),
|
||
realm=__config_sip_server.get("sipServerRealm"),
|
||
password=__config_sip_server.get("sipServerPass"),
|
||
sip_server_timeout=__config_sip_server.get("sipServerTimeout"),
|
||
sip_server_expiry=__config_sip_server.get("sipServerExpiry"),
|
||
sip_transfer_mode=__config_sip_server.get("sipTransferMode"),
|
||
rtp_transfer_mode=__config_sip_server.get("rtpTransferMode"),
|
||
rtp_transfer_audio_type=__config_sip_server.get("rtpTransferAudioType"),
|
||
auto_invite_after_rec_cate_log=__config_sip_server.get("autoInviteAfterRecCateLog"),
|
||
admin_host=g_config.adminHost,
|
||
zlm=g_zlm,
|
||
logger=g_logger
|
||
)
|
||
|
||
g_pull_stream_types = [
|
||
{"id": 1, "name": "RTSP"},
|
||
{"id": 2, "name": "RTMP"},
|
||
{"id": 3, "name": "FLV"},
|
||
{"id": 4, "name": "HLS"},
|
||
{"id": 21, "name": "GB28181"},
|
||
{"id": 31, "name": "cRTSP"},
|
||
{"id": 32, "name": "cRTMP"}
|
||
]
|
||
|
||
def get_audio_types(lang='zh'):
|
||
"""返回音频类型"""
|
||
T = LANG_UI_DICT.get(lang, {})
|
||
result = []
|
||
__audio_types = [
|
||
{"type": 0, "name": "静音", "name_key": "audio_pull_mute"},
|
||
{"type": 1, "name": "原始音频", "name_key": "audio_pull_original"}
|
||
]
|
||
for audio_type in __audio_types:
|
||
result.append({
|
||
"type": audio_type["type"],
|
||
"name": T.get(audio_type["name_key"], audio_type["name"])
|
||
})
|
||
return result
|
||
|
||
g_session_key_user = "user"
|
||
g_session_key_captcha = "captcha"
|
||
|
||
|
||
def _bool_cfg(v):
|
||
if v is None:
|
||
return False
|
||
if isinstance(v, bool):
|
||
return v
|
||
if isinstance(v, (int, float)):
|
||
return v != 0
|
||
return str(v).strip().lower() in ("1", "true", "yes", "on")
|
||
|
||
|
||
class GlobalUtils(object):
|
||
|
||
@staticmethod
|
||
def addStreamProxy(stream, lang=None):
|
||
"""开启流代理（拉流到ZLM）"""
|
||
__ret = False
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "msg_unknown_error")
|
||
|
||
if stream.pull_stream_type in [1, 2, 3, 4]:
|
||
enable_rtmp = 1 if g_config.isEnableMediaProxyRtmp else 0
|
||
add_key, add_msg = g_zlm.addStreamProxy(app=stream.app,
|
||
name=stream.name,
|
||
origin_url=stream.pull_stream_url,
|
||
is_audio=stream.is_audio,
|
||
enable_rtmp=enable_rtmp)
|
||
if add_key:
|
||
__ret = True
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "stream_forward_enabled_success")
|
||
else:
|
||
__msg = add_msg
|
||
elif stream.pull_stream_type == 21:
|
||
__ret, __msg = g_gb28181SipServer.request_invite(client_id=stream.camera_device_id, channel_id=stream.name)
|
||
if __ret:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "stream_forward_enabled_success")
|
||
elif stream.pull_stream_type in [31, 32]:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_push_stream_hint")
|
||
else:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_protocol_not_supported")
|
||
return __ret, __msg
|
||
|
||
@staticmethod
|
||
def addAllStreamProxy(lang=None):
|
||
"""开启所有流代理"""
|
||
ret = False
|
||
msg = LANG_VIEWS_USE_LANG_T(lang, "msg_unknown_error")
|
||
|
||
online_streams = g_zlm.getMediaList()
|
||
online_stream_dict = {}
|
||
|
||
if len(online_streams) == 0:
|
||
g_database.execute("update av_stream set forward_state=0")
|
||
else:
|
||
for d in online_streams:
|
||
an = "{app}_{name}".format(app=d["app"], name=d["name"])
|
||
online_stream_dict[an] = d
|
||
|
||
success_count = 0
|
||
error_count = 0
|
||
streams = StreamModel.objects.all()
|
||
for stream in streams:
|
||
stream_an = "{app}_{name}".format(app=stream.app, name=stream.name)
|
||
if online_stream_dict.get(stream_an):
|
||
success_count += 1
|
||
else:
|
||
__add_ret, __add_msg = GlobalUtils.addStreamProxy(stream, lang=lang)
|
||
if __add_ret:
|
||
stream.forward_state = 1
|
||
stream.save()
|
||
success_count += 1
|
||
else:
|
||
error_count += 1
|
||
ret = True
|
||
msg = LANG_VIEWS_USE_LANG_T(lang, "msg_batch_result") % (success_count, error_count)
|
||
return ret, msg
|
||
|
||
@staticmethod
|
||
def delStreamProxy(stream, lang=None):
|
||
"""关闭流代理"""
|
||
__ret = False
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "msg_unknown_error")
|
||
|
||
if stream.pull_stream_type in [1, 2, 3, 4]:
|
||
del_flag, del_msg = g_zlm.delStreamProxy(app=stream.app, name=stream.name)
|
||
if del_flag:
|
||
__ret = True
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_forward_stop_success")
|
||
else:
|
||
__msg = del_msg
|
||
elif stream.pull_stream_type == 21:
|
||
__ret, __msg = g_gb28181SipServer.request_bye(client_id=stream.camera_device_id, channel_id=stream.name)
|
||
if __ret:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_forward_stop_success")
|
||
elif stream.pull_stream_type == 31:
|
||
__ret, __msg = g_zlm.close_streams(schema="rtsp", app=stream.app, name=stream.name)
|
||
if __ret:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_forward_stop_success")
|
||
elif stream.pull_stream_type == 32:
|
||
__ret, __msg = g_zlm.close_streams(schema="rtmp", app=stream.app, name=stream.name)
|
||
if __ret:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_forward_stop_success")
|
||
else:
|
||
__msg = LANG_VIEWS_USE_LANG_T(lang, "media_protocol_not_supported")
|
||
return __ret, __msg
|
||
|
||
@staticmethod
|
||
def delAllStreamProxy(lang=None):
|
||
"""关闭所有流代理"""
|
||
online_streams = g_zlm.getMediaList()
|
||
for d in online_streams:
|
||
stream = StreamModel.objects.filter(app=d["app"], name=d["name"]).first()
|
||
if stream:
|
||
__ret, __msg = GlobalUtils.delStreamProxy(stream, lang=lang)
|
||
g_database.execute("update av_stream set forward_state=0")
|
||
|
||
|
||
@staticmethod
|
||
def apply_runtime_config(before_cfg):
|
||
"""保存 config.json 后热更新运行时组件（不重启 Django 进程）。"""
|
||
before_cfg = before_cfg or {}
|
||
after_cfg = g_config.to_dict(include_secrets=True)
|
||
notes = []
|
||
|
||
def _sip_snapshot(cfg):
|
||
sip = dict((cfg or {}).get("sipServer") or {})
|
||
return (
|
||
sip.get("sipServerIp"), sip.get("sipServerPort"), sip.get("sipServerId"),
|
||
sip.get("sipServerRealm"), sip.get("sipServerPass"), sip.get("sipServerTimeout"),
|
||
sip.get("sipServerExpiry"), sip.get("sipTransferMode"), sip.get("rtpTransferMode"),
|
||
sip.get("rtpTransferAudioType"), sip.get("autoInviteAfterRecCateLog"),
|
||
)
|
||
|
||
if _sip_snapshot(before_cfg) != _sip_snapshot(after_cfg):
|
||
try:
|
||
from app.services.lifecycle import is_service_leader
|
||
if is_service_leader():
|
||
GlobalUtils._reload_gb28181_sip()
|
||
g_logger.info("apply_runtime_config: GB28181 SIP reloaded")
|
||
else:
|
||
notes.append("gb28181")
|
||
except Exception as e:
|
||
g_logger.warning("apply_runtime_config: GB28181 reload failed: %s", e)
|
||
notes.append("gb28181")
|
||
|
||
before_rec = _bool_cfg(before_cfg.get("recordingEnabled"))
|
||
after_rec = bool(g_config.recordingEnabled)
|
||
if after_rec != before_rec:
|
||
try:
|
||
from app.services.lifecycle import get_service_manager
|
||
get_service_manager().reconcile()
|
||
except Exception as e:
|
||
g_logger.warning("apply_runtime_config: recording manager: %s", e)
|
||
|
||
if before_cfg.get("adminPort") != after_cfg.get("adminPort"):
|
||
notes.append("adminPort")
|
||
if _bool_cfg(before_cfg.get("logDebug")) != bool(g_config.logDebug):
|
||
notes.append("logDebug")
|
||
|
||
media_keys = (
|
||
"mediaHttpPort", "mediaRtspPort", "mediaRtmpPort",
|
||
"mediaStartPath", "mediaStartConfigPath", "mediaSecret",
|
||
)
|
||
if any(before_cfg.get(k) != after_cfg.get(k) for k in media_keys):
|
||
notes.append("zlm")
|
||
|
||
return notes
|
||
|
||
@staticmethod
|
||
def _reload_gb28181_sip():
|
||
global g_gb28181SipServer
|
||
try:
|
||
g_gb28181SipServer.stop()
|
||
except Exception:
|
||
pass
|
||
sip = dict(g_config.sipServer or {})
|
||
g_gb28181SipServer = GB28181SipServer(
|
||
server_ip=sip.get("sipServerIp"),
|
||
server_port=sip.get("sipServerPort"),
|
||
server_id=sip.get("sipServerId"),
|
||
realm=sip.get("sipServerRealm"),
|
||
password=sip.get("sipServerPass"),
|
||
sip_server_timeout=sip.get("sipServerTimeout"),
|
||
sip_server_expiry=sip.get("sipServerExpiry"),
|
||
sip_transfer_mode=sip.get("sipTransferMode"),
|
||
rtp_transfer_mode=sip.get("rtpTransferMode"),
|
||
rtp_transfer_audio_type=sip.get("rtpTransferAudioType"),
|
||
auto_invite_after_rec_cate_log=sip.get("autoInviteAfterRecCateLog"),
|
||
admin_host=g_config.adminHost,
|
||
zlm=g_zlm,
|
||
logger=g_logger,
|
||
)
|
||
g_gb28181SipServer.start()
|
||
|
||
|
||
class CheckServerUtils():
|
||
@staticmethod
|
||
def checkVersion(request_ip, peer_ip, peer_port, lang=None):
|
||
# Network and host information parameters are intentionally ignored.
|
||
from app.services.telemetry import check_update
|
||
return check_update(lang=lang)
|
||
|
||
@staticmethod
|
||
def reportHeart(report_count, lang=None):
|
||
from app.services.telemetry import send_heartbeat
|
||
return send_heartbeat(report_count)
|