Restructure package. Add flask adapter

PolymathVentures · Apr 22, 2016 · fd73a0b · fd73a0b
1 parent 45ab068
commit fd73a0b
Show file tree

Hide file tree

Showing 6 changed files with 24 additions and 0 deletions.
diff --git a/payu/__init__.py b/payu/__init__.py
@@ -0,0 +1 @@
+from payu.payu import PayU
diff --git a/payu/adapters/__init__.py b/payu/adapters/__init__.py
diff --git a/payu/adapters/flask_payu.py b/payu/adapters/flask_payu.py
@@ -0,0 +1,17 @@
+from payu import PayU as PayUBase
+
+class PayU(PayUBase):
+
+    def __init__(self, app, **config):
+        super(PayU, self).__init__(**config)
+        self.app = app
+        if app is not None:
+            self.init_app(app)
+
+
+    def init_app(self, app):
+        for key, value in self.app.config.items():
+            if not key.startswith('PAYU_'):
+                continue
+            k = key.split('PAYU_')[-1]
+            self.config[k] = value
diff --git a/helpers.py → payu/helpers.py b/helpers.py → payu/helpers.py
@@ -6,3 +6,9 @@ def generate_reference_code(id=None, prefix='payment_test_'):
         s = string.ascii_lowercase + string.digits
         id = ''.join(random.sample(s,10))
     return prefix + id
+
+
+def mask_credit_card_number(number):
+    """ Mask credit card, keeping the first 6 (Issuer Identification Number)
+    and last 4 digits visible """
+    return number[:6] + 'X' * 6 + number[-4:]
diff --git a/payu.py → payu/payu.py b/payu.py → payu/payu.py
diff --git a/test_payu.py → payu/test_payu.py b/test_payu.py → payu/test_payu.py