def test_infer_return_type_multiples_and_none(self): self.assertIs(str, tempfile._infer_return_type(None, '')) self.assertIs(str, tempfile._infer_return_type('', None)) self.assertIs(str, tempfile._infer_return_type(None, None)) self.assertIs(bytes, tempfile._infer_return_type(b'', None)) self.assertIs(bytes, tempfile._infer_return_type(None, b'')) with self.assertRaises(TypeError): tempfile._infer_return_type('', None, b'') with self.assertRaises(TypeError): tempfile._infer_return_type(b'', None, '')
def __init__(self, dir, pre, suf, bin): if bin: flags = self._bflags else: flags = self._tflags output_type = tempfile._infer_return_type(dir, pre, suf) self.fd, self.name = tempfile._mkstemp_inner( dir, pre, suf, flags, output_type)
def do_create(self, dir=None, pre=None, suf=None, bin=1): output_type = tempfile._infer_return_type(dir, pre, suf) if dir is None: if output_type is str: dir = tempfile.gettempdir() else: dir = tempfile.gettempdirb() if pre is None: pre = output_type() if suf is None: suf = output_type() file = self.mkstemped(dir, pre, suf, bin) self.nameCheck(file.name, dir, pre, suf) return file
def _sanitize_params(prefix, suffix, dir): """Common parameter processing for most APIs in this module.""" output_type = _infer_return_type(prefix, suffix, dir) if suffix is None: suffix = output_type() if prefix is None: if output_type is str: prefix = "tmp" else: prefix = os.fsencode("tmp") if dir is None: if output_type is str: dir = gettempdir() else: dir = os.fsencode(gettempdir()) return prefix, suffix, dir, output_type
def do_create(self, dir=None, pre=None, suf=None): output_type = tempfile._infer_return_type(dir, pre, suf) if dir is None: if output_type is str: dir = tempfile.gettempdir() else: dir = tempfile.gettempdirb() if pre is None: pre = output_type() if suf is None: suf = output_type() name = tempfile.mkdtemp(dir=dir, prefix=pre, suffix=suf) try: self.nameCheck(name, dir, pre, suf) return name except: os.rmdir(name) raise
def do_create(self, dir=None, pre=None, suf=None): output_type = tempfile._infer_return_type(dir, pre, suf) if dir is None: if output_type is str: dir = tempfile.gettempdir() else: dir = tempfile.gettempdirb() if pre is None: pre = output_type() if suf is None: suf = output_type() (fd, name) = tempfile.mkstemp(dir=dir, prefix=pre, suffix=suf) (ndir, nbase) = os.path.split(name) adir = os.path.abspath(dir) self.assertEqual(adir, ndir, "Directory '%s' incorrectly returned as '%s'" % (adir, ndir)) try: self.nameCheck(name, dir, pre, suf) finally: os.close(fd) os.unlink(name)
def test_infer_return_type_singles(self): self.assertIs(str, tempfile._infer_return_type('')) self.assertIs(bytes, tempfile._infer_return_type(b'')) self.assertIs(str, tempfile._infer_return_type(None))
def __init__(self, dir, pre, suf, bin): if bin: flags = self._bflags else: flags = self._tflags output_type = tempfile._infer_return_type(dir, pre, suf) (self.fd, self.name) = tempfile._mkstemp_inner(dir, pre, suf, flags, output_type)
def update_event(self, inp=-1): self.set_output_val(0, tempfile._infer_return_type())