screenbot.py

from pynput import keyboard
import slack_sdk, os, pyautogui, sys, win32api, win32con

def on_press(key):
    if key == keyboard.Key.print_screen:
        print('posting')
        myScreenshot = pyautogui.screenshot()
        myScreenshot.save('screen.png')
        try:
            client.files_upload_v2(channel=channel,
                file='screen.png',
                title='Test upload')
        except slack_sdk.errors.SlackApiError:
            e = sys.exc_info()
            print("Failed to upload to slack, maybe the token or channel are not correct?")
            print("Error traceback: ", e)

def setenv_var(varname, value):
    try:
        rkey = win32api.RegOpenKeyEx(win32con.HKEY_LOCAL_MACHINE, 'SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment',0 ,win32con.KEY_WRITE)
        try:
            win32api.RegSetValueEx(rkey, varname, 0, win32con.REG_SZ, value)
            print("Saved ", varname)
            return True
        except Exception:
            pass
    except Exception:
        print("No access to save environment variables, if you want to save them, please run this once in administrator mode")
        print("Running bot without save\n")
    finally:
        try:
            win32api.RegCloseKey(rkey)
        except UnboundLocalError:
            pass
    return False

if __name__ == "__main__":
    slack_token = ''
    channel = ''

    save = input("Would you like to store/edit your token and/or channel? (y/N): ")
    if str(save).lower() == 'y':
        token = input("Slack bot token (leave empty if you don't want to change it): ")
        if token != "":
            setenv_var("SLACK_BOT_TOKEN", token)
        chan = input("Slack channel ID (Open channel details, then the ID will be at the bottom) (leave empty if you don't want to change it): ")
        if chan != "":
            setenv_var("SLACK_CHANNEL", chan)
        print('Restart the program to reload your environment variables')
        sys.exit()

    if os.environ.get('SLACK_BOT_TOKEN') is None or os.environ.get('SLACK_CHANNEL') is None:
        print("Slack token not found in environment variables, make sure the variable is called 'SLACK_BOT_TOKEN' and 'SLACK_CHANNEL'\n")
        sys.exit()

    slack_token = os.environ.get('SLACK_BOT_TOKEN')
    channel = os.environ.get('SLACK_CHANNEL')

    print("Bot is ready")
    print("Press the 'print screen' button to send a screenshot, press 'escape' to quit")

    client = slack_sdk.WebClient(slack_token)

    with keyboard.Listener(on_press=on_press) as listener: listener.join()