a minor modification and a small bugfix to the SE import

1 files changed, 11 insertions, 5 deletions

diff --git a/askbot/importers/stackexchange/management/commands/load_stackexchange.py b/askbot/importers/stackexchange/management/commands/load_stackexchange.py
index 3d6bee05..f4c71f9a 100644
--- a/askbot/importers/stackexchange/management/commands/load_stackexchange.py
+++ b/askbot/importers/stackexchange/management/commands/load_stackexchange.py
@@ -1,4 +1,5 @@
 #todo: http://stackoverflow.com/questions/837828/how-to-use-a-slug-in-django 
+DEBUGME = False
 import os
 import re
 import sys
@@ -10,8 +11,7 @@ import askbot.importers.stackexchange.parse_models as se_parser
 from xml.etree import ElementTree as et
 from django.db.models import fields
 from django.db.utils import IntegrityError
-from django.db import models, transaction
-#from askbot.utils import dummy_transaction as transaction
+from django.db import models
 import askbot.models as askbot
 import askbot.deps.django_authopenid.models as askbot_openid
 import askbot.importers.stackexchange.models as se
@@ -25,6 +25,11 @@ from askbot.importers.stackexchange.management import is_ready as importer_is_re
 #from markdown2 import Markdown
 #markdowner = Markdown(html4tags=True)
 
+if DEBUGME == True:
+    from askbot.utils import dummy_transaction as transaction
+else:
+    from django.db import transaction
+
 xml_read_order = (
         'VoteTypes','UserTypes','Users','Users2Votes',
         'Badges','Users2Badges','CloseReasons',#'FlatPages',
@@ -301,8 +306,9 @@ class Command(BaseCommand):
             self.load_xml_file(item)
             transaction.commit()
             time_after = datetime.now()
-            print time_after - time_before
-            print HEAP.heap()
+            if DEBUGME == True:
+                print time_after - time_before
+                print HEAP.heap()
 
         #this is important so that when we clean up messages
         #automatically generated by the procedures below
@@ -392,7 +398,7 @@ class Command(BaseCommand):
         for m in se.Message.objects.all().iterator():
             if m.is_read:
                 continue
-            if m.user in None:
+            if m.user is None:
                 continue
             if m.user.id == -1:
                 continue