# HG changeset patch
# User Brian Neal <bgneal@gmail.com>
# Date 1372635031 18000
# Node ID 987a29d8bce354612aba869737493c42f680062f
# Parent  4705429d3b21701a04976df6da6ad28e3ce67057
Encrypt command-line command is working.

diff -r 4705429d3b21 -r 987a29d8bce3 m209/keylist/config.py
--- a/m209/keylist/config.py	Sun Jun 30 16:34:24 2013 -0500
+++ b/m209/keylist/config.py	Sun Jun 30 18:30:31 2013 -0500
@@ -59,7 +59,7 @@
 
     if indicator and indicator not in config.sections():
         return None
-    else:
+    elif indicator is None:
         # choose one at random
         indicator = random.choice(config.sections())
 
diff -r 4705429d3b21 -r 987a29d8bce3 m209/main.py
--- a/m209/main.py	Sun Jun 30 16:34:24 2013 -0500
+++ b/m209/main.py	Sun Jun 30 18:30:31 2013 -0500
@@ -26,6 +26,7 @@
 DEFAULT_KEY_LIST = 'm209keys.cfg'
 LOG_CHOICES = ['debug', 'info', 'warning', 'error', 'critical']
 SYS_IND_RE = re.compile(r'^[A-Z]{1}$')
+M209_ALPHABET_LOWER = set(c.lower() for c in M209_ALPHABET_SET)
 
 
 def validate_key_list_indicator(s):
@@ -94,21 +95,42 @@
     raise argparse.ArgumentTypeError('value must be 1 letter')
 
 
+def plaintext_filter(fp):
+    """Generator function to filter input plaintext.
+
+    * ASCII upper case letters are passed as-is.
+    * ASCII lower case letters are converted to upper case.
+    * Whitespace characters are converted to 'Z'.
+    * All other characters are dropped from the input.
+
+    """
+    for line in fp:
+        for c in line:
+            if c in M209_ALPHABET_SET:
+                yield c
+            elif c in M209_ALPHABET_LOWER:
+                yield c.upper()
+            elif c.isspace():
+                yield 'Z'
+
+
 def encrypt(args):
     """Encrypt subcommand processor"""
-    print('Encrypting!', args)
+    logging.info("Encrypting using key file %s", args.file)
 
-    logging.info("Encrypting using key file %s", args.file)
+    # Get a key list from the key list file
     if not os.path.isfile(args.file):
         sys.exit("key list file not found: {}\n".format(args.file))
 
-    # Get a key list from the key list file
     key_list = read_key_list(args.file, args.key_list_ind)
     if not key_list:
         sys.exit("key list not found in file: {}\n".format(args.file))
 
+    # Get the plaintext
+    infile = open(args.plaintext, 'r') if args.plaintext != '-' else sys.stdin
+    plaintext = plaintext_filter(infile)
+
     proc = StdProcedure(key_list=key_list)
-    plaintext = "HELLOZWORLDX"
     ct = proc.encrypt(plaintext, ext_msg_ind=args.ext_ind, sys_ind=args.sys_ind)
     print(ct)