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

Move streaming dependancies to their own "extra"

Browse Source
This commit is contained in:
jdiamond
2021-03-30 23:04:18 +00:00
parent 5ac45d8b20
commit 8cd3405378
2 changed files with 58 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
reolinkapi/mixins/stream.py
View File

@@ -5,12 +5,14 @@ from urllib import parse
from io import BytesIO from io import BytesIO
import requests import requests
from PIL.Image import Image, open as open_image
from reolinkapi.utils.rtsp_client import RtspClient try:
from PIL.Image import Image, open as open_image
from reolinkapi.utils.rtsp_client import RtspClient
class StreamAPIMixin: class StreamAPIMixin:
""" API calls for opening a video stream or capturing an image from the camera.""" """ API calls for opening a video stream or capturing an image from the camera."""
def open_video_stream(self, callback: Any = None, proxies: Any = None) -> Any: def open_video_stream(self, callback: Any = None, proxies: Any = None) -> Any:
@@ -50,3 +52,12 @@ class StreamAPIMixin:
except Exception as e: except Exception as e:
print("Could not get Image data\n", e) print("Could not get Image data\n", e)
raise raise
except ImportError:
class StreamAPIMixin:
""" API calls for opening a video stream or capturing an image from the camera."""
def open_video_stream(self, callback: Any = None, proxies: Any = None) -> Any:
raise ImportError('''open_video_stream requires streaming extra dependencies\nFor instance "pip install reolinkapi[streaming]"''')
def get_snap(self, timeout: float = 3, proxies: Any = None) -> Optional['Image']:
raise ImportError('''open_video_stream requires streaming extra dependencies\nFor instance "pip install reolinkapi[streaming]"''')

13
setup.py
View File

@@ -26,13 +26,17 @@ AUTHOR_EMAIL = 'alanoterblanche@gmail.com'
AUTHOR = 'Benehiko' AUTHOR = 'Benehiko'
LICENSE = 'GPL-3.0' LICENSE = 'GPL-3.0'
INSTALL_REQUIRES = [ INSTALL_REQUIRES = [
'numpy==1.19.4',
'opencv-python==4.4.0.46',
'Pillow==8.0.1',
'PySocks==1.7.1', 'PySocks==1.7.1',
'PyYaml==5.3.1', 'PyYaml==5.3.1',
'requests>=2.18.4', 'requests>=2.18.4',
] ]
EXTRAS_REQUIRE = {
'streaming': [
'numpy==1.19.4',
'opencv-python==4.4.0.46',
'Pillow==8.0.1',
],
}
here = os.path.abspath(os.path.dirname(__file__)) here = os.path.abspath(os.path.dirname(__file__))
@@ -53,5 +57,6 @@ setup(
url=URL, url=URL,
license=LICENSE, license=LICENSE,
install_requires=INSTALL_REQUIRES, install_requires=INSTALL_REQUIRES,
packages=find_packages(exclude=['examples', 'tests']) packages=find_packages(exclude=['examples', 'tests']),
extras_require=EXTRAS_REQUIRE
) )