Script_Runner.py

from tkinter import *
from tkinter.ttk import *
import tkinter as tk #used to force certain widget type
import tkinter.font as tkFont
import subprocess as sub
from sys import platform #for knowing if it is windows
from mmfunctions import *
import time
from threading import *
from pathlib import Path
import ugmfunctions
import distro #for figuring out what linux distro

#global variables
OS = distro.linux_distribution()
ops = OS[0]


def resource_path(relative_path):
    """ Get absolute path to resource, works for dev and for PyInstaller """
    try:
        # PyInstaller creates a temp folder and stores path in _MEIPASS
        base_path = sys._MEIPASS
    except Exception:
        base_path = os.path.abspath(".")

    return os.path.join(base_path, relative_path)


#THIS_FOLDER = os.path.dirname(os.path.abspath(__file__))
#print(THIS_FOLDER + '     HELLO')

#print(os.getcwd())

#def findCWD():
#    print(os.getcwd())
#    for root, dirs, files, in os.walk('Q:\\'):
#        if 'Script_Runner.py' in files:
#            newPath = os.chdir(os.path.join(root, 'Script_Runner.py'))
#            print(newPath)
#            print(os.getcwd())

#findCWD()

def rmvusrgrubutton(widget):
    #scrip = scriptrunnerGUI()
    #print(scrip.usrd)
    #scrip.usrd = 1
    #print(scrip.usrd)
    print(widget)
    print('Removing User / Group Settings buttons')
    #scrip.usrgru()
    widget.grid_remove()

def chngTOusr():
    scrip = scriptrunnerGUI()
    print(scrip.usrd)
    print(scrip.mmd)
    scrip.usrd = 0
    scrip.mmd = 1
    print(scrip.usrd)
    print(scrip.mmd)
    scrip.usrgru()

def rmvmmbuttons():
    scrip = scriptrunnerGUI()
    print(scrip.mmd)
    scrip.mmd = 1
    print(scrip.mmd)
    print('Removing main menu buttons...')
    scrip.mmenu()

def chngTOmm():
    scrip = scriptrunnerGUI()
    print(scrip.mmd)
    print(scrip.usrd)
    scrip.mmd = 0
    scrip.usrd = 1
    print(scrip.mmd)
    print(scrip.usrd)
    scrip.usrgru()


def aboutHowtoUse():
    window = Toplevel(root)
    window.iconphoto(False, tk.PhotoImage(file="cup2.png"))
    aboutContent = Label(window, text="""
+----------------------------------------+
|    H o w  T o  U s e  T h e  P r o g r a m    |
+----------------------------------------+

1.) Run as root
2.) Choose command that you would like to use
3.) Click it
4.) Sit back and relax while the command runs   :)

""", background='lightgreen')
    aboutContent.grid()
    window.configure(bg='lightgreen')


def aboutCreators():
    window = Toplevel(root)
    window.iconphoto(False, tk.PhotoImage(file="cup2.png"))
    aboutContent = Label(window, text="""
+----------------------+
|   A P P L E    C I D R     |
+----------------------+

This program/Application/Script was made by and for the Apple Cidr Cyber Patriot team

Creator: Michael Brenner
""", background='lightgreen')
    aboutContent.grid()
    window.configure(bg='lightgreen')


