muendelezaji · May 19, 2023 18:22 · May 19, 2023 · May 19, 2023 · May 19, 2023 · May 19, 2023
diff --git a/payment-reference.py b/payment-reference.py
@@ -32,6 +32,7 @@ def get_email_from_payment_ref(pay_ref):
     """
     # Split the payment reference string into groups by "-"
     groups = pay_ref.split("-")
+
     # Rejoin all groups after the 5th "-" (i.e. where email starts)
     # This ensures any emails with "-" will be extracted correctly
     email = "-".join(groups[5:])

diff --git a/payment-reference.py b/payment-reference.py
@@ -6,7 +6,8 @@ def generate_payment_ref(email, method, country):
 
     Examples of returned references:
     - [email protected]
-    - [email protected]
+    - [email protected]
+    - [email protected]
     - [email protected]
 
     :param email: Customer's email

diff --git a/payment-reference.py b/payment-reference.py
@@ -1,8 +1,8 @@
 from datetime import datetime
 
-def create_payment_ref(email, method, country):
+def generate_payment_ref(email, method, country):
     """
-    Creates a unique payment reference that will be sent to the payment provider.
+    Generates a unique payment reference that will be sent to the payment provider.
 
     Examples of returned references:
     - [email protected]

diff --git a/payment-reference.py b/payment-reference.py
@@ -10,7 +10,7 @@ def create_payment_ref(email, method, country):
     - [email protected]
 
     :param email: Customer's email
-    :param method: Payment method - CARD, MOBILE, etc.
+    :param method: Payment method - CARD, PUSH, BANK, USSD etc.
     :param country: Country of service - TZ, KE etc.
     :return: Unique payment reference to be sent to the provider
     """

diff --git a/payment-reference.py b/payment-reference.py
@@ -0,0 +1,37 @@
+from datetime import datetime
+
+def create_payment_ref(email, method, country):
+    """
+    Creates a unique payment reference that will be sent to the payment provider.
+
+    Examples of returned references:
+    - [email protected]
+    - [email protected]
+    - [email protected]
+
+    :param email: Customer's email
+    :param method: Payment method - CARD, MOBILE, etc.
+    :param country: Country of service - TZ, KE etc.
+    :return: Unique payment reference to be sent to the provider
+    """
+    # Generate date and time strings in the right format
+    date = datetime.now().strftime("%Y%m%d")
+    time = datetime.now().strftime("%H%M%S")
+
+    # Combine it with the customer's email, payment method and country
+    ref = f"TENDA-{country}-{method}-{date}-{time}-{email}"
+    return ref
+
+def get_email_from_payment_ref(pay_ref):
+    """
+    Extracts the customer's email from payment reference sent to the provider.
+
+    :param pay_ref: Payment reference
+    :return: Email of the customer that attempted payment
+    """
+    # Split the payment reference string into groups by "-"
+    groups = pay_ref.split("-")
+    # Rejoin all groups after the 5th "-" (i.e. where email starts)
+    # This ensures any emails with "-" will be extracted correctly
+    email = "-".join(groups[5:])
+    return email