diff options
| -rwxr-xr-x | countdistros.py | 6 | ||||
| -rwxr-xr-x | countdistros_pichart.py | 4 |
2 files changed, 7 insertions, 3 deletions
diff --git a/countdistros.py b/countdistros.py index ccdb19c..08b242e 100755 --- a/countdistros.py +++ b/countdistros.py @@ -5,7 +5,9 @@ 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' ] +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 @@ -30,7 +32,7 @@ def get_data(filename='reports_piiremoved.json'): try: with open(filename) as file: db = json.load(file) - except: + except Exception: err = filename + ' doesn\'t exist or isn\'t valid json!' print(err) sys.exit(1) diff --git a/countdistros_pichart.py b/countdistros_pichart.py index ed66d71..09d19bb 100755 --- a/countdistros_pichart.py +++ b/countdistros_pichart.py @@ -3,9 +3,10 @@ import countdistros as countd import matplotlib.pyplot as plt + def run(): data = countd.get_data() - + labels = [] values = [] for x, y in data.items(): @@ -16,5 +17,6 @@ def run(): plt.axis('equal') plt.show() + if __name__ == '__main__': run() |