Added minlength and exactlength

banana.py

@@ -1,6 +1,6 @@
 import argparse
 
-def get_args():
+def dec2banana():
     parser = argparse.ArgumentParser(description="Create banana words")
     parser.add_argument("num", help="number to be converted", type=int)
     parser.add_argument("--dictionary", help="Set dictionary", type=list, nargs='+', default=[list("bcdfglmnprstvz"), list("aeiou")]) # , list("123456")
@@ -10,34 +10,45 @@ def get_args():
     parser.add_argument("--shiftend", help="Set shift for ending dictionary", type=int, default=0)
     
     args = parser.parse_args()
-    return args
 
-
-
-
-def main():
-
-    args = get_args()
     dictionary = args.dictionary
     numdict = len(dictionary)
     minlength = args.minlength
     
     v = args.num
     st = ""
-    
-    i = (numdict - 1 + args.dictstart + args.shiftend) % numdict
+    l = 0
 
-    while not (v == 0 and i == (numdict - 1 + args.dictstart) % numdict):
-        r = v % len(dictionary[i])
-        v = int(v / len(dictionary[i]))
-        st = dictionary[i][r] + st
-        i = (i - 1) % numdict
+    
 
+    if args.exactlength == 0:
+        i = (numdict - 1 + args.dictstart + args.shiftend) % numdict
+        while not (v == 0 and i == (numdict - 1 + args.dictstart) % numdict and l >= args.minlength):
+            r = v % len(dictionary[i])
+            v = int(v / len(dictionary[i]))
+            st = dictionary[i][r] + st
+            i = (i - 1) % numdict
+            l += 1
+    else:
+        i = (numdict - 1 + args.dictstart + args.shiftend + args.exactlength) % numdict
+        while not (l >= args.exactlength):
+            r = v % len(dictionary[i])
+            v = int(v / len(dictionary[i]))
+            st = dictionary[i][r] + st
+            i = (i - 1) % numdict
+            l += 1
+        if v > 0:
+            print("Error, result longer than exactlength")
+            return
+            
     print(st)
 
 
+   
+
+
 
 
 
 if __name__ == "__main__":
-    main()
+    dec2banana()