X-Git-Url: https://projects.mako.cc/source/harrypotter-wikipedia-cdsw/blobdiff_plain/fcf662671099db62bdfd5607f9fe51c31747a999..ce5c13c094d659125fe85d59b9bc0e4c2bf40072:/hpwp-minor.py

diff --git a/hpwp-minor.py b/hpwp-minor.py
index 4e173f2..79135a1 100644
--- a/hpwp-minor.py
+++ b/hpwp-minor.py
@@ -1,16 +1,18 @@
+import encoding_fix
+
 from csv import DictReader
 
-input_file = open("hp_wiki.csv", 'r')
+input_file = open("hp_wiki.tsv", 'r', encoding="utf-8")
 
 num_edits = 0
 num_anon = 0
-for row in DictReader(input_file):
+for row in DictReader(input_file, delimiter="\t"):
     num_edits = num_edits + 1
-    if row["anon"] == "False":
+    if row["anon"] == "True":
         num_anon = num_anon + 1
 
 prop_anon = num_anon / num_edits
 
-print("total edits: %s" % num_edits)
-print("anon edits: %s" % num_anon)
-print("proportion anon: %s" % prop_anon)
+print("total edits: {0}".format(num_edits))
+print("anon edits: {0}".format(num_anon))
+print("proportion anon: {0}".format(prop_anon))