From 4536589e2c121c4798c86c6e3721fbd1282ee88b Mon Sep 17 00:00:00 2001 From: Pol Henarejos Date: Mon, 10 Oct 2022 11:02:07 +0200 Subject: [PATCH] Added error message if package is missing. Signed-off-by: Pol Henarejos --- tools/pico-hsm-tool.py | 26 ++++++++++++++++++-------- 1 file changed, 18 insertions(+), 8 deletions(-) diff --git a/tools/pico-hsm-tool.py b/tools/pico-hsm-tool.py index 8878ad3..9df8763 100755 --- a/tools/pico-hsm-tool.py +++ b/tools/pico-hsm-tool.py @@ -19,14 +19,24 @@ */ """ -from smartcard.CardType import AnyCardType -from smartcard.CardRequest import CardRequest -from smartcard.Exceptions import CardRequestTimeoutException -from cvc.certificates import CVC -from cvc.asn1 import ASN1 -from cvc.oid import oid2scheme -from cvc.utils import scheme_rsa -from cryptography.hazmat.primitives.asymmetric import ec +import sys +try: + from smartcard.CardType import AnyCardType + from smartcard.CardRequest import CardRequest + from smartcard.Exceptions import CardRequestTimeoutException + from cvc.certificates import CVC +except: + print('ERROR: smarctard module not found! Install pyscard package.\nTry with `pip install pyscard`') + sys.exit(-1) + +try: + from cvc.asn1 import ASN1 + from cvc.oid import oid2scheme + from cvc.utils import scheme_rsa + from cryptography.hazmat.primitives.asymmetric import ec +except: + print('ERROR: cvc module not found! Install pycvc package.\nTry with `pip install pycvc`') + sys.exit(-1) import json import urllib.request import base64