Use os.replace for Windows compat

Solves update script running on Windows, issue #23.
ko-zu · Jan 31, 2023 · 8ae0d58 · 8ae0d58
1 parent 07e8e87
commit 8ae0d58
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/publicsuffixlist/update.py b/publicsuffixlist/update.py
@@ -38,7 +38,12 @@ def updatePSL(psl_file=PSLFILE):
     with open(psl_file + ".swp", "rb") as f:
         psl = PublicSuffixList(f)
 
-    os.rename(psl_file + ".swp", psl_file)
+    try:
+        os.replace(psl_file + ".swp", psl_file)
+    except AttributeError:
+        # will not work on python2 on Win.
+        os.rename(psl_file + ".swp", psl_file)
+
     if lastmod:
         t = calendar.timegm(parsedate(lastmod))
         os.utime(psl_file, (t, t))