rnhmjoj/pirate-get
1
0
Fork 0
mirror of https://github.com/vikstrous/pirate-get synced 2025-01-09 09:59:51 +01:00
Code Issues Releases Wiki Activity

Merge pull request #55 from jebaum/allow-ranges

Browse Source 
allow ranges of values to be entered
This commit is contained in:
Viktor Stanchev 2015-08-07 21:10:32 -07:00
commit 5f11463a0f
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pirate-get.py
View File

@ -623,8 +623,19 @@ def main():
# Turn into list
l = re.sub(r'^[hdfp, ]*|[hdfp, ]*$', '', l)
l = re.sub('[ ,]+', ',', l)
l = re.sub('[^0-9,]', '', l)
choices = l.split(',')
l = re.sub('[^0-9,-]', '', l)
parsed_input = l.split(',')
# expand ranges
choices = []
for elem in parsed_input: # loop will generate a list of lists
left, sep, right = elem.partition('-')
if right:
choices.append(list(range(int(left), int(right) + 1)))
else:
choices.append([int(left)])
choices = sum(choices, []) # flatten list
choices = [str(elem) for elem in choices] # the current code stores the choices as strings instead of ints. not sure if necessary
# Act on option, if supplied
print('')