Fetch length top-up code attribute from providerdb if available

1 files changed, 11 insertions, 0 deletions

diff --git a/src/ppm/provider.py b/src/ppm/provider.py
index fbbd911..54bc0fe 100644
--- a/src/ppm/provider.py
+++ b/src/ppm/provider.py
@@ -62,6 +62,17 @@ class Provider(object):
         else:
             return False
 
+    def get_top_up_code_length(self):
+        """The length of the topup code"""
+        if self.has_top_up_cmd():
+            return self.top_up_cmds['ussd'][2]
+        else:
+            return 0
+
+    @property
+    def top_up_code_length(self):
+        return self.get_top_up_code_length()
+
     def fetch_balance(self, mm, reply_func=None, error_func=None):
         if self.has_fetch_balance_cmd():
             mm.ussd_initiate (self.fetch_balance_cmds['ussd'],