thebears/reolink_api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add proxy support back to recording snap()

Browse Source
This commit is contained in:
Karl Moos
2020-10-29 07:32:45 -05:00
parent ad08a8ce71
commit 00c00ecf04
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/recording.py
View File

@@ -41,10 +41,11 @@ class RecordingAPIMixin:
proxies={"host": "127.0.0.1", "port": 8000}) as rtsp_client: proxies={"host": "127.0.0.1", "port": 8000}) as rtsp_client:
rtsp_client.preview() rtsp_client.preview()
def get_snap(self, timeout: int = 3) -> Image or None: def get_snap(self, timeout: int = 3, proxies=None) -> Image or None:
""" """
Gets a "snap" of the current camera video data and returns a Pillow Image or None Gets a "snap" of the current camera video data and returns a Pillow Image or None
:param timeout: Request timeout to camera in seconds :param timeout: Request timeout to camera in seconds
:param proxies: http/https proxies to pass to the request object.
:return: Image or None :return: Image or None
""" """
data = {} data = {}
@@ -56,7 +57,7 @@ class RecordingAPIMixin:
parms = parse.urlencode(data).encode("utf-8") parms = parse.urlencode(data).encode("utf-8")
try: try:
response = requests.get(self.url, params=parms, timeout=timeout) response = requests.get(self.url, proxies=proxies, params=parms, timeout=timeout)
if response.status_code == 200: if response.status_code == 200:
return Image.open(BytesIO(response.content)) return Image.open(BytesIO(response.content))
print("Could not retrieve data from camera successfully. Status:", response.stats_code) print("Could not retrieve data from camera successfully. Status:", response.stats_code)