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
|
import argparse
import re
from abc import ABCMeta, abstractmethod
import dateutil.relativedelta
from quarg.quassel.types import BufferType, MessageFlag, MessageType
from quarg.utils import errx, parse_isodate
class ParseEnum(argparse.Action, metaclass=ABCMeta):
def __call__(self, parser, namespace, value, option_string=None):
key = value.upper()
if key not in self.enumclass.__members__:
possible = ', '.join([e.name.lower() for e in self.enumclass])
desc = self.enumclass.describe()
errx(f'Not a valid {desc}: {value}\nPossible {desc}s are: {possible}')
saved = getattr(namespace, self.dest) or []
saved.append(self.enumclass[key])
setattr(namespace, self.dest, saved)
@property
@abstractmethod
def enumclass(self):
pass
class ParseMessageType(ParseEnum):
@property
def enumclass(self):
return MessageType
class ParseMessageFlag(ParseEnum):
@property
def enumclass(self):
return MessageFlag
class ParseBufferType(ParseEnum):
@property
def enumclass(self):
return BufferType
class ParseDate(argparse.Action):
def __call__(self, parser, namespace, datespec, option_string=None):
setattr(namespace, self.dest, parse_isodate(datespec))
class ParseOrder(argparse.Action):
def __call__(self, parser, namespace, orderspec, option_string=None):
if orderspec not in ['asc', 'desc']:
errx(f'Invalid order \'{orderspec}\'. Possible values are: asc, desc')
setattr(namespace, self.dest, orderspec)
class ParseAround(argparse.Action):
def __call__(self, parser, namespace, aroundspec, option_string=None):
if '/' in aroundspec:
datespec, rangespec = aroundspec.split('/', 1)
if not rangespec:
errx('Missing range for --around')
match = re.match(r'^(?P<range>\d+)(?P<unit>[hm])?$', rangespec)
if not match:
errx(f'Invalid range for --around: \'{rangespec}\'')
unit = match.group('unit')
try:
time_range = int(match.group('range'))
except ValueError as err:
errx(f'Error when parsing range for --around: {err}')
else:
datespec, time_range, unit = (aroundspec, 12, 'h')
date = parse_isodate(datespec)
if unit == 'm':
offset = dateutil.relativedelta.relativedelta(minutes=time_range)
else:
offset = dateutil.relativedelta.relativedelta(hours=time_range)
setattr(namespace, self.dest, (date - offset, date + offset))
|
