From 40254c54a323b186fde9354a701c353c4e6f5559 Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Sat, 30 Nov 2019 18:22:25 +0000
Subject: [PATCH] remove stripe columns in User model

---
 app/models.py                        |  4 ---
 migrations/versions/0b28518684ae_.py | 39 ++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 4 deletions(-)
 create mode 100644 migrations/versions/0b28518684ae_.py

diff --git a/app/models.py b/app/models.py
index 02d56ad7..e80cb36f 100644
--- a/app/models.py
+++ b/app/models.py
@@ -92,10 +92,6 @@ class User(db.Model, ModelMixin, UserMixin):
 
     activated = db.Column(db.Boolean, default=False, nullable=False)
 
-    stripe_customer_id = db.Column(db.String(128), unique=True)
-    stripe_card_token = db.Column(db.String(128), unique=True)
-    stripe_subscription_id = db.Column(db.String(128), unique=True)
-
     profile_picture_id = db.Column(db.ForeignKey(File.id), nullable=True)
 
     # feature flag
diff --git a/migrations/versions/0b28518684ae_.py b/migrations/versions/0b28518684ae_.py
new file mode 100644
index 00000000..7dced928
--- /dev/null
+++ b/migrations/versions/0b28518684ae_.py
@@ -0,0 +1,39 @@
+"""empty message
+
+Revision ID: 0b28518684ae
+Revises: a8d8aa307b8b
+Create Date: 2019-11-30 18:22:02.869387
+
+"""
+import sqlalchemy_utils
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '0b28518684ae'
+down_revision = 'a8d8aa307b8b'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('users_stripe_card_token_key', 'users', type_='unique')
+    op.drop_constraint('users_stripe_customer_id_key', 'users', type_='unique')
+    op.drop_constraint('users_stripe_subscription_id_key', 'users', type_='unique')
+    op.drop_column('users', 'stripe_customer_id')
+    op.drop_column('users', 'stripe_card_token')
+    op.drop_column('users', 'stripe_subscription_id')
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('users', sa.Column('stripe_subscription_id', sa.VARCHAR(length=128), autoincrement=False, nullable=True))
+    op.add_column('users', sa.Column('stripe_card_token', sa.VARCHAR(length=128), autoincrement=False, nullable=True))
+    op.add_column('users', sa.Column('stripe_customer_id', sa.VARCHAR(length=128), autoincrement=False, nullable=True))
+    op.create_unique_constraint('users_stripe_subscription_id_key', 'users', ['stripe_subscription_id'])
+    op.create_unique_constraint('users_stripe_customer_id_key', 'users', ['stripe_customer_id'])
+    op.create_unique_constraint('users_stripe_card_token_key', 'users', ['stripe_card_token'])
+    # ### end Alembic commands ###