OneDrivePath.py

import os
import sys
from pathlib import Path

# Determine local OneDrive path, accounting for multiple OneDrive accounts
# (e.g. personal + one or more business/school accounts)


def _normalize(s):
    return "".join(ch for ch in s.lower() if ch.isalnum())


def _win_accounts():
    # Registry has one subkey per signed-in account, so it's the only way to
    # see every account (env vars only expose one personal + one business).
    # Matched by folder name, same as macOS/Linux.
    accounts = []
    try:
        import winreg
        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r"Software\Microsoft\OneDrive\Accounts")
        for i in range(winreg.QueryInfoKey(key)[0]):
            with winreg.OpenKey(key, winreg.EnumKey(key, i)) as sub:
                try:
                    folder = Path(winreg.QueryValueEx(sub, "UserFolder")[0])
                except OSError:
                    continue
                accounts.append((folder.name, folder))
    except OSError:
        pass
    if not accounts:
        for var in ("OneDriveConsumer", "OneDriveCommercial", "OneDrive"):
            val = os.getenv(var)
            if val:
                accounts.append((Path(val).name, Path(val)))
    return accounts


def _mac_accounts():
    accounts = []
    cloud_storage = Path.home() / "Library" / "CloudStorage"
    if cloud_storage.exists():
        for folder in sorted(cloud_storage.glob("OneDrive-*")):
            accounts.append((folder.name, folder))
    legacy = Path.home() / "OneDrive"
    if legacy.exists():
        accounts.append((legacy.name, legacy))
    return accounts


def _linux_accounts():
    # No official Linux client; unofficial clients (e.g. abraunegg/onedrive)
    # default to syncing a single account into ~/OneDrive
    folder = Path.home() / "OneDrive"
    return [(folder.name, folder)] if folder.exists() else []


_default_account = None


def _resolve(account):
    if sys.platform == "win32":
        accounts = _win_accounts()
    elif sys.platform == "darwin":
        accounts = _mac_accounts()
    else:
        accounts = _linux_accounts()

    accounts = [(label, path) for label, path in accounts if path.exists()]

    if account is not None:
        needle = _normalize(account)
        accounts = [a for a in accounts if needle in _normalize(a[0])]
        if not accounts:
            raise FileNotFoundError(f"No OneDrive account matching '{account}' found")
    elif not accounts:
        raise FileNotFoundError("OneDrive folder not found")

    return accounts[0][1]


def setAccount(account: str) -> None:
    # Validate the account exists now, so later onedrivePath() calls don't
    # surprise you with a lookup failure
    global _default_account
    _resolve(account)
    _default_account = account


def onedrivePath(name: str = None, account: str = None) -> str:
    local_onedrive_path = _resolve(account if account is not None else _default_account)

    # Append the relative path to the local OneDrive path
    if name is not None:
        local_onedrive_path /= name.lstrip("/\\")

    return str(local_onedrive_path)


# Now you can use local relative paths within this directory