initial commit

1 files changed, 59 insertions, 0 deletions

diff --git a/countdistros.py b/countdistros.py
new file mode 100755
index 0000000..ccdb19c
--- /dev/null
+++ b/countdistros.py
@@ -0,0 +1,59 @@
+#!/usr/bin/env python
+import json
+from multiprocessing import Manager, Process
+import re
+import sys
+
+# array of distro names (EDIT THIS AND PROBABLY USE CAPITALIZATION)
+arr = [ 'Ubuntu', 'Linux Mint', 'Pop!_OS', 'Debian', 'Arch Linux', 'Manjaro Linux', 'Fedora' ]
+
+def get_count(distro, db, data):
+    # print how many times distro appears in report
+    # by matching its' name to data using regex
+    rg = re.compile(distro)
+    count = 0
+    for item in db:
+        if re.search(rg, item['systemInfo']['os']):
+            count += 1
+    data[distro] = str(count)
+
+
+def get_data(filename='reports_piiremoved.json'):
+    # use default file name if file is not supplied
+    if len(sys.argv) > 1:
+        filename = sys.argv[1]
+    else:
+        print('defaulting to ' + filename + ' since no argument was supplied')
+
+    # try to load file as json
+    db = None
+    try:
+        with open(filename) as file:
+            db = json.load(file)
+    except:
+        err = filename + ' doesn\'t exist or isn\'t valid json!'
+        print(err)
+        sys.exit(1)
+
+    # get distro count asynchronously :trollfig:
+    data = Manager().dict()
+    procs = []
+    for distro in arr:
+        proc = Process(target=get_count, args=(distro, db, data))
+        procs.append(proc)
+        proc.start()
+
+    for proc in procs:
+        proc.join()
+
+    return data
+
+
+def print_data():
+    data = get_data()
+    for key in data:
+        print(key + ': ' + data[key])
+
+
+if __name__ == "__main__":
+    print_data()