# Author: Stefan Weijers https://github.com/Stxfie

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()

cwd = os.getcwd()

motdscript.write('#!/bin/bash\n{}/start.sh'.format(cwd))

#Create start script
startscript = open('start.sh', 'w')
startscript.write('#!/bin/bash\npython3 {}/time.py\necho\npython3 {}/diskusage.py\necho\npython3 {}/services.py'.format(cwd, cwd, cwd))

print('All done!')