#!/usr/bin/env python
-# -*- mode: python; coding: utf-8-unix; -*-
+# -*- mode: python; coding: utf-8; -*-
-import argparse, errno, os, re, readline, subprocess, sys, tempfile, textwrap, threading
+import argparse, errno, gettext, itertools, os, re, readline, subprocess, \
+ sys, tempfile, textwrap, threading
if sys.version_info[0] == 2:
from ScrolledText import *
from Queue import *
+
def input_string (prompt=""):
return raw_input (prompt)
+
+
+ def gettext_install ():
+ gettext.install ("crypto-install", unicode = True)
elif sys.version_info[0] > 2:
from tkinter import *
from tkinter.messagebox import *
from tkinter.scrolledtext import *
from queue import *
+
def input_string (prompt=""):
return input (prompt)
+
+
+ def gettext_install ():
+ gettext.install ("crypto-install")
else:
raise Exception ("Unsupported Python version {}".format (sys.version_info))
"-v", "--version",
dest = "version",
action = "version",
- version = "crypto-install.py version GIT-TAG (GIT-COMMIT/GIT-BRANCH)",
- help = "Display version.")
+ version = "crypto-install version GIT-TAG (GIT-COMMIT/GIT-BRANCH)",
+ help = _ ("Display version."))
parser.add_argument (
"--no-gui",
dest = "gui",
action = "store_false",
- help = "Disable GUI, use text mode.")
+ help = _ ("Disable GUI, use text interface."))
gnupg_group = parser.add_argument_group (
- "GnuPG",
- "Options related to the GnuPG setup.")
+ _ ("GnuPG"),
+ _ ("Options related to the GnuPG setup."))
gnupg_group.add_argument (
"--no-gpg",
dest = "gnupg",
action = "store_false",
- help = "Disable GnuPG setup.")
+ help = _ ("Disable GnuPG setup."))
gnupg_group.add_argument (
"--gpg-home",
dest = "gnupg_home",
default = os.getenv("GNUPGHOME") or "~/.gnupg",
- metavar = "PATH",
- help = "Default directory for GnuPG files.")
+ metavar = _ ("PATH"),
+ help = _ ("Default directory for GnuPG files."))
openssh_group = parser.add_argument_group (
- "OpenSSH",
- "Options related to the OpenSSH setup.")
+ _ ("OpenSSH"),
+ _ ("Options related to the OpenSSH setup."))
openssh_group.add_argument (
"--no-ssh",
dest = "openssh",
action = "store_false",
- help = "Disable OpenSSH setup.")
+ help = _ ("Disable OpenSSH setup."))
openssh_group.add_argument (
"--ssh-home",
dest = "openssh_home",
default = "~/.ssh",
- metavar = "PATH",
- help = "Default directory for OpenSSH files.")
+ metavar = _ ("PATH"),
+ help = _ ("Default directory for OpenSSH files."))
return parser.parse_args ()
return value.strip () != ""
+def valid_user (value):
+ return value.strip () != ""
+
+
+def valid_host (value):
+ return value.strip () != ""
+
+
def valid_comment (value):
return True
return ""
+def redirect_to_stdout (process):
+ # TODO: argh. there has to be a better way
+ process.stdin.close ()
+ while process.poll () is None:
+ sys.stdout.write (process.stdout.readline ())
+
+ while True:
+ line = process.stdout.readline ()
+ if len (line) == 0:
+ break
+ sys.stdout.write (line.decode ("UTF-8"))
+
+
def gnupg_setup (arguments, name = None, email = None, comment = None):
gnupg_home = os.path.expanduser (arguments.gnupg_home)
gnupg_secring = os.path.join (gnupg_home, "secring.gpg")
if gnupg_exists (arguments):
- print ("GnuPG secret keyring already exists at '{}'."
+ print (_ ("GnuPG secret keyring already exists at '{}'.")
.format (gnupg_secring))
return
if not arguments.gui:
- print (filled ("""
+ print (filled (_ ("""
No default GnuPG key available. Please enter your information to
- create a new key."""))
+ create a new key.""")))
- name = read_input_string ("What is your name (e.g. 'John Doe')? ",
+ name = read_input_string (_ ("What is your name (e.g. 'John Doe')? "),
default_name ())
- email = read_input_string (dedented ("""
- What is your email address (e.g. 'test@example.com')? """),
+ email = read_input_string (dedented (_ ("""
+ What is your email address (e.g. 'test@example.com')? """)),
default_email ())
- comment = read_input_string (dedented ("""
- What is your comment phrase, if any (e.g. 'key for 2014')? """),
+ comment = read_input_string (dedented (_ ("""
+ What is your comment phrase, if any (e.g. 'key for 2014')? """)),
default_comment ())
if not os.path.exists (gnupg_home):
- print ("Creating GnuPG directory at '{}'.".format (gnupg_home))
+ print (_ ("Creating GnuPG directory at '{}'.").format (gnupg_home))
ensure_directories (gnupg_home, 0o700)
with tempfile.NamedTemporaryFile () as tmp:
stderr = subprocess.STDOUT,
env = batch_env)
- # TODO: argh. there has to be a better way
- gnupg_process.stdin.close ()
- while gnupg_process.poll () is None:
- sys.stdout.write (gnupg_process.stdout.readline ())
-
- while True:
- line = gnupg_process.stdout.readline ()
- if len (line) == 0:
- break
- sys.stdout.write (line.decode ("UTF-8"))
+ redirect_to_stdout (gnupg_process)
if gnupg_process.returncode != 0:
raise Exception ("Couldn't create GnuPG key.")
openssh_config = os.path.join (openssh_home, "config")
if not os.path.exists (openssh_config):
- print ("Creating OpenSSH directory at '{}'.".format (openssh_home))
+ print (_ ("Creating OpenSSH directory at '{}'.").format (openssh_home))
ensure_directories (openssh_home, 0o700)
- print ("Creating OpenSSH configuration at '{}'."
+ print (_ ("Creating OpenSSH configuration at '{}'.")
.format (openssh_config))
with open (openssh_config, "w") as config:
config.write (ldedented ("""
openssh_key = os.path.join (openssh_home, "id_rsa")
if os.path.exists (openssh_key):
- print ("OpenSSH key already exists at '{}'.".format (openssh_key))
+ print (_ ("OpenSSH key already exists at '{}'.").format (openssh_key))
return
openssh_key_dsa = os.path.join (openssh_home, "id_dsa")
if os.path.exists (openssh_key_dsa):
- print ("OpenSSH key already exists at '{}'.".format (openssh_key_dsa))
+ print (_ ("OpenSSH key already exists at '{}'.").format (openssh_key_dsa))
return
- print (filled ("No OpenSSH key available. Generating new key."))
+ print (filled (_ ("No OpenSSH key available. Generating new key.")))
if not arguments.gui:
comment = "{}@{}".format (default_username (), default_hostname ())
- comment = read_input_string (ldedented ("""
- What is your comment phrase (e.g. 'user@mycomputer')? """), comment)
+ comment = read_input_string (ldedented (_ ("""
+ What is your comment phrase (e.g. 'user@mycomputer')? """)), comment)
passphrase = input_passphrase (arguments)
stderr = subprocess.STDOUT,
env = batch_env)
- # TODO: argh. there has to be a better way
- openssh_process.stdin.close ()
- while openssh_process.poll () is None:
- sys.stdout.write (openssh_process.stdout.readline ())
-
- while True:
- line = openssh_process.stdout.readline ()
- if len (line) == 0:
- break
- sys.stdout.write (line.decode ("UTF-8"))
+ redirect_to_stdout (openssh_process)
if openssh_process.returncode != 0:
raise Exception ("Couldn't create OpenSSH key.")
+def _state (value):
+ return NORMAL if value else DISABLED
+
+
+def _valid (value):
+ return "black" if value else "red"
+
+
+def setitem (object, name, value):
+ return object.__setitem__ (name, value)
+
+
+def setitems (name, value, objects):
+ return map (lambda object: setitem (object, name, value), objects)
+
+
+def _fg (value, *objects):
+ setitems ("fg", value, objects)
+
+
# http://www.blog.pythonlibrary.org/2014/07/14/tkinter-redirecting-stdout-stderr/
# http://www.virtualroadside.com/blog/index.php/2012/11/10/glib-idle_add-for-tkinter-in-python/
class RedirectText (object):
self.redirect = RedirectText (self.parent, self.text)
- self._quit = Button (self)
- self._quit["text"] = "Quit"
- self._quit["command"] = self.quit
+ self._quit = Button (self,
+ text = _ ("Quit"),
+ command = self.quit)
self.balloon.bind_widget (self._quit,
- msg = "Quit the program immediately")
+ msg = _ ("Quit the program immediately"))
self._quit.pack ()
self.arguments = arguments
self.resizable (width = False, height = False)
- self.title ("Crypto Install Wizard")
+ self.title (_ ("Crypto Install Wizard"))
self.create_widgets ()
def create_widgets (self):
+ self.fields = {}
+
self.balloon = Balloon (self, initwait = 250)
self.info_frame = Frame (self)
self.info_frame.pack (fill = X)
- self.user_label = Label (self.info_frame)
- self.user_label["text"] = "Username"
+ msg = dedented (_ ("""
+ Username on the local machine (e.g. 'user')
+ """))
+ self.user_label = Label (self.info_frame, text = _ ("Username"))
+ self.balloon.bind_widget (self.user_label, msg = msg)
self.user_label.grid ()
- self.user_var = StringVar ()
- self.user_var.set (default_username ())
+ self.user_var = StringVar (self, default_username ())
self.user_var.trace ("w", self.update_widgets)
- self.user = Entry (self.info_frame, textvariable = self.user_var,
- state = DISABLED)
- self.balloon.bind_widget (self.user, msg = dedented ("""
- Username on the local machine (e.g. 'user')
- """))
+ self.user = Entry (self.info_frame, textvariable = self.user_var)
+ self.balloon.bind_widget (self.user, msg = msg)
self.user.grid (row = 0, column = 1)
- self.host_label = Label (self.info_frame)
- self.host_label["text"] = "Host Name"
+ self.fields["user"] = [self.user_var, valid_user,
+ self.user, self.user_label]
+
+ msg = dedented (_ ("""
+ Host name of the local machine (e.g. 'mycomputer')
+ """))
+ self.host_label = Label (self.info_frame, text = _ ("Host Name"))
+ self.balloon.bind_widget (self.host_label, msg = msg)
self.host_label.grid ()
- self.host_var = StringVar ()
- self.host_var.set (default_hostname ())
+ self.host_var = StringVar (self, default_hostname ())
self.host_var.trace ("w", self.update_widgets)
- self.host = Entry (self.info_frame, textvariable = self.host_var,
- state = DISABLED)
- self.balloon.bind_widget (self.host, msg = dedented ("""
- Host name of the local machine (e.g. 'mycomputer')
- """))
+ self.host = Entry (self.info_frame, textvariable = self.host_var)
+ self.balloon.bind_widget (self.host, msg = msg)
self.host.grid (row = 1, column = 1)
- self.name_label = Label (self.info_frame)
- self.name_label["text"] = "Full Name"
+ self.fields["host"] = [self.host_var, valid_host,
+ self.host, self.host_label]
+
+ msg = dedented (_ ("""
+ Full name as it should appear in the key description (e.g. 'John Doe')
+ """))
+ self.name_label = Label (self.info_frame, text = _ ("Full Name"))
+ self.balloon.bind_widget (self.name_label, msg = msg)
self.name_label.grid ()
- self.name_var = StringVar ()
- self.name_var.set (default_name ())
+ self.name_var = StringVar (self, default_name ())
self.name_var.trace ("w", self.update_widgets)
self.name = Entry (self.info_frame, textvariable = self.name_var)
- self.balloon.bind_widget (self.name, msg = dedented ("""
- Full name as it should appear in the key description (e.g. 'John Doe')
- """))
+ self.balloon.bind_widget (self.name, msg = msg)
self.name.grid (row = 2, column = 1)
- self.email_label = Label (self.info_frame)
- self.email_label["text"] = "Email address"
+ self.fields["name"] = [self.name_var, valid_name,
+ self.name, self.name_label]
+
+ msg = dedented (_ ("""
+ Email address associated with the name (e.g. '<test@example.com>')
+ """))
+ self.email_label = Label (self.info_frame, text = _ ("Email address"))
+ self.balloon.bind_widget (self.email_label, msg = msg)
self.email_label.grid ()
- self.email_var = StringVar ()
- self.email_var.set (default_email ())
+ self.email_var = StringVar (self, default_email ())
self.email_var.trace ("w", self.update_widgets)
self.email = Entry (self.info_frame, textvariable = self.email_var)
- self.balloon.bind_widget (self.email, msg = dedented ("""
- Email address associated with the name (e.g. '<test@example.com>')
- """))
+ self.balloon.bind_widget (self.email, msg = msg)
self.email.grid (row = 3, column = 1)
- self.comment_label = Label (self.info_frame)
- self.comment_label["text"] = "Comment phrase"
+ self.fields["email"] = [self.email_var, valid_email,
+ self.email, self.email_label]
+
+ msg = dedented (_ ("""
+ Comment phrase for the GnuPG key, if any (e.g. 'key for 2014')
+ """))
+ self.comment_label = Label (self.info_frame, text = _ ("Comment phrase"))
+ self.balloon.bind_widget (self.comment_label, msg = msg)
self.comment_label.grid ()
- self.comment_var = StringVar ()
- self.comment_var.set (default_comment ())
+ self.comment_var = StringVar (self, default_comment ())
self.comment_var.trace ("w", self.update_widgets)
self.comment = Entry (self.info_frame, textvariable = self.comment_var)
- self.balloon.bind_widget (self.comment, msg = dedented ("""
- Comment phrase for the GnuPG key, if any (e.g. 'key for 2014')
- """))
+ self.balloon.bind_widget (self.comment, msg = msg)
self.comment.grid (row = 4, column = 1)
+ self.fields["comment"] = [self.comment_var, valid_comment,
+ self.comment, self.comment_label]
+
self.options_frame = Frame (self)
self.options_frame.pack (fill = X)
- self.gnupg_label = Label (self.options_frame)
- self.gnupg_label["text"] = "Generate GnuPG key"
+ self.gnupg_label = Label (self.options_frame,
+ text = _ ("Generate GnuPG key"))
self.gnupg_label.grid ()
- self.gnupg_var = IntVar ()
- self.gnupg_var.set (1 if self.arguments.gnupg else 0)
+ self.gnupg_var = IntVar (self, 1 if self.arguments.gnupg else 0)
self.gnupg_var.trace ("w", self.update_widgets)
self.gnupg = Checkbutton (self.options_frame,
variable = self.gnupg_var)
self.gnupg.grid (row = 0, column = 1)
- self.openssh_label = Label (self.options_frame)
- self.openssh_label["text"] = "Generate OpenSSH key"
+ self.openssh_label = Label (self.options_frame,
+ text = _ ("Generate OpenSSH key"))
self.openssh_label.grid ()
- self.openssh_var = IntVar ()
- self.openssh_var.set (1 if self.arguments.openssh else 0)
+ self.openssh_var = IntVar (self, 1 if self.arguments.openssh else 0)
self.openssh_var.trace ("w", self.update_widgets)
self.openssh = Checkbutton (self.options_frame,
self.button_frame = Frame (self)
self.button_frame.pack (fill = X)
- self._generate = Button (self.button_frame)
- self._generate["text"] = "Generate Keys"
- self._generate["command"] = self.generate
+ self._generate = Button (self.button_frame, text = _ ("Generate Keys"),
+ command = self.generate)
self.balloon.bind_widget (
self._generate,
- msg = "Generate the keys as configured above")
+ msg = _ ("Generate the keys as configured above"))
self._generate.pack (side = LEFT, fill = Y)
- self._quit = Button (self.button_frame)
- self._quit["text"] = "Quit"
- self._quit["command"] = self.quit
+ self._quit = Button (self.button_frame, text = _ ("Quit"),
+ command = self.quit)
self.balloon.bind_widget (self._quit,
- msg = "Quit the program immediately")
+ msg = _ ("Quit the program immediately"))
self._quit.pack (side = LEFT)
self.update_widgets ()
if gnupg_exists (self.arguments) and openssh_exists (self.arguments):
return False
+ if not valid_name (self.user_var.get ()):
+ return False
+
+ if not valid_host (self.host_var.get ()):
+ return False
+
if not valid_email (self.email_var.get ()):
return False
return True
- def update_widgets (self, *args):
- valid = self.valid_state ()
+ def update_field (self, name):
+ field = self.fields[name]
- self._generate["state"] = NORMAL if valid else DISABLED
+ _fg (_valid (field[1] (field[0].get ())), field[2], field[3])
- name = self.name_var.get ()
-
- valid = valid_name (name)
- self.name["fg"] = "black" if valid else "red"
- self.name_label["fg"] = "black" if valid else "red"
-
- email = self.email_var.get ()
-
- valid = valid_email (email)
- self.email["fg"] = "black" if valid else "red"
- self.email_label["fg"] = "black" if valid else "red"
-
- comment = self.comment_var.get ()
-
- valid = valid_comment (comment)
- self.comment["fg"] = "black" if valid else "red"
- self.comment_label["fg"] = "black" if valid else "red"
+ def update_widgets (self, *args):
+ self._generate["state"] = _state (self.valid_state ())
- exists = gnupg_exists (self.arguments)
- self.gnupg["state"] = NORMAL if not exists else DISABLED
+ for field in ["user", "host", "name", "email", "comment"]:
+ self.update_field (field)
- exists = openssh_exists (self.arguments)
- self.openssh["state"] = NORMAL if not exists else DISABLED
+ self.gnupg["state"] = _state (not gnupg_exists (self.arguments))
+ self.openssh["state"] = _state (not openssh_exists (self.arguments))
- gnupg_key = name
- if comment.strip () != "":
+ gnupg_key = self.name_var.get ().strip ()
+ comment = self.comment_var.get ().strip ()
+ if comment != "":
gnupg_key + " ({}) ".format (comment)
- gnupg_key += "<{}>".format (email)
+ gnupg_key += "<{}>".format (self.email_var.get ().strip ())
- user = self.user_var.get ()
- host = self.host_var.get ()
+ user = self.user_var.get ().strip ()
+ host = self.host_var.get ().strip ()
openssh_key = "{}@{}".format (user, host)
- msg = dedented ("""
+ msg = dedented (_ ("""
Generate a GnuPG key for '{}' and configure a default setup for it
- """).format (gnupg_key)
+ """)).format (gnupg_key)
self.balloon.bind_widget (self.gnupg, msg = msg)
self.balloon.bind_widget (self.gnupg_label, msg = msg)
- msg = dedented ("""
+ msg = dedented (_ ("""
Generate an OpenSSH key for '{}' and configure a default setup for it
- """).format (openssh_key)
+ """)).format (openssh_key)
self.balloon.bind_widget (self.openssh, msg = msg)
self.balloon.bind_widget (self.openssh_label, msg = msg)
sys.stdout = stdout
def _on_idle ():
- while True:
- try:
+ try:
+ while True:
self.progress.redirect.write (self.progress.queue.get (block = False))
- except Empty:
- break
+ except Empty:
+ pass
def generate (self):
self.progress = CryptoInstallProgress (self)
def main ():
+ gettext_install ()
+
arguments = parse_arguments ()
if arguments.gui: