...

import os
import re
from pathlib import Path

import pytest

SIMPLE_VERSION_REGEX = re.compile(r'([0-9]+)\.([0-9]+)\.([0-9]+)')
ILLEGAL_FILENAME_CHARS = r'\/:*?"<>|'

def shorten_path(path: Path) -> Path:
    """
    Returns the path relative to the current working directory if possible. Otherwise return the path unchanged.
    """
    try:
        return path.relative_to(os.getcwd())
    except ValueError:
        return path

def get_valid_filename(s: str) -> str:
    """
    Return the given string converted to a string that can be used for a clean filename.
    Taken from https://github.com/django/django/blob/master/django/utils/text.py
    """
    s = str(s).strip().replace(' ', '_')
    s = re.sub(r'(?u)[^-\w.]', '', s)
    s = {'': 'empty', '.': 'dot', '..': 'dotdot'}.get(s, s)
    return s

def might_be_valid_filename(s: str) -> bool:
    """
    Returns false if the given string is definitely a path traversal or not a valid filename.
    Returns true if the string might be a valid filename.

Note: This isn't secure, it just catches most accidental path traversals or invalid filenames.
    """
    return not (
        len(s) == 0
        or s == '.'
        or s == '..'
        or any(c in s for c in ILLEGAL_FILENAME_CHARS)
    )

def simple_version_parse(version: str):
    """
    Returns a 3 tuple of the versions major, minor, and patch.
    Raises a value error if the version string is unsupported.
    """
    match = SIMPLE_VERSION_REGEX.match(version)
    if match is None:
        raise ValueError('Unsupported version format')

return tuple(int(part) for part in match.groups())

def _pytest_expected_on_right() -> bool:
    """
    Returns true if pytest prints string diffs correctly for:

assert tested_value == expected_value

Returns false if pytest prints string diffs correctly for:

assert expected_value == tested_value
    """
    # pytest diffs before version 5.4.0 assumed expected to be on the left hand side.
    try:
        pytest_version = simple_version_parse(pytest.__version__)
    except ValueError:
        return True
    else:
        return pytest_version >= (5, 4, 0)

def flatten_dict(d: dict):
    """
    Returns the flattened dict representation of the given dict.

Example:

>>> flatten_dict({
        ...     'a': 1,
        ...     'b': {
        ...         'c': 2
        ...     },
        ...     'd': {},
        ... })
        [(['a'], 1), (['b', 'c'], 2)]
    """
    assert type(d) is dict
    result = []
    _flatten_dict(d, result, [])
    return result

def _flatten_dict(obj, result, prefix):
    if type(obj) is dict:
        for k, v in obj.items():
            prefix.append(k)
            _flatten_dict(v, result, prefix)
            prefix.pop()
    else:
        result.append((list(prefix), obj))

def flatten_filesystem_dict(d):
    """
    Returns the flattened dict of a nested dictionary structure describing a filesystem.

Raises ``ValueError`` if any of the dictionary keys are invalid filenames.

Example:

>>> flatten_filesystem_dict({
        ...     'file1.txt': '111',
        ...     'dir1': {
        ...         'file2.txt': '222'
        ...     },
        ... })
        {'file1.txt': '111', 'dir1/file2.txt': '222'}
    """
    result = {}
    for key_list, obj in flatten_dict(d):
        for i, key in enumerate(key_list):
            if not might_be_valid_filename(key):
                key_list_str = ''.join('[{!r}]'.format(k) for k in key_list[:i])
                raise ValueError('Key {!r} in d{} must be a valid file name.'.format(key, key_list_str))
        result['/'.join(key_list)] = obj

return result

.	Edit
..	Edit
__init__.py	Edit
__pycache__	Edit
_utils.py	Edit
_version.py	Edit
plugin.py	Edit