From: Nicolas Marie-Magdelaine <redacted>
Date: Thu, 21 Jun 2018 14:27:08 +0000 (+0200)
Subject: Add capitalization as a case choice
X-Git-Url: http://git.99rst.org/?a=commitdiff_plain;h=a847d66693e37231279fda802e2436d70e5cfa2b;p=redacted-XKCD-password-generator.git

Add capitalization as a case choice
---

diff --git a/xkcdpass/xkcd_password.py b/xkcdpass/xkcd_password.py
index 823b514..894e586 100755
--- a/xkcdpass/xkcd_password.py
+++ b/xkcdpass/xkcd_password.py
@@ -245,6 +245,12 @@ def lower_case(words):
     """
     return [w.lower() for w in words]
 
+def capitalize_case(words):
+    """
+    Set first letter of each words to UPPER case aka Capitalize.
+    """
+    return [w.capitalize() for w in words]
+
 
 def random_case(words, testing=False):
     """
@@ -266,7 +272,8 @@ def random_case(words, testing=False):
 CASE_METHODS = {"alternating": alternating_case,
                 "upper": upper_case,
                 "lower": lower_case,
-                "random": random_case}
+                "random": random_case,
+                "capitalize":capitalize_case}
 
 
 def set_case(words, method="lower", testing=False):