stark821 commited on
Commit
8c1afab
·
unverified ·
1 Parent(s): 84faec3

Adding list data type to 'res' parameter of filter() function.

Browse files

Adding list data type in addition to the existing str type to the 'res' and 'resolution' parameter of the filter() function of the StreamQuery class. This enables us to streams with multiple resolutions in a single query.

Files changed (1) hide show
  1. pytube/query.py +4 -1
pytube/query.py CHANGED
@@ -123,7 +123,10 @@ class StreamQuery(Sequence):
123
  """
124
  filters = []
125
  if res or resolution:
126
- filters.append(lambda s: s.resolution == (res or resolution))
 
 
 
127
 
128
  if fps:
129
  filters.append(lambda s: s.fps == fps)
 
123
  """
124
  filters = []
125
  if res or resolution:
126
+ if isinstance(res, str) or isinstance(resolution, str):
127
+ filters.append(lambda s: s.resolution == (res or resolution))
128
+ elif isinstance(res, list) or isinstance(resolution, list):
129
+ filters.append(lambda s: s.resolution in (resolution or res))
130
 
131
  if fps:
132
  filters.append(lambda s: s.fps == fps)