glob_load.py

from pymol import cmd
import glob
import os

def glob_load(pattern, quiet=0):
    """
    Load all files matching a glob pattern.
    
    USAGE:
        glob_load pattern
    
    EXAMPLES:
        glob_load *.pdb
        glob_load /path/to/folder/*.cif
        glob_load nipah_pT10_*.cif
    """
    files = glob.glob(os.path.expanduser(pattern))
    
    if not files:
        print(f"Error: No files found matching pattern: {pattern}")
        return
    
    if not quiet:
        print(f"Loading {len(files)} file(s)...")
    
    for filepath in files:
        object_name = os.path.splitext(os.path.basename(filepath))[0]
        cmd.load(filepath, object_name)
        if not quiet:
            print(f"  Loaded: {object_name}")
    
    if not quiet:
        print(f"Successfully loaded {len(files)} structure(s)")

cmd.extend("glob_load", glob_load)