1 files changed, 3 insertions, 2 deletions
diff --git a/libhal.py b/libhal.py
index 3063398..53e5fe8 100644
--- a/libhal.py
+++ b/libhal.py
@@ -391,8 +391,9 @@ class PKey(Handle):
     def verify(self, hash = 0, data = "", signature = None):
         self.hsm.pkey_verify(self, hash = hash, data = data, signature = signature)
 
-    def set_attributes(self, attributes):
-        self.hsm.pkey_set_attributes(self, attributes)
+    def set_attributes(self, attributes = None, **kwargs):
+        assert if attributes is None or not kwargs
+        self.hsm.pkey_set_attributes(self, attributes or kwargs)
 
     def get_attributes(self, attributes):
         attrs = self.hsm.pkey_get_attributes(self, attributes, 0)