From 71d6d0fe31db13f4ddf5c75209b9bba88a1e0a32 Mon Sep 17 00:00:00 2001 From: Nataliia Solomko Date: Sat, 17 Aug 2024 13:26:51 +0300 Subject: op_mode: T3961: Generate PKI expect 2 character country code --- src/op_mode/pki.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/op_mode/pki.py b/src/op_mode/pki.py index 84b080023..ea7e93931 100755 --- a/src/op_mode/pki.py +++ b/src/op_mode/pki.py @@ -316,7 +316,13 @@ def generate_certificate_request(private_key=None, key_type=None, return_request default_values = get_default_values() subject = {} - subject['country'] = ask_input('Enter country code:', default=default_values['country']) + while True: + country = ask_input('Enter country code:', default=default_values['country']) + if len(country) != 2: + print("Country name must be a 2 character country code") + continue + subject['country'] = country + break subject['state'] = ask_input('Enter state:', default=default_values['state']) subject['locality'] = ask_input('Enter locality:', default=default_values['locality']) subject['organization'] = ask_input('Enter organization name:', default=default_values['organization']) -- cgit v1.2.3