adding URL helper function

cinemascience · Nov 8, 2023 · 6f696d5 · 6f696d5
1 parent f0f626f
commit 6f696d5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/pycinema/Core.py b/pycinema/Core.py
@@ -44,6 +44,13 @@ def shape(self):
     def resolution(self):
         return self.shape[:2][::-1]
 
+def isURL(path):
+    s = path.strip()
+    if s.startswith("http") or s.startswith("HTTP"):
+        return True
+    else:
+        return False
+
 def isNumber(s):
     t = type(s)
     if t == int or t == float:

diff --git a/pycinema/filters/ImageReader.py b/pycinema/filters/ImageReader.py
@@ -1,4 +1,4 @@
-from pycinema import Filter, Image
+from pycinema import Filter, Image, isURL
 
 import PIL
 import numpy
@@ -73,7 +73,7 @@ def _update(self):
                 file.close()
 
             elif str.lower(extension) in ['png','jpg','jpeg']:
-                if "https" in path:
+                if isURL(path):
                     rawImage = PIL.Image.open(requests.get(path, stream=True).raw)
                 else:
                     rawImage = PIL.Image.open(path)