Clean up crypto iv generation

jasonacox · Nov 15, 2023 · 159ac7a · 159ac7a
1 parent 286f080
commit 159ac7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tinytuya/core.py b/tinytuya/core.py
@@ -251,7 +251,7 @@ def __init__(self, key):
 
     @classmethod
     def get_encryption_iv( cls, iv ):
-        if not self.CRYPTOLIB_HAS_GCM:
+        if not cls.CRYPTOLIB_HAS_GCM:
             raise NotImplementedError( 'Crypto library does not support GCM' )
         if iv is True:
             if log.isEnabledFor( logging.DEBUG ):
@@ -262,7 +262,7 @@ def get_encryption_iv( cls, iv ):
 
     @classmethod
     def get_decryption_iv( cls, iv, data ):
-        if not self.CRYPTOLIB_HAS_GCM:
+        if not cls.CRYPTOLIB_HAS_GCM:
             raise NotImplementedError( 'Crypto library does not support GCM' )
         if iv is True:
             iv = data[:12]