aboutsummaryrefslogtreecommitdiffstats
path: root/minidinstall/SafeWriteFile.py
diff options
context:
space:
mode:
Diffstat (limited to 'minidinstall/SafeWriteFile.py')
-rwxr-xr-xminidinstall/SafeWriteFile.py10
1 files changed, 4 insertions, 6 deletions
diff --git a/minidinstall/SafeWriteFile.py b/minidinstall/SafeWriteFile.py
index 1777d36..591c4f0 100755
--- a/minidinstall/SafeWriteFile.py
+++ b/minidinstall/SafeWriteFile.py
@@ -21,9 +21,7 @@
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
-from types import StringType
from shutil import copy2
-from string import find
from os import rename
class ObjectNotAllowed(Exception):
@@ -37,14 +35,14 @@ class InvalidMode(Exception):
class SafeWriteFile:
def __init__(self, newname, realname, mode="w", bufsize=-1):
- if type(newname)!=StringType:
+ if not isinstance(newname, str):
raise ObjectNotAllowed(newname)
- if type(realname)!=StringType:
+ if not isinstance(realname, str):
raise ObjectNotAllowed(realname)
- if find(mode, "r")>=0:
+ if "r" in mode:
raise InvalidMode(mode)
- if find(mode, "a")>=0 or find(mode, "+") >= 0:
+ if "a" in mode or "+" in mode:
copy2(realname, newname)
self.fobj=open(newname, mode, bufsize)
self.newname=newname