From f81643aca57ecad2cceb659dbca33715a8626d0d Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Sun, 21 Nov 2021 23:13:29 -0500 Subject: [PATCH] [F] mkdirs --- src/raw_collect/twitter.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/raw_collect/twitter.py b/src/raw_collect/twitter.py index ec8bf8b..4a2e8f1 100644 --- a/src/raw_collect/twitter.py +++ b/src/raw_collect/twitter.py @@ -159,6 +159,10 @@ def download_users(api: API, start_point: str, n: float = math.inf, if base_dir.endswith('/'): base_dir = base_dir[:-1] + # Ensure directory exists + Path(f'{base_dir}/users').mkdir(parents=True, exist_ok=True) + Path(f'{base_dir}/meta').mkdir(parents=True, exist_ok=True) + # Limit request rate start_time = time.time() num_requests = 0