How to Map Network Drive Paths to Local Paths in Python

"""Map mock SMB network drive paths to local placeholder paths."""
from dataclasses import dataclass

@dataclass(frozen=True)
class NetworkDrive:
    letter: str
    remote_path: str

DRIVES = {
    "S:": NetworkDrive("S", r"\\server01\shares\sales"),
    "M:": NetworkDrive("M", r"\\server02\media\movies"),
    "X:": NetworkDrive("X", r"\\nas\public\documents"),
}

LOCAL_BASE = {"S": "/mnt/sales", "M": "/mnt/media", "X": "/mnt/docs"}

def smb_to_local(smb_path: str, drives: dict = DRIVES) -> str:
    """Convert a mock SMB path (e.g. 'S:\\reports\\q1.xlsx') to a local path."""
    if not smb_path or smb_path[0] not in drives:
        raise ValueError(f"Drive letter not mapped: {smb_path}")
    letter = smb_path[0]
    rest = smb_path[2:].replace("\\", "/")
    return f"{LOCAL_BASE[letter]}/{rest}".rstrip("/")

def local_to_smb(local_path: str) -> str:
    """Convert a mock local path back to an SMB path."""
    for letter, base in LOCAL_BASE.items():
        if local_path.startswith(base + "/") or local_path == base:
            suffix = local_path[len(base):].replace("/", "\\")
            return drives[letter + ":"].remote_path + suffix
    raise ValueError(f"No matching drive for path: {local_path}")

if __name__ == "__main__":
    test_paths = [
        "S:\\reports\\q1.xlsx",
        "M:\\movies\\inception\\movie.mp4",
        "S:",
    ]
    for p in test_paths:
        local = smb_to_local(p)
        back = local_to_smb(local)
        print(f"{p!r} -> {local!r} -> {back!r}")