25 lines
624 B
Python
25 lines
624 B
Python
import os,re
|
|
|
|
table = str.maketrans({
|
|
"<": "<",
|
|
">": ">",
|
|
"&": "&",
|
|
"'": "'",
|
|
'"': """,
|
|
})
|
|
def xmlesc(txt):
|
|
return txt.translate(table)
|
|
|
|
def get_size(file_path, unit='bytes'):
|
|
file_size = os.path.getsize(file_path)
|
|
exponents_map = {'bytes': 0, 'kb': 1, 'mb': 2, 'gb': 3}
|
|
if unit not in exponents_map:
|
|
raise ValueError("Must select from \
|
|
['bytes', 'kb', 'mb', 'gb']")
|
|
else:
|
|
size = file_size / 1024 ** exponents_map[unit]
|
|
return round(size, 1)
|
|
|
|
def get_cvdb(string):
|
|
return re.findall('(?<=\[CVDB)(.*)(?=].)', string[0].text)[0]
|