Move streaming dependancies to their own "extra"

reolinkapi/mixins/stream.py

@@ -5,48 +5,59 @@ from urllib import parse
 from io import BytesIO
 
 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:
-    """ API calls for opening a video stream or capturing an image from the camera."""
+    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:
-        """
-        'https://support.reolink.com/hc/en-us/articles/360007010473-How-to-Live-View-Reolink-Cameras-via-VLC-Media-Player'
-        Blocking function creates a generator and returns the frames as it is spawned
-        :param callback:
-        :param proxies: Default is none, example: {"host": "localhost", "port": 8000}
-        """
-        rtsp_client = RtspClient(
-            ip=self.ip, username=self.username, password=self.password, proxies=proxies, callback=callback)
-        return rtsp_client.open_stream()
+        def open_video_stream(self, callback: Any = None, proxies: Any = None) -> Any:
+            """
+            'https://support.reolink.com/hc/en-us/articles/360007010473-How-to-Live-View-Reolink-Cameras-via-VLC-Media-Player'
+            Blocking function creates a generator and returns the frames as it is spawned
+            :param callback:
+            :param proxies: Default is none, example: {"host": "localhost", "port": 8000}
+            """
+            rtsp_client = RtspClient(
+                ip=self.ip, username=self.username, password=self.password, proxies=proxies, callback=callback)
+            return rtsp_client.open_stream()
 
-    def get_snap(self, timeout: float = 3, proxies: Any = None) -> Optional[Image]:
-        """
-        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 proxies: http/https proxies to pass to the request object.
-        :return: Image or None
-        """
-        data = {
-            'cmd': 'Snap',
-            'channel': 0,
-            'rs': ''.join(random.choices(string.ascii_uppercase + string.digits, k=10)),
-            'user': self.username,
-            'password': self.password,
-        }
-        parms = parse.urlencode(data).encode("utf-8")
+        def get_snap(self, timeout: float = 3, proxies: Any = None) -> Optional[Image]:
+            """
+            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 proxies: http/https proxies to pass to the request object.
+            :return: Image or None
+            """
+            data = {
+                'cmd': 'Snap',
+                'channel': 0,
+                'rs': ''.join(random.choices(string.ascii_uppercase + string.digits, k=10)),
+                'user': self.username,
+                'password': self.password,
+            }
+            parms = parse.urlencode(data).encode("utf-8")
 
-        try:
-            response = requests.get(self.url, proxies=proxies, params=parms, timeout=timeout)
-            if response.status_code == 200:
-                return open_image(BytesIO(response.content))
-            print("Could not retrieve data from camera successfully. Status:", response.status_code)
-            return None
+            try:
+                response = requests.get(self.url, proxies=proxies, params=parms, timeout=timeout)
+                if response.status_code == 200:
+                    return open_image(BytesIO(response.content))
+                print("Could not retrieve data from camera successfully. Status:", response.status_code)
+                return None
 
-        except Exception as e:
-            print("Could not get Image data\n", e)
-            raise
+            except Exception as e:
+                print("Could not get Image data\n", e)
+                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]"''')

setup.py

@@ -26,13 +26,17 @@ AUTHOR_EMAIL = 'alanoterblanche@gmail.com'
 AUTHOR = 'Benehiko'
 LICENSE = 'GPL-3.0'
 INSTALL_REQUIRES = [
-    'numpy==1.19.4',
-    'opencv-python==4.4.0.46',
-    'Pillow==8.0.1',
     'PySocks==1.7.1',
     'PyYaml==5.3.1',
     '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__))
@@ -53,5 +57,6 @@ setup(
     url=URL,
     license=LICENSE,
     install_requires=INSTALL_REQUIRES,
-    packages=find_packages(exclude=['examples', 'tests'])
+    packages=find_packages(exclude=['examples', 'tests']),
+    extras_require=EXTRAS_REQUIRE
 )