From 708b723f8669ca711269012ac12b9f4e06b89c81 Mon Sep 17 00:00:00 2001 From: Blair Vanderlugt Date: Tue, 27 Jul 2021 09:48:03 -0700 Subject: [PATCH] if user dir not ready just retry instead of bailing --- apps/data-seeding/cic_ussd/import_balance.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/apps/data-seeding/cic_ussd/import_balance.py b/apps/data-seeding/cic_ussd/import_balance.py index df5bdfe4..513c2ef6 100644 --- a/apps/data-seeding/cic_ussd/import_balance.py +++ b/apps/data-seeding/cic_ussd/import_balance.py @@ -2,6 +2,7 @@ import argparse import logging import sys +import time import os # external imports @@ -124,7 +125,16 @@ def main(): # TODO get decimals from token balances = {} - f = open('{}/balances.csv'.format(user_dir, 'r')) + f = None + while f is None: + try: + f = open('{}/balances.csv'.format(user_dir, 'r')) + except FileNotFoundError: + print('{} not found...', user_dir) + time.sleep(30) + + + remove_zeros = 10 ** 6 i = 0 while True: