Added automatic stuff

app/payment/views.py

@@ -6,24 +6,46 @@ from app.classes.cards import Card_Card, Card_Card_schema
 from flask_login import current_user
 
 
-def set_card_main(user_id):
+def set_card_main(user_id, card_id):
     """
     updates a card of a user
     """
-    get_card_count = db.session \
-        .query(Card_Card) \
-        .filter(Card_Card.user_id == user_id) \
+    get_card_count = (
+         db.session 
+        .query(Card_Card) 
+        .filter(Card_Card.user_id == user_id) 
         .count()
+        )
+    
+    get_card = (
+         db.session 
+        .query(Card_Card) 
+        .filter(Card_Card.user_id == user_id) 
+        .filter(Card_Card.id == card_id) 
+        .first()
+        )
+
     if get_card_count > 0:
-        get_old_card = db.session \
-            .query(Card_Card) \
-            .filter(Card_Card.main_card == True) \
-            .filter(Card_Card.user_id == user_id) \
+        get_old_card = (
+             db.session 
+            .query(Card_Card) 
+            .filter(Card_Card.main_card == True) 
+            .filter(Card_Card.user_id == user_id) 
             .first()
+        )
+
         get_old_card.main_card = False
+        get_card.main_card = True
 
         db.session.add(get_old_card)
         db.session.commit()
+    else:
+
+
+        get_card.main_card = True
+
+        db.session.add(get_card)
+        db.session.commit()
 
 
 @payment.route("/cards/<int:user_id>", methods=["GET"])
@@ -105,9 +127,9 @@ def create_user_card(user_id):
         accepted_or_declined=None,
         main_card=main_card,
     )
-    print(main_card)
+    db.session.flush()
     if main_card is True:
-        set_card_main(user_id=get_customer.id)
+        set_card_main(user_id=get_customer.id, card_id=create_new_card.id)
     db.session.add(create_new_card)
     db.session.commit()