Run black(1) on the UnitaryTests/*.py files

Author: LudovicRousseau

UnitaryTests/CheckAutoPowerOff.py

@@ -19,10 +19,10 @@
 
 # check the card is not auto powered off after a SCardReconnect()
 
-from smartcard.scard import *
-from smartcard.pcsc.PCSCExceptions import *
 import sys
 import time
+from smartcard.scard import *
+from smartcard.pcsc.PCSCExceptions import *
 
 
 RED = "\033[0;31m"
@@ -36,14 +36,15 @@
 hresult, readers = SCardListReaders(hcontext, [])
 if hresult != SCARD_S_SUCCESS:
     raise ListReadersException(hresult)
-print('PC/SC Readers:', readers)
+print("PC/SC Readers:", readers)
 
 reader = readers[0]
 print("Using reader:", reader)
 
 # Connect in SCARD_SHARE_SHARED mode
-hresult, hcard, dwActiveProtocol = SCardConnect(hcontext, reader,
-    SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY)
+hresult, hcard, dwActiveProtocol = SCardConnect(
+    hcontext, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)
 
@@ -54,8 +55,9 @@
     if hresult == SCARD_W_REMOVED_CARD:
         print("\nInsert the card")
         while hresult != SCARD_S_SUCCESS:
-            hresult, dwActiveProtocol = SCardReconnect(hcard,
-                SCARD_SHARE_EXCLUSIVE, SCARD_PROTOCOL_ANY, SCARD_LEAVE_CARD)
+            hresult, dwActiveProtocol = SCardReconnect(
+                hcard, SCARD_SHARE_EXCLUSIVE, SCARD_PROTOCOL_ANY, SCARD_LEAVE_CARD
+            )
             print(".", end="")
             sys.stdout.flush()
             time.sleep(1)
@@ -65,7 +67,7 @@
 
     if hresult != SCARD_S_SUCCESS:
         raise BaseSCardException(hresult)
-    #print(SCardGetErrorMessage(hresult))
+    # print(SCardGetErrorMessage(hresult))
 
     print(".", end="")
     sys.stdout.flush()
@@ -76,7 +78,7 @@
 
     if count >= 0:
         count += 1
-        if (count > 10):
+        if count > 10:
             print(BLUE + "\nTest passed" + NORMAL)
             break
     time.sleep(1)

UnitaryTests/FEATURE_CCID_ESC_COMMAND_Gemalto_features.py

@@ -19,28 +19,35 @@
 #   with this program; if not, see <http://www.gnu.org/licenses/>.
 
 from smartcard.System import readers
-from smartcard.pcsc.PCSCPart10 import (SCARD_SHARE_DIRECT,
-    SCARD_LEAVE_CARD, FEATURE_CCID_ESC_COMMAND,
-    FEATURE_GET_TLV_PROPERTIES, getTlvProperties,
-    getFeatureRequest, hasFeature)
+from smartcard.pcsc.PCSCPart10 import (
+    SCARD_SHARE_DIRECT,
+    SCARD_LEAVE_CARD,
+    FEATURE_CCID_ESC_COMMAND,
+    FEATURE_GET_TLV_PROPERTIES,
+    getTlvProperties,
+    getFeatureRequest,
+    hasFeature,
+)
 from smartcard.Exceptions import SmartcardException
+
 try:
     from itertools import izip
 except ImportError:
     izip = zip
 
 
 # http://www.usb.org/developers/docs/USB_LANGIDs.pdf
-USBLangID = {0x0409: "English (United States)",
-        0x040C: "French (Standard)",
-        0x0425: "Estonian",
-        0x0419: "Russian",
-        0x0415: "Polish",
-        0x0416: "Portuguese (Brazil)",
-        0x0405: "Czech",
-        0x040A: "Spanish (Traditional Sort)",
-        0x041B: "Slovak"
-        }
+USBLangID = {
+    0x0409: "English (United States)",
+    0x040C: "French (Standard)",
+    0x0425: "Estonian",
+    0x0419: "Russian",
+    0x0415: "Polish",
+    0x0416: "Portuguese (Brazil)",
+    0x0405: "Czech",
+    0x040A: "Spanish (Traditional Sort)",
+    0x041B: "Slovak",
+}
 
 
 def test_bit(value, bit):
@@ -49,19 +56,18 @@ def test_bit(value, bit):
 
 
 def main():
-    """ main """
+    """main"""
     card_connection = readers()[0].createConnection()
-    card_connection.connect(mode=SCARD_SHARE_DIRECT,
-        disposition=SCARD_LEAVE_CARD)
+    card_connection.connect(mode=SCARD_SHARE_DIRECT, disposition=SCARD_LEAVE_CARD)
 
     feature_list = getFeatureRequest(card_connection)
 
     get_tlv_properties = hasFeature(feature_list, FEATURE_GET_TLV_PROPERTIES)
     if get_tlv_properties:
         tlv = getTlvProperties(card_connection)
         print("Reader:   ", readers()[0])
-        print("IdVendor:  0x%04X" % tlv['PCSCv2_PART10_PROPERTY_wIdVendor'])
-        print("IdProduct: 0x%04X" % tlv['PCSCv2_PART10_PROPERTY_wIdProduct'])
+        print("IdVendor:  0x%04X" % tlv["PCSCv2_PART10_PROPERTY_wIdVendor"])
+        print("IdProduct: 0x%04X" % tlv["PCSCv2_PART10_PROPERTY_wIdProduct"])
 
     ccid_esc_command = hasFeature(feature_list, FEATURE_CCID_ESC_COMMAND)
     if ccid_esc_command is None:
@@ -137,5 +143,6 @@ def main():
         print("FirewalledCommand_SW2: 0x%02X" % res[17])
     # bytes 18-20 are RFU
 
+
 if __name__ == "__main__":
     main()

UnitaryTests/FEATURE_CCID_ESC_COMMAND_Xiring.py

@@ -25,8 +25,13 @@
 # option of the CCID driver Info.plist file
 
 from smartcard.System import readers
-from smartcard.pcsc.PCSCPart10 import (getFeatureRequest, hasFeature,
-    getTlvProperties, FEATURE_CCID_ESC_COMMAND, SCARD_SHARE_DIRECT)
+from smartcard.pcsc.PCSCPart10 import (
+    getFeatureRequest,
+    hasFeature,
+    getTlvProperties,
+    FEATURE_CCID_ESC_COMMAND,
+    SCARD_SHARE_DIRECT,
+)
 
 # use the first reader
 card_connection = readers()[0].createConnection()
@@ -43,18 +48,19 @@
 tlv = getTlvProperties(card_connection)
 
 # check we are using a Xiring Leo v1 or v2 reader
-if tlv['PCSCv2_PART10_PROPERTY_wIdVendor'] == 0x0F14 \
-    and (tlv['PCSCv2_PART10_PROPERTY_wIdProduct'] in [0x0037, 0x0038]):
+if tlv["PCSCv2_PART10_PROPERTY_wIdVendor"] == 0x0F14 and (
+    tlv["PCSCv2_PART10_PROPERTY_wIdProduct"] in [0x0037, 0x0038]
+):
 
     # proprietary escape command for Xiring Leo readers
     version = [ord(c) for c in "VERSION"]
     res = card_connection.control(ccid_esc_command, version)
     print(res)
-    print("VERSION:", ''.join([chr(x) for x in res]))
+    print("VERSION:", "".join([chr(x) for x in res]))
 
     serial = [ord(c) for c in "GET_SN"]
     res = card_connection.control(ccid_esc_command, serial)
     print(res)
-    print("GET_SN:", ''.join([chr(x) for x in res]))
+    print("GET_SN:", "".join([chr(x) for x in res]))
 else:
     print("Xiring Leo reader not found")

UnitaryTests/FEATURE_GET_TLV_PROPERTIES.py

@@ -39,8 +39,10 @@
     tlv = getTlvProperties(card_connection)
 
     for key in sorted(tlv):
-        if key in ["PCSCv2_PART10_PROPERTY_wIdProduct",
-                "PCSCv2_PART10_PROPERTY_wIdVendor"]:
+        if key in [
+            "PCSCv2_PART10_PROPERTY_wIdProduct",
+            "PCSCv2_PART10_PROPERTY_wIdVendor",
+        ]:
             print("%s: 0x%04X" % (key, tlv[key]))
         else:
             print("%s: %s" % (key, tlv[key]))

UnitaryTests/MCT_ReaderDirect.py

@@ -19,13 +19,18 @@
 #   with this program; if not, see <http://www.gnu.org/licenses/>.
 
 from smartcard.System import readers
-from smartcard.pcsc.PCSCPart10 import (SCARD_SHARE_DIRECT,
-    SCARD_LEAVE_CARD, FEATURE_MCT_READER_DIRECT, getFeatureRequest, hasFeature)
+from smartcard.pcsc.PCSCPart10 import (
+    SCARD_SHARE_DIRECT,
+    SCARD_LEAVE_CARD,
+    FEATURE_MCT_READER_DIRECT,
+    getFeatureRequest,
+    hasFeature,
+)
 from smartcard.util import toHexString
 
 
 def parse_info(bytes):
-    """ parse the SECODER INFO answer """
+    """parse the SECODER INFO answer"""
     print("parse the SECODER INFO answer:", toHexString(bytes))
 
     sw = bytes[-2:]
@@ -34,36 +39,34 @@ def parse_info(bytes):
     while len(bytes):
         tag = bytes[0]
         length = bytes[1]
-        data = bytes[2:2 + length]
+        data = bytes[2 : 2 + length]
 
-        print("tag: %02X, length: %2d:" % (tag, length), end=' ')
+        print("tag: %02X, length: %2d:" % (tag, length), end=" ")
         if tag in [0x40, 0x80, 0x81, 0x83, 0x84]:
-            print("'%s'" % ''.join([chr(x) for x in data]))
+            print("'%s'" % "".join([chr(x) for x in data]))
         else:
             print(toHexString(data))
 
-        del bytes[:2 + length]
+        del bytes[: 2 + length]
     print("SW:", toHexString(sw))
 
 
 def parse_select(bytes):
-    """ parse the SECODER SELECT APPLICATION answer """
-    print("parse the SECODER SELECT APPLICATION answer:",
-            toHexString(bytes))
+    """parse the SECODER SELECT APPLICATION answer"""
+    print("parse the SECODER SELECT APPLICATION answer:", toHexString(bytes))
 
     print("Activation ID:", toHexString(bytes[0:4]))
-    print("Interface Version: '%s'" % ''.join([chr(x) for x in bytes[5:11]]))
+    print("Interface Version: '%s'" % "".join([chr(x) for x in bytes[5:11]]))
     print("Language Code:", toHexString(bytes[11:15]))
     print("CSI:", toHexString(bytes[15:18]))
     print("Application Identifier:", toHexString(bytes[18:23]))
     print("SW:", toHexString(bytes[23:25]))
 
 
 def main():
-    """ main """
+    """main"""
     card_connection = readers()[0].createConnection()
-    card_connection.connect(mode=SCARD_SHARE_DIRECT,
-        disposition=SCARD_LEAVE_CARD)
+    card_connection.connect(mode=SCARD_SHARE_DIRECT, disposition=SCARD_LEAVE_CARD)
 
     feature_list = getFeatureRequest(card_connection)
     # print(getPinProperties(card_connection))
@@ -76,11 +79,40 @@ def main():
     res = card_connection.control(mct_reader_direct, secoder_info)
     parse_info(res)
 
-    secoder_select = [0x20, 0x71, 0x00, 0x00, 0x00, 0x00, 0x14, 0x00, 0x80,
-        0x05, 0x31, 0x2E, 0x31, 0x2E, 0x30, 0x84, 0x02, 0x64, 0x65,
-        0x90, 0x01, 0x01, 0x85, 0x03, ord('g'), ord('k'), ord('p'), 0x00, 0x00]
+    secoder_select = [
+        0x20,
+        0x71,
+        0x00,
+        0x00,
+        0x00,
+        0x00,
+        0x14,
+        0x00,
+        0x80,
+        0x05,
+        0x31,
+        0x2E,
+        0x31,
+        0x2E,
+        0x30,
+        0x84,
+        0x02,
+        0x64,
+        0x65,
+        0x90,
+        0x01,
+        0x01,
+        0x85,
+        0x03,
+        ord("g"),
+        ord("k"),
+        ord("p"),
+        0x00,
+        0x00,
+    ]
     res = card_connection.control(mct_reader_direct, secoder_select)
     parse_select(res)
 
+
 if __name__ == "__main__":
     main()

UnitaryTests/SCardBeginTransaction_Disconnect.py

@@ -31,11 +31,13 @@
 hresult, readers = SCardListReaders(hcontext, [])
 if hresult != SCARD_S_SUCCESS:
     raise ListReadersException(hresult)
-print('PC/SC Readers:', readers)
+print("PC/SC Readers:", readers)
 reader = readers[0]
 print("Using reader:", reader)
 
-hresult, hcard, dwActiveProtocol = SCardConnect(hcontext, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY)
+hresult, hcard, dwActiveProtocol = SCardConnect(
+    hcontext, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)
 

UnitaryTests/SCardBeginTransaction_ExclusiceMode.py

@@ -29,19 +29,23 @@
 hresult, readers = SCardListReaders(hcontext1, [])
 if hresult != SCARD_S_SUCCESS:
     raise ListReadersException(hresult)
-print('PC/SC Readers:', readers)
+print("PC/SC Readers:", readers)
 reader = readers[0]
 print("Using reader:", reader)
 
 hresult, hcontext2 = SCardEstablishContext(SCARD_SCOPE_USER)
 if hresult != SCARD_S_SUCCESS:
     raise EstablishContextException(hresult)
 
-hresult, hcard1, dwActiveProtocol = SCardConnect(hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY)
+hresult, hcard1, dwActiveProtocol = SCardConnect(
+    hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)
 
-hresult, hcard2, dwActiveProtocol = SCardConnect(hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY)
+hresult, hcard2, dwActiveProtocol = SCardConnect(
+    hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)
 
@@ -64,7 +68,9 @@
 
 # Connect should not return an error since hcard1 and hcard2 are now
 # disconnected
-hresult, hcard1, dwActiveProtocol = SCardConnect(hcontext1, reader, SCARD_SHARE_EXCLUSIVE, SCARD_PROTOCOL_ANY)
+hresult, hcard1, dwActiveProtocol = SCardConnect(
+    hcontext1, reader, SCARD_SHARE_EXCLUSIVE, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)
 

UnitaryTests/SCardBeginTransaction_Reset.py

@@ -30,19 +30,23 @@
 hresult, readers = SCardListReaders(hcontext1, [])
 if hresult != SCARD_S_SUCCESS:
     raise ListReadersException(hresult)
-print('PC/SC Readers:', readers)
+print("PC/SC Readers:", readers)
 reader = readers[0]
 print("Using reader:", reader)
 
 hresult, hcontext2 = SCardEstablishContext(SCARD_SCOPE_USER)
 if hresult != SCARD_S_SUCCESS:
     raise EstablishContextException(hresult)
 
-hresult, hcard1, dwActiveProtocol = SCardConnect(hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY)
+hresult, hcard1, dwActiveProtocol = SCardConnect(
+    hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)
 
-hresult, hcard2, dwActiveProtocol = SCardConnect(hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY)
+hresult, hcard2, dwActiveProtocol = SCardConnect(
+    hcontext1, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
+)
 if hresult != SCARD_S_SUCCESS:
     raise BaseSCardException(hresult)