From 717597f7583a143eeb3d7da48b35118839537e1d Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Jan 05 2017 15:24:18 +0000
Subject: Add the alembic migration allowing to migrate tags to the new model


---

diff --git a/alembic/versions/208b0cd232ab_colored_tags.py b/alembic/versions/208b0cd232ab_colored_tags.py
new file mode 100644
index 0000000..de1e3dd
--- /dev/null
+++ b/alembic/versions/208b0cd232ab_colored_tags.py
@@ -0,0 +1,69 @@
+"""colored tags
+
+
+Revision ID: 208b0cd232ab
+Revises: 588eabcd394c
+Create Date: 2016-12-27 11:52:53.355838
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '208b0cd232ab'
+down_revision = '588eabcd394c'
+
+import collections
+import datetime
+
+from alembic import op
+import sqlalchemy as sa
+
+
+try:
+    from pagure.lib import model
+except ImportError:
+    import sys
+    sys.path.insert(0, '.')
+    from pagure.lib import model
+
+
+def upgrade():
+    """ Alter the DB schema for the changes related to colored tags. """
+    engine = op.get_bind()
+    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
+    Session.configure(bind=engine)
+    session = Session()
+
+    projects = collections.defaultdict(list)
+    for issue in session.query(model.Issue).all():
+        for issuetag in issue.old_tags:
+            tag = issuetag.tag
+            # Add the tag to the project if it isn't already there
+            if tag not in projects[issue.project.id]:
+                tagobj = model.TagColored(
+                    tag=tag,
+                    project_id=issue.project.id)
+                session.add(tagobj)
+                session.flush()
+                projects[issue.project.id].append(tag)
+            else:
+                tagobj = session.query(
+                    model.TagColored
+                ).filter(
+                    model.TagColored.tag == tag
+                ).filter(
+                    model.TagColored.project_id == issue.project.id
+                ).first()
+
+            # Link the tag to the ticket as it was
+            tagissueobj = model.TagIssueColored(
+                tag_id=tagobj.id,
+                issue_uid=issue.uid,
+                date_created=tagobj.date_created,
+            )
+            session.add(tagissueobj)
+            session.flush()
+    session.commit()
+
+
+def downgrade():
+    raise ValueError("The colored tags feature can not be un-done")