summaryrefslogtreecommitdiffstats
path: root/contrib/python/PyHamcrest/py3/hamcrest/library/text/stringmatches.py
blob: e7e8dd4190f26eec08ed9f5b67a8adf2999707de (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import re
from typing import Pattern, Union

from hamcrest.core.base_matcher import BaseMatcher
from hamcrest.core.description import Description
from hamcrest.core.matcher import Matcher

__author__ = "Chris Rose"
__copyright__ = "Copyright 2011 hamcrest.org"
__license__ = "BSD, see License.txt"


class StringMatchesPattern(BaseMatcher[str]):
    def __init__(self, pattern) -> None:
        self.pattern = pattern

    def describe_to(self, description: Description) -> None:
        description.append_text("a string matching '").append_text(
            self.pattern.pattern
        ).append_text("'")

    def _matches(self, item: str) -> bool:
        return self.pattern.search(item) is not None


def matches_regexp(pattern: Union[str, Pattern[str]]) -> Matcher[str]:
    """Matches if object is a string containing a match for a given regular
    expression.

    :param pattern: The regular expression to search for.

    This matcher first checks whether the evaluated object is a string. If so,
    it checks if the regular expression ``pattern`` matches anywhere within the
    evaluated object.

    """
    if isinstance(pattern, str):
        pattern = re.compile(pattern)

    return StringMatchesPattern(pattern)