Add ability to disable revert detection

Also add test to ensure functionality works.

Signed-off-by: Will Beason <willbeason@gmail.com>

test/Wikiq_Unit_Test.py

@@ -160,6 +160,23 @@ class WikiqTestCase(unittest.TestCase):
         baseline = pd.read_table(tester.baseline_file)
         assert_frame_equal(test, baseline, check_like=True)
 
+    def test_WP_no_revert_radius(self):
+        tester = WikiqTester(IKWIKI, "no_revert_radius")
+
+        try:
+            tester.call_wikiq("-rr 0")
+        except subprocess.CalledProcessError as exc:
+            self.fail(exc.stderr.decode("utf8"))
+
+        copyfile(tester.call_output, tester.test_file)
+
+        # as a test let's make sure that we get equal data frames
+        test = pd.read_table(tester.test_file)
+        num_reverted = sum(i is None for i in test.revert)
+        self.assertEqual(num_reverted, 0)
+        baseline = pd.read_table(tester.baseline_file)
+        assert_frame_equal(test, baseline, check_like=True)
+
     def test_noargs(self):
         tester = WikiqTester(SAILORMOON, "noargs", in_compression="7z")
 

wikiq

@@ -523,7 +523,11 @@ class WikiqParser:
                 if namespace not in self.namespace_filter:
                     continue
 
+            # Disable detecting reverts if radius is 0.
+            if self.revert_radius > 0:
                 rev_detector = mwreverts.Detector(radius=self.revert_radius)
+            else:
+                rev_detector = None
 
             if self.persist != PersistMethod.none:
                 window = deque(maxlen=PERSISTENCE_RADIUS)
@@ -574,6 +578,7 @@ class WikiqParser:
                     rev_data.text_chars = len(rev.text)
 
                     # generate revert data
+                    if rev_detector is not None:
                         revert = rev_detector.process(text_sha1, rev.id)
 
                         if revert: