skip fname duplicates

2014-06-21 20:09:15 +04:00 · 2014-06-21 20:09:15 +04:00 · e2009e7f08
commit e2009e7f08
parent ab9a7e34c1
1 changed files with 11 additions and 3 deletions
--- a/nevernote.py
+++ b/nevernote.py
@ -78,9 +78,17 @@ def write_file(page):
    parser = TitleParser(strict=False)
    parser.feed(page)

-    fname = parser.title + '.html'
-    with open(fname, 'w') as a_file:
+    fname = parser.title.replace('/', '_') + '.html'
+    inc = 1
+    while True:
+        try:
+            with open(fname, 'x') as a_file:
+                print('Saving in file "%s"' % fname)
                a_file.write(page)
+                break
+        except FileExistsError:
+            inc += 1
+            fname = parser.title.replace('/', '_') + '_%d.html' % inc


 def main():