From 1a57f53a3971e8f98b81c36b76276497f6894b28 Mon Sep 17 00:00:00 2001 From: cbarber Date: Sat, 26 Feb 2011 21:41:59 -0500 Subject: [PATCH] Check if file exists before opening it --- nbt.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/nbt.py b/nbt.py index cb9ca41..b1bdf4e 100644 --- a/nbt.py +++ b/nbt.py @@ -16,11 +16,15 @@ import gzip, zlib import struct import StringIO +import os # decorator to handle filename or object as first parameter def _file_loader(func): def wrapper(fileobj, *args): if isinstance(fileobj, basestring): + if not os.path.isfile(fileobj): + return None + # Is actually a filename fileobj = open(fileobj, 'rb') return func(fileobj, *args)