administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 01:35:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Nick Mowen 2022-10-09 12:21:41 -06:00
parent 448501fe22
commit 3754f66ace
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/test/test_camera_pw_clean.py
View File

@ -6,12 +6,10 @@ from frigate.util import clean_camera_user_pass
class TestUserPassCleanup(unittest.TestCase): class TestUserPassCleanup(unittest.TestCase):
def setUp(self) -> None: def setUp(self) -> None:
self.rtsp_with_pass = "rtsp://user:password@192.168.0.2:554/live" self.rtsp_with_pass = "rtsp://user:password@192.168.0.2:554/live"
self.rtsp_no_pass = "rtsp://192.168.0.3/live" self.rtsp_no_pass = "rtsp://192.168.0.3/live"
def test_cleanup(self): def test_cleanup(self):
"""Test that user / pass are cleaned up.""" """Test that user / pass are cleaned up."""
clean = clean_camera_user_pass(self.rtsp_with_pass) clean = clean_camera_user_pass(self.rtsp_with_pass)

1
frigate/util.py
View File

@ -627,6 +627,7 @@ def load_labels(path, encoding="utf-8"):
else: else:
return {index: line.strip() for index, line in enumerate(lines)} return {index: line.strip() for index, line in enumerate(lines)}
def clean_camera_user_pass(line: str) -> str: def clean_camera_user_pass(line: str) -> str:
"""Removes user and password from line.""" """Removes user and password from line."""
return re.sub(REGEX_CAMERA_USER_PASS, "*:*@", line) return re.sub(REGEX_CAMERA_USER_PASS, "*:*@", line)