validate function

h4ck3rm1k3 · Aug 14, 2012 · 2351fe8 · 2351fe8
1 parent 0ec2c9c
commit 2351fe8
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/speedydeletion.py b/speedydeletion.py
@@ -8,6 +8,7 @@
 import replace
 import xmlreader
 
+
 from shove import Shove
 file_store = Shove('file://wikiaupload')
 
@@ -22,6 +23,9 @@ def signal_handler(signal, frame):
 import subprocess    
 
 def main(*args):
+
+    print "ARGS:%s\n" % sys.argv
+
     genFactory = pagegenerators.GeneratorFactory()
     # If xmlfilename is None, references will be loaded from the live wiki.
     xmlfilename = None
@@ -44,10 +48,13 @@ def main(*args):
         print "cannot open %s\n" % xmlfilename
         exit (0)
 
-    tempfile = "%s.tmp" % xmlfilename
-
-    status = subprocess.call("xmllint --recover  %s -o %s" % (xmlfilename,tempfile) , shell=True)
-    print "status %d\n" % status
+    if sys.argv[1] == "--validate" :
+        tempfile = "%s.tmp" % xmlfilename
+        status = subprocess.call("xmllint --recover  %s -o %s" % (xmlfilename,tempfile) , shell=True)
+        print "status %d\n" % status
+    else:
+        tempfile = xmlfilename
+
 
     dump = xmlreader.XmlDump(tempfile)
     count = 0