Merge pull request #26 from ksauzz/feature/get-or-init

add KrbTicket.get_or_init
ksauzz · Dec 10, 2019 · 3f7e1b5 · 3f7e1b5
2 parents ae4248e + 447c7c8
commit 3f7e1b5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/krbticket/ticket.py b/krbticket/ticket.py
@@ -122,6 +122,14 @@ def init_by_config(config):
         KrbCommand.kinit(config)
         return KrbTicket.get_by_config(config)
 
+    @staticmethod
+    def get_or_init(principal, keytab=None, **kwargs):
+        config = KrbConfig(principal=principal, keytab=keytab, **kwargs)
+        try:
+            return KrbTicket.get_by_config(config)
+        except NoCredentialFound:
+            return KrbTicket.init_by_config(config)
+
     @staticmethod
     def get(principal, keytab=None, **kwargs):
         config = KrbConfig(principal=principal, keytab=keytab, **kwargs)

diff --git a/tests/test_ticket.py b/tests/test_ticket.py
@@ -12,6 +12,13 @@ def test_init(config):
     KrbTicket.init(DEFAULT_PRINCIPAL, DEFAULT_KEYTAB)
 
 
+def test_get_or_init(config):
+    KrbCommand.kdestroy(config)
+    ticket0 = KrbTicket.get_or_init(DEFAULT_PRINCIPAL, DEFAULT_KEYTAB)
+    ticket1 = KrbTicket.get_or_init(DEFAULT_PRINCIPAL, DEFAULT_KEYTAB)
+    assert_ticket(ticket0, ticket1)
+
+
 def test_init_with_keytab_env(config):
     KrbCommand.kdestroy(config)
     os.environ['KRB5_KTNAME'] = config.keytab