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

copy d341813e62

Browse Source
This commit is contained in:
Viktor Stanchev 2015-08-29 22:22:45 -07:00
parent ecbd264d02
commit f60e0bec3f
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pirate/pirate.py
View File

@ -235,8 +235,20 @@ 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('')