diff --git a/init.py b/init.py index be38092..26df26c 100644 --- a/init.py +++ b/init.py @@ -4,12 +4,12 @@ import os from config import bcolors #Create motd entry -#try: -motdscript = open('/etc/update-motd/stats', 'w') -#except: -# print('[' + bcolors.FULL + 'IMPORTANT' + bcolors.RESET + ']') -# print('Make sure you run the program as root!') -# quit() +try: + motdscript = open('/etc/update-motd.d/stats', 'w') +except: + print('[' + bcolors.FULL + 'IMPORTANT' + bcolors.RESET + ']') + print('Make sure you run the program as root!') + quit() cwd = os.getcwd()