Creating Shell Shortcuts

The winshell module provides direct access to Windows special folders, simpliyfing the generation of links for the desktop and Start menu. Rather than manually resloving paths via ctypes, developers can utilize built-in helpers to locate target directories.

Unified Shortcut Generator

The following utility function handles link creation for both the desktop and the Start menu. It dynamically resolves the target directory based on the specified location parameter.

import os
import winshell
from pathlib import Path

def build_shell_link(file_name, destination='desktop', icon_file=None):
    work_dir = Path.cwd()
    executable_path = work_dir / file_name

    if destination == 'desktop':
        folder_path = Path(winshell.desktop())
    elif destination == 'start_menu':
        folder_path = Path(winshell.programs())
    else:
        raise ValueError("Unsupported destination")

    link_location = folder_path / f"{file_name}.lnk"

    icon_config = icon_file if icon_file else (str(executable_path), 0)

    winshell.CreateShortcut(
        Path=str(link_location),
        Target=str(executable_path),
        Icon=icon_config,
        Description="Automated Shell Link",
        StartIn=str(work_dir),
        Arguments="--launch-mode"
    )

# Example usage
build_shell_link('main_script.py', destination='desktop')

Handling Command-Line Arguments

When a shortcut includes arguments, the target script receives them via sys.argv. The Arguments field in the shortcut properties appends data to the execution command.

import sys

# Retrieve passed arguments
params = sys.argv

# Example output if launched with '--launch-mode':
# ['C:\\Path\\main_script.py', '--launch-mode']
print(params)

Implementing System Tray Icons

For background operations, integrating an icon into the system tray allows users to interact with the process without a visible window. The pystray library, combined with Pillow for image handling, facilitates this functionality.

Tray Implementation Logic

The implementation requires defining an icon image, a menu structure, and callback functions for menu interactions.

import pystray
from PIL import Image
import sys

def process_menu_selection(tray, item):
    selection = str(item)
    if selection == "Run Diagnostic":
        print(">>> Running Diagnostic")
    elif selection == "Show Status":
        print(">>> Status: OK")
    elif selection == "Terminate":
        tray.stop()
        sys.exit(0)

def generate_icon_image():
    # Create a solid blue 64x64 image
    return Image.new('RGB', (64, 64), color='blue')
    # Alternatively, load an existing file:
    # return Image.open('icon.png')

def start_tray_service():
    tray_icon = pystray.Icon(
        "BackgroundService",
        generate_icon_image(),
        "Service Active",
        menu=pystray.Menu(
            pystray.MenuItem("Run Diagnostic", process_menu_selection),
            pystray.MenuItem("Show Status", process_menu_selection),
            pystray.MenuItem("Terminate", process_menu_selection)
        )
    )
    tray_icon.run()

if __name__ == "__main__":
    start_tray_service()