Port to Python 3

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