cas.py

import subprocess

#Python3.8
# Initial function to loop valid options
def start():
    menu_choice = 99
    print()
    print("***********************************************************************")
    print("*              Welcome to CaS - Command and Script v1.0.0             *")
    print("* Dependency: Run with sudo. Install screen (sudo apt install screen) *")
    print("* by https://github.com/fpatrick/cas                          and dxT *")
    print("***********************************************************************")
    print()
    print(">>> Choose an option: | 1 - Run commands in background | 2 - Create script to run on Cron | 0 - Exit ")

    menu_choice = int(input())
    while menu_choice > 2:
        menu_choice = int(input(">>> Please choose a valid option (0 to 2): "))

    return menu_choice


# Call initial function to start program
menu = start()


# Make folder to store scripts
def make_folder():
    print(">>> Please enter *!* FULL PATH *!* where the script will be created (ex: /example/yourfolder):")
    folder_path = input()
    command = "mkdir -p " + folder_path
    result_command = subprocess.run(command, shell=True, capture_output=True, text=True)
    if result_command.returncode == 0:
        return folder_path
    else:
        print(">>> The folder couldn't be created due to an error:")
        print(result_command.stderr)
        return folder_path


# Make file and add the user commands on provided folder
def make_file(user_folder, user_command):
    print(">>> What will be the script name?")
    script_name = input()
    create_lock = "touch " + user_folder + "/" + script_name + ".lock && "
    remove_lock = " ; rm -f " + user_folder + "/" + script_name + ".lock"
    screen_command = "screen -dmS section_created_by_script  bash -c " + "'" + create_lock + user_command + remove_lock + "'"

    lock = user_folder + "/" + script_name + ".lock"
    try:
        f_path = user_folder + '/' + script_name + '.py'
        with open(f_path, 'w') as f:
            f.write('import subprocess\n')
            f.write('import os.path\n')
            f.write('from datetime import datetime\n')
            f.write('now = datetime.now()\n')
            f.write('dt_string = now.strftime("%d/%m/%Y %H:%M:%S")\n')
            f.write('if os.path.isfile("' + lock + '"):\n')
            f.write('    print(f"{dt_string}: ' + script_name + '.lock found, the command is still running. Delete lock file to force run.")\n')
            f.write('else:\n')
            f.write('    subprocess.run("' + screen_command + '", shell=True, capture_output=True, text=True)\n')

        print(f"\n*************** Script created on {f_path} ***************\n")
        return f_path
    except:
        print(">>> The script couldn't be created")


def close_program():
    print(">>> Would you like to exit this program? y / n")
    choice = input()
    while choice != "y" and choice != "n":
        choice = input(">>> Please write y or n: ")

    if choice == "y":
        exit()
    else:
        return start()


# If choice is just to run a command
if menu == 1:
    print("\n>>> * Info: you can install screen with sudo apt install screen *\n")
    print(">>> What command would you like to run?: ")
    userCommand = input()
    screenCommand = "screen -dmS section_created_by_script  bash -c " + "'" + userCommand + "'"
    subprocess.run(screenCommand, shell=True, capture_output=True, text=True)
    result = subprocess.run("screen -ls", shell=True, capture_output=True, text=True)
    print(">>> * Info: Your screen section may remain open when command finish. ( sudo screen -list  and then sudo screen kill <number>) *")
    print()
    print(result.stdout)
    input("\n\nPress enter to exit. ")

# If choice is to create a script
elif menu == 2:
    print(">>> What command would you like to run? ")
    userCommand = input()

    folder = make_folder()
    fullPath = make_file(folder, userCommand)
    print(">>> Instructions on how to make a cron job with the script created:\n")
    print("> Enter https://crontab.guru (and get the schedule)")
    print("> Enter on terminal: which python3 (and get the python's path)")
    print("> Enter on terminal: sudo crontab -e")
    print("> Write on same line: <schedule> <python3 path> <script full path>  <optional log>")
    print(f"> Example: * * * * * /usr/bin/python3 /folders/{fullPath} >> ~/cron.log 2>&1 \n\n")
    print("> *!* ATTENTION *!* recreate the script if you need to move it to another folder!!! \n\n")

    input("Press enter to exit. ")

elif menu == 0:
    exit()