+def UrlOrAfsValidator(value):
+ if value.startswith('/mit/') or value.startswith('/afs/'):
+ return
+ else:
+ try:
+ URLValidator()(value)
+ except ValidationError:
+ raise ValidationError('Provide a valid URL or AFS path')
+
+def pag_check_helper(fn, args, aklog=False, ccname=None, **kwargs):
+ if 'executable' in kwargs:
+ raise ValueError('"executable" not supported with pag_check_*')
+
+ env = None
+ if 'env' in kwargs:
+ env = kwargs['env']
+ del kwargs['env']
+ if ccname:
+ if env is not None:
+ env = dict(env)
+ else:
+ env = dict(os.environ)
+ env['KRB5CCNAME'] = ccname
+
+ pagsh_cmd = 'exec "$@"'
+ if aklog: pagsh_cmd = "aklog && " + pagsh_cmd
+ args = ['pagsh', '-c', pagsh_cmd, 'exec', ] + args
+
+ return fn(args, env=env, **kwargs)
+
+def pag_check_call(args, **kwargs):
+ return pag_check_helper(subprocess.check_call, args, **kwargs)
+def pag_check_output(args, **kwargs):
+ return pag_check_helper(subprocess.check_output, args, **kwargs)
+
+def kinit(keytab=None, principal=None, autodelete=True, ):
+ if not keytab:
+ keytab = settings.KRB_KEYTAB
+ if not principal:
+ principal = settings.KRB_PRINCIPAL
+ assert keytab and principal
+ fd = tempfile.NamedTemporaryFile(mode='rb', prefix="krb5cc_djmit_", delete=autodelete, )
+ env = dict(KRB5CCNAME=fd.name)
+ kinit_cmd = ['kinit', '-k', '-t', keytab, principal, ]
+ subprocess.check_call(kinit_cmd, env=env)
+ return fd
+