work_connect.py - master - rnekrasov/Python

work_connect.py
65 строк3 KB
slowy07
refactor: clean code
30 янв 2022, 04:33
30 янв 2022, 04:33f0af0c4
# Script Name		: work_connect.py
# Author				: Craig Richards
# Created				: 11th May 2012
# Last Modified		: 31st October 2012
# Version				: 1.1

# Modifications		: 1.1 - CR - Added some extra code, to check an argument is passed to the script first of all, then check it's a valid input

# Description			: This simple script loads everything I need to connect to work etc

import os  # Load the Library Module
import subprocess  # Load the Library Module
import sys  # Load the Library Module
import time  # Load the Library Module

dropbox = os.getenv(
    "dropbox"
)  # Set the variable dropbox, by getting the values of the environment setting for dropbox
rdpfile = "remote\\workpc.rdp"  # Set the variable logfile, using the arguments passed to create the logfile
conffilename = os.path.join(
    dropbox, rdpfile
)  # Set the variable conffilename by joining confdir and conffile together
remote = (
    r"c:\windows\system32\mstsc.exe "  # Set the variable remote with the path to mstsc
)

text = """You need to pass an argument
	-c Followed by login password to connect
	-d to disconnect"""  # Text to display if there is no argument passed or it's an invalid option - 1.2

if len(sys.argv) < 2:  # Check there is at least one option passed to the script - 1.2
    print(text)  # If not print the text above - 1.2
    sys.exit()  # Exit the program - 1.2

if (
    "-h" in sys.argv or "--h" in sys.argv or "-help" in sys.argv or "--help" in sys.argv
):  # Help Menu if called
    print(text)  # Print the text, stored in the text variable - 1.2
    sys.exit(0)  # Exit the program
else:
    if sys.argv[1].lower().startswith("-c"):  # If the first argument is -c then
        passwd = sys.argv[
            2
        ]  # Set the variable passwd as the second argument passed, in this case my login password
        subprocess.Popen(
            (
                r"c:\Program Files\Checkpoint\Endpoint Connect\trac.exe connect -u username -p "
                + passwd
            )
        )
        subprocess.Popen((r"c:\geektools\puttycm.exe"))
        time.sleep(
            15
        )  # Sleep for 15 seconds, so the checkpoint software can connect before opening mstsc
        subprocess.Popen([remote, conffilename])
    elif (
        sys.argv[1].lower().startswith("-d")
    ):  # If the first argument is -d then disconnect my checkpoint session.
        subprocess.Popen(
            (r"c:\Program Files\Checkpoint\Endpoint Connect\trac.exe disconnect ")
        )
    else:
        print(
            "Unknown option - " + text
        )  # If any other option is passed, then print Unknown option and the text from above - 1.2