Merge branch 'master' of github.com:bamos/python-scripts

Author: bamos

python3/rank-writing.py

@@ -5,9 +5,21 @@
 from subprocess import Popen,PIPE
 
 parser = argparse.ArgumentParser()
+parser.add_argument('--aspell',action='store_true')
+parser.add_argument('--diction',action='store_true')
+parser.add_argument('--write-good',action='store_true')
 parser.add_argument('files',type=str, nargs='+')
 args = parser.parse_args()
 
+# If no arguments are provided, show everything.
+if not (args.aspell or args.diction or args.write_good):
+  args.aspell = args.diction = args.write_good = True
+
+programs = []
+if args.aspell: programs.append("aspell")
+if args.diction: programs.append("diction")
+if args.write_good: programs.append("write-good")
+
 def call(cmd,in_f=None):
   if in_f:
     with open(in_f,'r') as f:
@@ -19,34 +31,31 @@ def call(cmd,in_f=None):
     raise Exception("Error running {}".format(cmd))
   return out
 
-def getNumWriteGoodSuggestions(f_name):
-  out = call(["write-good",f_name])
-  return out.count("-------------")
-
-def getNumDictionSuggestions(f_name):
-  out = call(["diction","--suggest",f_name])
-  r = re.search("(\S*) phrases? in (\S*) sentences? found.",out)
-  if r:
-    if r.group(1) == "No":
-      return 0
-    else:
-      return int(r.group(1))
-
-def getNumAspellSuggestions(f_name):
-  out = call(["aspell","list"],in_f=f_name)
-  return len(out.split())
+def getNumSuggestions(program,f_name):
+  if program == "write-good":
+    out = call(["write-good",f_name])
+    return out.count("-------------")
+  elif program == "aspell":
+    out = call(["diction","--suggest",f_name])
+    r = re.search("(\S*) phrases? in (\S*) sentences? found.",out)
+    if r:
+      if r.group(1) == "No":
+        return 0
+      else:
+        return int(r.group(1))
+  elif program == "diction":
+    out = call(["aspell","list"],in_f=f_name)
+    return len(out.split())
+  else:
+    raise Exception("Unrecognized program: {}".format(program))
 
 def getSuggestions(f_name):
-  suggestions = [
-    getNumDictionSuggestions(f_name),
-    getNumWriteGoodSuggestions(f_name),
-    getNumAspellSuggestions(f_name)
-  ]
+  suggestions = list(map(lambda x: getNumSuggestions(x,f_name),programs))
   return (sum(suggestions), suggestions, f_name)
 
 for tup in sorted(map(getSuggestions,args.files),reverse=True):
   print("\n=== {} ===".format(tup[2]))
   print("  Total: {}".format(tup[0]))
-  print("  ├── diction: {}".format(tup[1][0]))
-  print("  ├── write-good: {}".format(tup[1][1]))
-  print("  └── aspell: {}".format(tup[1][2]))
+  div = ["├──"]*(len(programs)-1) + ["└──"]
+  for program_output in list(zip(div,programs,tup[1])):
+    print("  {} {}: {}".format(*program_output))