proxy-scraper-checker/proxy_scraper_checker/fs.py

from __future__ import annotations

import logging
from typing import TYPE_CHECKING

import platformdirs

if TYPE_CHECKING:
    from pathlib import Path

_logger = logging.getLogger(__name__)
CACHE_PATH = platformdirs.user_cache_path("proxy_scraper_checker")


def add_permission(
    path: Path, permission: int, /, *, missing_ok: bool = False
) -> None:
    try:
        current_permissions = path.stat().st_mode
        new_permissions = current_permissions | permission
        if current_permissions != new_permissions:
            path.chmod(new_permissions)
            _logger.info(
                "Changed permissions of %s from %o to %o",
                path,
                current_permissions,
                new_permissions,
            )
    except FileNotFoundError:
        if not missing_ok:
            raise


def create_or_fix_dir(path: Path, /, *, permission: int) -> None:
    try:
        path.mkdir(parents=True)
    except FileExistsError:
        if not path.is_dir():
            msg = f"{path} is not a directory"
            raise ValueError(msg) from None
        add_permission(path, permission)