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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
import re
from datetime import datetime, timezone
from uuid import UUID
# TODO use pytest_assertrepr_compare
class CloseToNow:
def __init__(self, delta=2):
self.delta: float = delta
self.now = datetime.utcnow()
self.match = False
self.other = None
def __eq__(self, other):
self.other = other
if not isinstance(other, datetime):
try:
from pydantic.v1.datetime_parse import parse_datetime
except ImportError: # pragma: no cover
raise ImportError('pydantic is required to use CloseToNow, please run `pip install pydantic`')
other = parse_datetime(other)
if other.tzinfo:
self.now = self.now.replace(tzinfo=timezone.utc)
self.match = -self.delta < (self.now - other).total_seconds() < self.delta
return self.match
def __repr__(self):
if self.match:
# if we've got the correct value return it to aid in diffs
return repr(self.other)
else:
# else return something which explains what's going on.
return f'<CloseToNow(delta={self.delta}, now={self.now:%Y-%m-%dT%H:%M:%S})>'
class AnyInt:
def __init__(self):
self.v = None
def __eq__(self, other):
if type(other) == int:
self.v = other
return True
def __repr__(self):
if self.v is None:
return '<AnyInt>'
else:
return repr(self.v)
class RegexStr:
re = re
def __init__(self, regex, flags=re.S):
self._regex = re.compile(regex, flags=flags)
self.v = None
def __eq__(self, other):
if self._regex.fullmatch(other):
self.v = other
return True
return False
def __repr__(self):
if self.v is None:
return f'<RegexStr(regex={self._regex!r}>'
else:
return repr(self.v)
class IsUUID:
def __init__(self):
self.v = None
def __eq__(self, other):
if isinstance(other, UUID):
self.v = other
return True
# could also check for regex
def __repr__(self):
return repr(self.v) if self.v else '<UUID(*)>'
|
