added checks emergency fixed bugs

app/payment/views.py

@@ -3,7 +3,7 @@ from app.payment import payment
 from app import db
 from app.classes.customer import Customer_Customer
 from app.classes.cards import Card_Card, Card_Card_schema
-from flask_login import current_user
+
 
 
 def set_card_main(user_id, card_id):
@@ -111,6 +111,7 @@ def create_user_card(user_id):
     type_of_card = request.json["type_of_card"]
     security_number = request.json["security_number"]
     main_card = request.json["main_card"]
+    zip_code = request.json["main_card"]
 
     card_number = request.json["card_number"]
     last_four = card_number[-4]
@@ -126,10 +127,15 @@ def create_user_card(user_id):
         security_number=security_number,
         accepted_or_declined=None,
         main_card=main_card,
+        zip_code=zip_code
     )
+    db.session.add(create_new_card)
     db.session.flush()
     if main_card is True:
-        set_card_main(user_id=get_customer.id, card_id=create_new_card.id)
+        try:
+            set_card_main(user_id=get_customer.id, card_id=create_new_card.id)
+        except:
+            pass
     db.session.add(create_new_card)
     db.session.commit()
 
@@ -186,6 +192,7 @@ def update_user_card(card_id):
     security_number = request.json["security_number"]
     card_number = request.json["card_number"]
     main_card = request.json["main_card"]
+    zip_code = request.json["zip_code"]
 
     get_card.user_id = get_customer.id
     get_card.card_number = card_number
@@ -195,6 +202,7 @@ def update_user_card(card_id):
     get_card.type_of_card = type_of_card
     get_card.security_number = security_number
     get_card.main_card = main_card
+    get_card.zip_code = zip_code
 
     if main_card is True:
         set_card_main(user_id=get_customer.id, card_id=get_card.id)