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
|
# Copyright 2010-2015 Lars Wirzenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import re
import obnamlib
class UnitError(obnamlib.ObnamError):
pass
class SizeSyntaxError(UnitError):
msg = '"{size}" is not a valid size'
class UnitNameError(UnitError):
msg = '"{unit}" is not a valid unit'
class ByteSizeParser(object):
'''Parse sizes of data in bytes, kilobytes, kibibytes, etc.'''
pat = re.compile(r'^(?P<size>\d+(\.\d+)?)\s*'
r'(?P<unit>[kmg]?i?b?)?$', re.I)
units = {
'b': 1,
'k': 1000,
'kb': 1000,
'kib': 1024,
'm': 1000**2,
'mb': 1000**2,
'mib': 1024**2,
'g': 1000**3,
'gb': 1000**3,
'gib': 1024**3,
}
def __init__(self):
self.set_default_unit('B')
def set_default_unit(self, unit):
if unit.lower() not in self.units:
raise UnitNameError(size=unit)
self.default_unit = unit
def parse(self, string):
m = self.pat.match(string)
if not m:
raise SizeSyntaxError(size=string)
size = float(m.group('size'))
unit = m.group('unit')
if not unit:
unit = self.default_unit
elif unit.lower() not in self.units:
raise UnitNameError(unit=unit)
factor = self.units[unit.lower()]
return int(size * factor)
|
