|
1 | 1 | import logging |
2 | 2 | import os |
3 | 3 | import shutil |
| 4 | +import subprocess |
4 | 5 | import traceback |
5 | 6 | from typing import Optional |
6 | 7 |
|
7 | 8 | import psutil |
8 | 9 | import requests |
9 | | -from jf_agent.config_file_reader import get_ingest_config |
10 | | -from jf_agent.data_manifests.git.generator import get_instance_slug |
11 | | -from jf_agent.git import GithubGqlClient, get_git_client, get_nested_repos_from_git |
12 | | -from jf_agent.util import upload_file |
13 | 10 | from jf_ingest.validation import ( |
14 | 11 | GitConnectionHealthCheckResult, |
15 | 12 | IngestionHealthCheckResult, |
|
20 | 17 | ) |
21 | 18 |
|
22 | 19 | from jf_agent import write_file |
| 20 | +from jf_agent.config_file_reader import get_ingest_config |
| 21 | +from jf_agent.data_manifests.git.generator import get_instance_slug |
| 22 | +from jf_agent.git import GithubGqlClient, get_git_client, get_nested_repos_from_git |
| 23 | +from jf_agent.util import upload_file |
23 | 24 |
|
24 | 25 | logger = logging.getLogger(__name__) |
25 | 26 |
|
@@ -173,7 +174,12 @@ def validate_memory(config): |
173 | 174 | f" Available memory: {round(psutil.virtual_memory().available / (1024 * 1024), 2)} MB" |
174 | 175 | ) |
175 | 176 |
|
176 | | - output_dir_size = os.popen(f'du -hs {config.outdir}').readlines()[0].split("\t")[0] |
| 177 | + proc = subprocess.run( |
| 178 | + ['du', '-hs', os.path.expanduser(config.outdir)], |
| 179 | + encoding='utf-8', |
| 180 | + stdout=subprocess.PIPE, |
| 181 | + ) |
| 182 | + output_dir_size = proc.stdout.split("\t")[0] |
177 | 183 | usage = shutil.disk_usage(config.outdir) |
178 | 184 |
|
179 | 185 | print( |
|
0 commit comments