diff --git a/scripts/drand.py b/scripts/drand.py new file mode 100644 index 0000000..e35bb05 --- /dev/null +++ b/scripts/drand.py @@ -0,0 +1,27 @@ +import os, os.path +import sys +from random import randint + +check_start = False +start = '' + +dirn = sys.argv[1] + +if len(sys.argv) > 2: + start = sys.argv[2] + check_start = True + +dirlist = os.listdir(dirn) + +if check_start: + fs = [] + for i in dirlist: + if i.startswith(start): + fs.append(i) + numitems = len(fs) + i = randint(0,numitems-1) + print(fs[i]) +else: + numitems = len(dirlist) + i = randint(0,numitems-1) + print(dirlist[i]) diff --git a/scripts/frand.py b/scripts/frand.py index 1511366..ec67901 100644 --- a/scripts/frand.py +++ b/scripts/frand.py @@ -1,10 +1,30 @@ -import os, os.path import sys from random import randint -dirn = sys.argv[1] -dirlist = os.listdir(dirn) -numitems = len(dirlist) -i = randint(0,numitems-1) -print(dirlist[i]) +check_start = False +start = '' +fn = sys.argv[1] + +if len(sys.argv) > 2: + start = sys.argv[2] + check_start = True + +lines = [] +with open(fn) as f: + for line in f: + nline = line.replace("\n", "") + lines.append(nline) + +if check_start: + line_sub = [] + for i in lines: + if i.startswith(start): + line_sub.append(i) + numitems = len(line_sub) + i = randint(0,numitems-1) + print(line_sub[i]) +else: + numitems = len(lines) + i = randint(0,numitems-1) + print(lines[i]) diff --git a/scripts/scripts.plugin.zsh b/scripts/scripts.plugin.zsh index 296a263..0f9fa0c 100644 --- a/scripts/scripts.plugin.zsh +++ b/scripts/scripts.plugin.zsh @@ -1,4 +1,5 @@ SDIR="~/.oh-my-zsh/custom/plugins/scripts" alias frand="python3 ${SDIR}/frand.py" +alias drand="python3 ${SDIR}/drand.py" alias randnums="python3 ${SDIR}/randomnums.py"