feat: draft version

db_repo/versions/003_clone_duration_column.py

@@ -0,0 +1,26 @@
+from sqlalchemy import Table, MetaData, Column, Float
+
+meta = MetaData()
+
+
+def _get_table(meta):
+    meta.reflect()
+    AnalyzeTask = Table("analyze_tasks", meta, autoload=True, autoload_with=meta.bind)
+    return AnalyzeTask
+
+
+def _get_column():
+    return Column("clone_duration", Float, nullable=False, default=0, server_default="0")
+
+
+def upgrade(migrate_engine):
+    meta.bind = migrate_engine
+    table = _get_table(meta)
+    col = _get_column()
+    col.create(table, populate_default=True)
+
+
+def downgrade(migrate_engine):
+    meta.bind = migrate_engine
+    table = _get_table(meta)
+    table.c.clone_duration.drop()