diff --git a/scripts/fnordload b/scripts/fnordload index 31d1d74..8b1dda0 100755 --- a/scripts/fnordload +++ b/scripts/fnordload @@ -8,7 +8,8 @@ import logging import logging.config import logging.handlers import max7301 -import upay +import upay.common +import upay.client import sys from decimal import Decimal @@ -245,7 +246,7 @@ class Fnordload(object): time.sleep(2) def _upay_show_balance(self): - token_reader = upay.USBTokenReader() + token_reader = upay.common.USBTokenReader() t0 = time.time() self._lcd.write("Fnordload UPAY", "", "Please insert your", "USB drive") @@ -257,7 +258,7 @@ class Fnordload(object): found_medium_callback = lambda x: self._lcd.write("Fnordload UPAY", "", "Reading USB drive", "")) break - except upay.NoTokensAvailableError: + except upay.common.NoTokensAvailableError: time.sleep(1) if not tokens: @@ -266,7 +267,7 @@ class Fnordload(object): return self._lcd.write("Fnordload UPAY", "", "Validating tokens", "") - token_client = upay.TokenClient() + token_client = upay.client.TokenClient() tokens = token_client.validate_tokens(tokens) value = sum([token.value for token in tokens]) @@ -276,8 +277,8 @@ class Fnordload(object): def _upay_get_tokens(self): self._logger.info("Entered get tokens menu") self._lcd.write("Fnordload UPAY", "", "Reading internal", "purse...") - local_tokens = upay.read_tokens_from_file('local-purse', max_tokens = 100, max_size = 300 * 1024) - token_client = upay.TokenClient() + local_tokens = upay.common.read_tokens_from_file('local-purse', max_tokens = 100, max_size = 300 * 1024) + token_client = upay.client.TokenClient() self._lcd.write("Fnordload UPAY", "", "Validating internal", "purse...") local_tokens = token_client.validate_tokens(local_tokens) @@ -299,7 +300,7 @@ class Fnordload(object): self._lcd.write("Fnordload UPAY", "", "Reading USB drive", "") local.purse_path = path - token_reader = upay.USBTokenReader() + token_reader = upay.common.USBTokenReader() t0 = time.time() self._lcd.write("Fnordload UPAY", "", "Please insert your", "USB drive") @@ -307,7 +308,7 @@ class Fnordload(object): while time.time() < t0 + 30: try: tokens = token_reader.read_tokens(found_medium_callback = found_usb_drive) - except upay.NoTokensAvailableError: + except upay.common.NoTokensAvailableError: time.sleep(1) if local.purse_path: