py:key_credential_links: allow encoding=='auto'

'auto' is the same as None. This is helpful to samba-tool.

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Gary Lockyer <gary@catalyst.net.nz>

diff --git a/python/samba/key_credential_link.py b/python/samba/key_credential_link.py
index b17581bf46df3fd50a0506a6f5ce2aec5a803861..64bca98b946951a6f5c5a9054b4c1ea34f617c62 100644 (file)
--- a/python/samba/key_credential_link.py
+++ b/python/samba/key_credential_link.py
@@ -195,7 +195,7 @@ class KeyCredentialLinkDn(BinaryDn):
         return pem.decode()
 
 
-def get_public_key(data:bytes, encoding:str):
+def get_public_key(data:bytes, encoding:Optional[str] = None) -> RSAPublicKey:
     """decode a key in PEM or DER format.
 
     If it turns out to be a certificate or something, we try to get
@@ -268,6 +268,9 @@ def create_key_credential_link(samdb: SamDB,
         if len(res) == 0:
             raise ValueError(f"link target {target} does not exist")
 
+    if encoding == 'auto':
+        encoding = None
+
     key = get_public_key(data, encoding)
 
     if key.key_size != 2048: