update scripts

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])

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])

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"