gpg autodetermined in python script

1 files changed, 6 insertions, 1 deletions

diff --git a/bin/imappwd.py b/bin/imappwd.py
index 9b25fcc..d630b4f 100755
--- a/bin/imappwd.py
+++ b/bin/imappwd.py
@@ -1,10 +1,15 @@
 import os.path
 import subprocess
+import distutils.spawn
 home = os.path.expanduser("~")
 def mailpasswd(acct):
   acct = os.path.basename(acct)
   path = "%s/.local/share/muttwizard/%s.gpg" % (home,acct)
-  args = ["gpg2", "--use-agent", "--quiet", "--batch", "-d", path]
+  if distutils.spawn.find_executable("gpg"):
+      GPG="gpg"
+  else:
+      GPG="gpg2"
+  args = [GPG, "--use-agent", "--quiet", "--batch", "-d", path]
   try:
     return subprocess.check_output(args).strip().decode('UTF-8')
   except subprocess.CalledProcessError: