From 162fe78aa6f22123d92590fb5204d05ad9951f16 Mon Sep 17 00:00:00 2001 From: Martin Zimmermann Date: Mon, 11 Nov 2013 12:02:49 +0100 Subject: [PATCH] ask to continue import if DB is not empty --- isso/migrate.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/isso/migrate.py b/isso/migrate.py index 57bfd01..47dbecb 100644 --- a/isso/migrate.py +++ b/isso/migrate.py @@ -9,6 +9,11 @@ import textwrap from time import mktime, strptime from collections import defaultdict +try: + input = raw_input +except NameError: + pass + try: from urlparse import urlparse except ImportError: @@ -43,6 +48,10 @@ def insert(db, thread, posts): def disqus(db, xmlfile): + if db.execute("SELECT * FROM comments").fetchone(): + if input("Isso DB is not empty! Continue? [y/N]: ") not in ("y", "Y"): + raise SystemExit("Abort.") + tree = ElementTree.parse(xmlfile) res = defaultdict(list)