Add a mechanism for converting an identifier to a PTS ID.

Signed-off-by: Evan Broder <broder@mit.edu>

diff --git a/afs/_pts.pyx b/afs/_pts.pyx
index 83fa010..8a8937c 100644 (file)
--- a/afs/_pts.pyx
+++ b/afs/_pts.pyx
@@ -212,6 +212,17 @@ cdef class PTS:
         ubik_ClientDestroy(self.client)
         rx_Finalize()
 
         ubik_ClientDestroy(self.client)
         rx_Finalize()
 

+    def NameOrId(self, ident):
+        """
+        Given an identifier, convert it to a PTS ID by looking up the
+        name if it's a string, or otherwise just converting it to an
+        integer.
+        """
+        if isinstance(ident, (str, unicode)):
+            return self.NameToId(ident)
+        else:
+            return int(ident)
+

     def NameToId(self, name):
         """
         Converts a user or group to an AFS ID.
     def NameToId(self, name):
         """
         Converts a user or group to an AFS ID.