anime-dl
/
anime-downloader
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve regex and add more conditions to statement

master
AbdullahM0hamed 2021-02-14 23:12:56 +00:00
parent 55cbcf25d4
commit 97adb94b7d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
anime_downloader/commands/dl.py
View File

@ -88,8 +88,12 @@ def command(ctx, anime_url, episode_range, url, player, skip_download, quality,
""" Download the anime using the url or search for it.
"""
if episode_range and not re.compile("^[0-9:]+$").search(episode_range):
raise UsageError("Invalid value for '--episode' / '-e': {} is not a valid range".format(episode_range))
if episode_range:
regexed_range = re.compile("^:?(\d+)?:?(\d+)?$").search(episode_range)
# Prevent such cases as: :5: and :1:1
if not regexed_range or (len(regexed_range.groups()) >= episode_range.count(":") and episode_range.count(":") != 1):
raise UsageError(
"Invalid value for '--episode' / '-e': {} is not a valid range".format(episode_range))
query = anime_url[:]