bugfix. column name text_chars

[mediawiki_dump_tools.git] / wikiq

diff --git a/wikiq b/wikiq
index bffbbf4cd10aefffcb206908c51a894a7fc70344..fb198f3a72f8cbaa37096d3dfd7e5da7fef2e573 100755 (executable)
--- a/wikiq
+++ b/wikiq
@@ -254,7 +254,7 @@ class RevDataBase():
         pa.field("title",pa.string()),
         pa.field("namespace",pa.int32()),
         pa.field("deleted",pa.bool_()),
-        pa.field("test_chars",pa.int32()),
+        pa.field("text_chars",pa.int32()),
         pa.field("revert",pa.bool_()),
         pa.field("reverteds",pa.list_(pa.int64())),
         pa.field("sha1",pa.string()),
@@ -429,12 +429,12 @@ class WikiqParser():
         else:
             sys.exit('Each regular expression *must* come with a corresponding label and vice versa.')
 
-    def matchmake(self, rev, rev_data):
-        rev_data = self.matchmake_revision(rev.text, rev_data)
+    def matchmake_revision(self, rev, rev_data):
+        rev_data = self.matchmake_text(rev.text, rev_data)
         rev_data = self.matchmake_comment(rev.comment, rev_data)
         return rev_data
 
-    def matchmake_revision(self, text, rev_data):
+    def matchmake_text(self, text, rev_data):
          return self.matchmake_pairs(text, rev_data, self.regex_revision_pairs)
 
     def matchmake_comment(self, comment, rev_data):
@@ -518,7 +518,7 @@ class WikiqParser():
                                              namespace = namespace
                                              )
 
-                rev_data = self.matchmake(rev, rev_data)
+                rev_data = self.matchmake_revision(rev, rev_data)
 
                 if not rev.deleted.text:
                     # rev.text can be None if the page has no text
@@ -661,7 +661,7 @@ class WikiqParser():
 
         outtable = rows_to_table(self.parquet_buffer, self.schema)
         if self.pq_writer is None:
-            self.pq_writer = pq.ParquetWriter(self.output_file, schema, flavor='spark')
+            self.pq_writer = pq.ParquetWriter(self.output_file, self.schema, flavor='spark')
 
         self.pq_writer.write_table(outtable)
         self.parquet_buffer = []