Add badge for admitted proofs

1 files changed, 36 insertions, 0 deletions

diff --git a/scripts/statistics.py b/scripts/statistics.py
new file mode 100644
index 0000000..927baba
--- /dev/null
+++ b/scripts/statistics.py
@@ -0,0 +1,36 @@
+import urllib.request
+import subprocess
+import sys
+
+
+def collect(d):
+    try:
+        n_admitted = len(subprocess.check_output(
+            ["grep", "-r", "-E", "Admitted", d]).splitlines())
+    except: n_admitted = 0
+
+    try: n_theorems = len(subprocess.check_output(
+            ["grep", "-r", "-E", "Theorem|Lemma|Remark", d]).splitlines())
+    except: n_theorems = 0
+
+    try: n_qed = len(subprocess.check_output(
+            ["grep", "-r", "-E", "Qed|Defined", d]).splitlines())
+    except: n_qed = 0
+
+    return (n_admitted, n_theorems, n_qed)
+
+
+def main(d):
+    n_admitted, _, _ = collect(d)
+
+    url = "https://img.shields.io/badge/Admitted%20Proofs-{}-{}?style=flat".format(
+        str(n_admitted), "orange")
+
+    req = urllib.request.Request(url, headers={"User-Agent": "Mozilla/5.0"})
+    with open("admitted.svg", "wb") as f:
+        with urllib.request.urlopen(req) as r:
+            f.write(r.read())
+
+
+if __name__ == "__main__":
+    main(sys.argv[1])