From dc58136c343babb076516f6f4e56170318f7e47b Mon Sep 17 00:00:00 2001
From: Aleksey Lobanov <i@likemath.ru>
Date: Thu, 7 Jan 2021 22:33:00 +0300
Subject: [PATCH] db: Add new index

---
 .../005_second_step_generation_index.py       | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 db_repo/versions/005_second_step_generation_index.py

diff --git a/db_repo/versions/005_second_step_generation_index.py b/db_repo/versions/005_second_step_generation_index.py
new file mode 100644
index 0000000..4f90902
--- /dev/null
+++ b/db_repo/versions/005_second_step_generation_index.py
@@ -0,0 +1,30 @@
+from sqlalchemy import Table, MetaData, Index
+
+meta = MetaData()
+
+
+def _get_indices(meta):
+    meta.reflect()
+    Tasks = Table("analyze_tasks", meta, autoload=True, autoload_with=meta.bind)
+    return [
+        Index(
+            "tasks_second_step_idx",
+            Tasks.c.repository_id,
+            Tasks.c.status,
+            Tasks.c.level,
+            Tasks.c.improvement_absolute,
+            Tasks.c.improvement_relative,
+        )
+    ]
+
+
+def upgrade(migrate_engine):
+    meta.bind = migrate_engine
+    for idx in _get_indices(meta):
+        idx.create()
+
+
+def downgrade(migrate_engine):
+    meta.bind = migrate_engine
+    for idx in _get_indices(meta):
+        idx.drop()