Merge 19aa5acd62
into 61b9f81e39
This commit is contained in:
commit
92635bc1f7
2 changed files with 22 additions and 15 deletions
|
@ -177,3 +177,9 @@ privileges:
|
|||
## usage: schema://user:password@host:port/database_name
|
||||
## example: postgres://szuru:dog@localhost:5432/szuru_test
|
||||
#database:
|
||||
|
||||
|
||||
## get the data dir usage
|
||||
## false disabled
|
||||
## true enabled ( default )
|
||||
data_dir_get_usage: true
|
||||
|
|
|
@ -10,22 +10,23 @@ _cache_result = None # type: Optional[int]
|
|||
|
||||
|
||||
def _get_disk_usage() -> int:
|
||||
global _cache_time, _cache_result
|
||||
threshold = timedelta(hours=48)
|
||||
now = datetime.utcnow()
|
||||
if _cache_time and _cache_time > now - threshold:
|
||||
assert _cache_result is not None
|
||||
return _cache_result
|
||||
total_size = 0
|
||||
for dir_path, _, file_names in os.walk(config.config["data_dir"]):
|
||||
for file_name in file_names:
|
||||
file_path = os.path.join(dir_path, file_name)
|
||||
try:
|
||||
total_size += os.path.getsize(file_path)
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
_cache_time = now
|
||||
_cache_result = total_size
|
||||
if config.config["data_dir_get_usage"]:
|
||||
global _cache_time, _cache_result
|
||||
threshold = timedelta(hours=48)
|
||||
now = datetime.utcnow()
|
||||
if _cache_time and _cache_time > now - threshold:
|
||||
assert _cache_result is not None
|
||||
return _cache_result
|
||||
for dir_path, _, file_names in os.walk(config.config["data_dir"]):
|
||||
for file_name in file_names:
|
||||
file_path = os.path.join(dir_path, file_name)
|
||||
try:
|
||||
total_size += os.path.getsize(file_path)
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
_cache_time = now
|
||||
_cache_result = total_size
|
||||
return total_size
|
||||
|
||||
|
||||
|
|
Reference in a new issue