From ec3350e45f411a3feb3f8284b7a737bb91d4092f Mon Sep 17 00:00:00 2001
From: Danny Skubak <skubakdj@gmail.com>
Date: Wed, 13 Nov 2019 18:59:35 -0500
Subject: [PATCH] Proposal Migration Script (#47)

* init script

* only modify object in not dry runs
---
 backend/grant/app.py               |  1 +
 backend/grant/proposal/commands.py | 46 ++++++++++++++++++++++++++++--
 2 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/backend/grant/app.py b/backend/grant/app.py
index 9a0a7673..f401c208 100644
--- a/backend/grant/app.py
+++ b/backend/grant/app.py
@@ -163,5 +163,6 @@ def register_commands(app):
     app.cli.add_command(commands.reset_db_chain_data)
     app.cli.add_command(proposal.commands.create_proposal)
     app.cli.add_command(proposal.commands.create_proposals)
+    app.cli.add_command(proposal.commands.retire_v1_proposals)
     app.cli.add_command(user.commands.set_admin)
     app.cli.add_command(task.commands.create_task)
diff --git a/backend/grant/proposal/commands.py b/backend/grant/proposal/commands.py
index e713ef99..3443a787 100644
--- a/backend/grant/proposal/commands.py
+++ b/backend/grant/proposal/commands.py
@@ -7,7 +7,7 @@ from flask.cli import with_appcontext
 from .models import Proposal, db
 from grant.milestone.models import Milestone
 from grant.comment.models import Comment
-from grant.utils.enums import ProposalStatus, Category, ProposalStageEnum
+from grant.utils.enums import ProposalStatus, Category, ProposalStage
 from grant.user.models import User
 
 
@@ -35,9 +35,9 @@ def create_proposals(count):
     user = User.query.filter_by().first()
     for i in range(count):
         if i < 5:
-            stage = ProposalStageEnum.WIP
+            stage = ProposalStage.WIP
         else:
-            stage = ProposalStageEnum.COMPLETED
+            stage = ProposalStage.COMPLETED
         p = Proposal.create(
             stage=stage,
             status=ProposalStatus.LIVE,
@@ -76,3 +76,43 @@ def create_proposals(count):
 
     db.session.commit()
     print(f'Added {count} LIVE fake proposals')
+
+
+@click.command()
+@click.argument('dry', required=False)
+@with_appcontext
+def retire_v1_proposals(dry):
+    now = datetime.datetime.now()
+    proposals = Proposal.query.filter_by(stage="FUNDING_REQUIRED").all()
+    modified_count = 0
+
+    if not proposals:
+        print("No proposals found. Exiting...")
+        return
+
+    print(f"Found {len(proposals)} proposals to modify")
+    if dry:
+        print(f"This is a dry run. Changes will not be committed to database")
+
+    confirm = input("Continue? (y/n) ")
+
+    if confirm != "y":
+        print("Exiting...")
+        return
+
+    for p in proposals:
+        if not dry:
+            new_deadline = (now - p.date_published).total_seconds()
+            p.stage = ProposalStage.FAILED
+            p.deadline_duration = int(new_deadline)
+            db.session.add(p)
+            modified_count += 1
+
+        print(f"Modified proposal {p.id} - {p.title}")
+
+    if not dry:
+        print(f"Committing changes to database")
+        db.session.commit()
+
+    print(f"Modified {modified_count} proposals")
+