Skip to content

Commit 13890d7

Browse files
Check if DRM scheme is supported
1 parent 60505b6 commit 13890d7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

lib/inputstreamhelper.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,10 @@ def __init__(self, protocol, drm=None):
2424
raise self.InputStreamException('UnsupportedProtocol')
2525
else:
2626
self._inputstream_addon = config.INPUTSTREAM_PROTOCOLS[self.protocol]
27+
if self.drm and self.drm not in config.DRM_SCHEMES:
28+
raise self.InputStreamException('UnsupportedDRMScheme')
29+
else:
30+
self.drm = config.DRM_SCHEMES[self.drm]
2731

2832
class InputStreamException(Exception):
2933
pass

0 commit comments

Comments
 (0)