Port to Python 3

1 files changed, 4 insertions, 6 deletions

diff --git a/minidinstall/DpkgDatalist.py b/minidinstall/DpkgDatalist.py
index 0c11612..68f9940 100644
--- a/minidinstall/DpkgDatalist.py
+++ b/minidinstall/DpkgDatalist.py
@@ -21,10 +21,9 @@
 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 
 import os, sys
-from UserDict import UserDict
-from OrderedDict import OrderedDict
+from collections import UserDict
+from collections import OrderedDict
 from minidinstall.SafeWriteFile import SafeWriteFile
-from types import StringType
 
 class DpkgDatalistException(Exception):
     UNKNOWN     = 0
@@ -55,15 +54,14 @@ class _DpkgDatalist:
             self._store(sys.stdout)
             return
 
-        # Write to a temporary file first
-        if type(fn) == StringType:
+        if isinstance(fn, str):
             vf=SafeWriteFile(fn+".new", fn, "w")
         else:
             vf=fn
         try:
             self._store(vf)
         finally:
-            if type(fn) == StringType:
+            if isinstance(fn, str):
                 vf.close()