Added Card PaymentMethod. (mmohades#34)

bellyfat · Mar 23, 2021 · 1eb72c9 · 1eb72c9
1 parent b726ad1
commit 1eb72c9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/venmo_api/models/payment_method.py b/venmo_api/models/payment_method.py
@@ -54,6 +54,9 @@ class BankAccount(PaymentMethod, BaseModel):
     def __init__(self, pid, p_role, p_name, p_type, json=None):
         super().__init__(pid, p_role, p_name, p_type, json)
 
+class Card(PaymentMethod, BaseModel):
+    def __init__(self, pid, p_role, p_name, p_type, json=None):
+        super().__init__(pid, p_role, p_name, p_type, json)
 
 class PaymentRole(Enum):
     DEFAULT = 'default'
@@ -67,4 +70,4 @@ class PaymentPrivacy(Enum):
     FRIENDS = 'friends'
 
 
-payment_type = {'bank': BankAccount, 'balance': VenmoBalance}
+payment_type = {'bank': BankAccount, 'balance': VenmoBalance, 'card': Card}