class scriptrunnerGUI():
    ssh = ''
    ftp = ''
    proftpd = ''
    vsftpd = ''
    web = ''
    apaweb = ''
    nginweb = ''
    smb = ''
    sql = ''
    rsnc = ''
    buttons = []
    usrgrubuttons = []
    usrd = 0
    mmd = 0
    fconf = 0
    #commands = [mmfunc.search_media, mmfunc.updates, self.usrgru]

    #def __init__(self):
    #    print('')
    #    self.firstConf()


    def __init__(self):
        # buttonNames=[]
        cwd = os.getcwd()
        #if platform == 'win32':
        variableCheck = Path(cwd + '/config.py')
        #elif ops == 'Ubuntu' or ops == 'debian' or ops == 'darwin':
        #    variableCheck = Path()
        print(variableCheck)
        print(cwd)
        variableCheck.is_file()
        if variableCheck.is_file():
            print('Configuration file has been loaded...')

            self.mmenu()
        else:
            print('Ello, you have some configurations to do!')
            self.mmenu()


            #Use Radio buttons for the yes or no questions



            #self.header = Label(text='First Time Configuration')
            #self.header.config(font=24, background='lightblue')
            #self.header.grid(row=0, sticky='W')
            #self.qalbl = Label(text='Does your system require the following services?')
            #self.qalbl.config(font=14, background='lightblue')
            #self.qalbl.grid(row=1)

            # Box Labels
            #self.sshlbl = Label(text='SSH [y/n] : ')
            #self.sshlbl.config(background='lightblue')
            #self.sshlbl.grid(row=2, column=0, sticky='W', pady='20')
            #self.ftplbl = Label(text='FTP [y/n] : ')
            #self.ftplbl.config(background='lightblue')
            #self.ftplbl.grid(row=3, column=0, sticky='W')

            # Entry/Checkbuttons Boxes

            #self.ssh1 = Checkbutton(root, text='yes', offvalue=0, onvalue=1)
            #self.ssh1.config()
            #self.ssh2 = Checkbutton(root, text='no', offvalue=0, onvalue=1)
            #self.ssh2.config()
            #self.ssh1.grid(row=2, column=0, sticky='W', padx='60')
            #self.ssh2.grid(row=2, column=0, sticky='W', padx='100')

            #self.ftp1 = Entry(textvariable=self.ftp)
            #self.ftp1.grid(row=3, column=0, sticky='W', padx='60')


            #self.cont = Button(text='Continue', command=self.mmenu)
            #self.cont.grid(row=1, column=1, sticky='SW')

    def mmenu(self):

        buttonNames = ['Search For Prohibited Media', 'Updates', 'User / Group Settings', 'Firewall Settings', 'Services Settings', 'Malware Removal', 'Audit System', 'Basic Configurations', 'Remove Prohibited Software']
        thred = ThreadmmFunc()

        commands = [thred.threaderSRCH, thred.threaderUPDT, chngTOusr, thred.threaderFWL, thred.threaderServ, thred.threaderMALREM, thred.threaderALYN, thred.threaderBASEconf, thred.threaderRMproCont]
        # commands = [mmfunc.search_media, mmfunc.updates, self.usrgru]
        gridrow = ['5', '1', '1', '2', '2', '3', '4', '4', '3']
        gridcolumn = ['0', '0', '1', '0', '1', '1', '0', '1', '0']

        fontSize = tkFont.Font(size=24)

        for i in range(0, len(buttonNames)):
            self.quit = Button(footerR, text='Quit', width=20, command=root.destroy)
            self.header = Label(fHead, text='Main Menu')
            self.buttons.append(Button(fCont, text=buttonNames[i], width='40', command=commands[i]))
            if self.mmd == 0:
                self.quit.grid_rowconfigure(0, weight=1)
                self.quit.grid(row=0, sticky='E', padx=30, pady=10)
                self.header.config(font=fontSize, background='lightblue')
                self.header.grid(row=0, sticky='WE', padx=30, pady=20)
                self.buttons[i].grid(row=gridrow[i], column=gridcolumn[i], pady='2', padx='5')
                print('In Main Menu')
            else:
                self.quit.grid_remove()
                self.header.grid_remove()
                self.buttons[i].grid_remove()

    def usrgru(self):
        buttonNames = ['Add User to System', 'Remove User from System', 'Add Group to System', 'Remove Group from System', 'Add User to Group', 'Remove User from Group', 'List Local Users', 'List Local Groups', 'List Members of Group', 'List the Groups an User is in', 'Change all Users Passwords at Once']
        gridrow = ['1', '1', '2', '2', '3', '3', '4', '4', '5', '5', '6']
        gridcolumn = ['0', '1', '0', '1', '0', '1', '0', '1', '0', '1', '0']
        usrgru = ugmfunctions.usrGruFunc()
        targets = [usrgru.addusr, usrgru.rmuser, usrgru.adgru, usrgru.rmgru, usrgru.adusrtogru, usrgru.rmusrfrogru, usrgru.lslocausr, usrgru.lslocagru, usrgru.lsmemgru, usrgru.lsgruusrin, usrgru.chngusrspass]

        fontSize = tkFont.Font(size=24)

        for i in range(0, len(buttonNames)):
            self.Back = Button(footerL, text='Back to Main Menu', width=20, command=chngTOmm)
            self.header2 = Label(fHead, text='User and Group Settings')
            self.usrgrubuttons.append(Button(fCont, text=buttonNames[i], width='40', command= lambda i=i: threader(i)))
            if self.usrd == 0:
                self.Back.grid(row=0, sticky='W', padx=30, pady=10)
                self.Back.grid_rowconfigure(1, weight=0)
                self.header2.grid(row=0, sticky='WE', padx=30, pady=20)
                self.header2.config(font=fontSize, background='lightblue')
                self.usrgrubuttons[i].grid(row=gridrow[i], column=gridcolumn[i], pady='2', padx='5')
                print('In user / group menu')
            else:
                print('starting to delete buttons')
                self.Back.grid_remove()
                self.header2.grid_remove()
                self.usrgrubuttons[i].grid_remove()

        def threader(com):
            try:
                print(targets[com])
                threader = Thread(target=targets[com])
                threader.start()
            except Exception as e:
                print(e)
                print('Could not start thread')

###################################################

if __name__ == '__main__':
    # creation of GUI
    root = Tk()
    root.title('Apple CIDR Script Runner')
    root.iconphoto(False, tk.PhotoImage(file="cup2.png"))
    if platform == 'win32':
        root.geometry("520x350")
    elif ops == 'darwin':
        root.geometry("735x350")
    else:
        root.geometry("680x350")
    # root.resizable(0, 0)
    root.grid_rowconfigure(0, weight=1)
    root.grid_columnconfigure(0, weight=1)

    fCont = tk.Frame(root, background='black')
    fCont.grid(row=0, column=0, columnspan=2, sticky='nsew')
    #fCont.grid_rowconfigure(1, weight=1)
    #fCont.grid_columnconfigure(0, weight=1)

    fHead = tk.Frame(fCont, background='blue')
    fHead.grid(row=0, column=0, columnspan=2, sticky='ew')

    footerR = tk.Frame(root, background='red')
    footerR.grid(row=1, column=1, sticky='se')

    footerL = tk.Frame(root, background='green')
    footerL.grid(row=1, column=0, sticky='sw')

    menubar = Menu(root)
    aboutmenu = Menu(menubar, tearoff=0)
    menubar.add_cascade(label='Help', menu=aboutmenu)
    aboutmenu.add_command(label='About Creator', command=aboutCreators)
    aboutmenu.add_command(label='How To Use', command=aboutHowtoUse)


    main = scriptrunnerGUI()
    root.configure(menu=menubar, bg='lightblue')
    root.mainloop()

####################################################