Mercurial > libervia-backend

comparison sat/tools/common/utils.py @ 3524:584379473925

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
tools (common/utils): new methods to parse and generate file size with symbols: - `parseSize` lets parse a size which can be either a int, or a string with a symbol like `Mio` - `getSizeMultiplier` returns a tuple of size with it's multiplier symbol - `getHumanSize` returns a string with size and adapted multiplier.
author Goffi <goffi@goffi.org>
date Wed, 05 May 2021 15:37:33 +0200
parents 7550ae9cfbac
children 524856bd7b19
comparison
equal deleted inserted replaced
3523:30779935c0aa 3524:584379473925
17 # along with this program. If not, see <http://www.gnu.org/licenses/>. 17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
18 18
19 """Misc utils for both backend and frontends""" 19 """Misc utils for both backend and frontends"""
20 20
21 import collections.abc 21 import collections.abc
22 size_units = {
23 "b": 1,
24 "kb": 1000,
25 "mb": 1000**2,
26 "gb": 1000**3,
27 "tb": 1000**4,
28 "pb": 1000**5,
29 "eb": 1000**6,
30 "zb": 1000**7,
31 "yb": 1000**8,
32 "o": 1,
33 "ko": 1000,
34 "mo": 1000**2,
35 "go": 1000**3,
36 "to": 1000**4,
37 "po": 1000**5,
38 "eo": 1000**6,
39 "zo": 1000**7,
40 "yo": 1000**8,
41 "kib": 1024,
42 "mib": 1024**2,
43 "gib": 1024**3,
44 "tib": 1024**4,
45 "pib": 1024**5,
46 "eib": 1024**6,
47 "zib": 1024**7,
48 "yib": 1024**8,
49 "kio": 1024,
50 "mio": 1024**2,
51 "gio": 1024**3,
52 "tio": 1024**4,
53 "pio": 1024**5,
54 "eio": 1024**6,
55 "zio": 1024**7,
56 "yio": 1024**8,
57 }
22 58
23 59
24 def per_luminance(red, green, blue): 60 def per_luminance(red, green, blue):
25 """Caculate the perceived luminance of a RGB color 61 """Caculate the perceived luminance of a RGB color
26 62
71 except KeyError: 107 except KeyError:
72 pass 108 pass
73 109
74 def update(self, items): 110 def update(self, items):
75 self._dict.update({i: None for i in items}) 111 self._dict.update({i: None for i in items})
112
113
114 def parseSize(size):
115 """Parse a file size with optional multiple symbole"""
116 try:
117 return int(size)
118 except ValueError:
119 number = []
120 symbol = []
121 try:
122 for c in size:
123 if c == " ":
124 continue
125 if c.isdigit():
126 number.append(c)
127 elif c.isalpha():
128 symbol.append(c)
129 else:
130 raise ValueError("unexpected char in size: {c!r} (size: {size!r})")
131 number = int("".join(number))
132 symbol = "".join(symbol)
133 if symbol:
134 try:
135 multiplier = size_units[symbol.lower()]
136 except KeyError:
137 raise ValueError(
138 "unknown size multiplier symbole: {symbol!r} (size: {size!r})")
139 else:
140 return number * multiplier
141 return number
142 except Exception as e:
143 raise ValueError(f"invalid size: {e}")
144
145
146 def getSizeMultiplier(size, suffix="o"):
147 """Get multiplier of a file size"""
148 size = int(size)
149 #  cf. https://stackoverflow.com/a/1094933 (thanks)
150 for unit in ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi"]:
151 if abs(size) < 1024.0:
152 return size, f"{unit}{suffix}"
153 size /= 1024.0
154 return size, f"Yi{suffix}"
155
156
157 def getHumanSize(size, suffix="o", sep=" "):
158 size, symbol = getSizeMultiplier(size, suffix)
159 return f"{size:.2f}{sep}{symbol}"