Adds migrations management

migrations/20171111135341_v2-to-v3.py

@@ -0,0 +1,99 @@
+"""
+a caribou migration
+
+name: v2-to-v3
+version: 20171111135341
+"""
+
+def upgrade(connection):
+    # Rename old table
+    sql = """
+        ALTER TABLE trades RENAME TO trades_orig;
+        """
+    connection.execute(sql)
+
+    # Create new table
+    sql = """
+        CREATE TABLE trades (
+            id INTEGER NOT NULL,
+            exchange VARCHAR NOT NULL,
+            pair VARCHAR NOT NULL,
+            fee FLOAT NOT NULL DEFAULT 0.0,
+            is_open BOOLEAN NOT NULL,
+            open_rate FLOAT,
+            close_rate FLOAT,
+            close_profit FLOAT,
+            stake_amount FLOAT NOT NULL,
+            amount FLOAT,
+            open_date DATETIME NOT NULL,
+            close_date DATETIME,
+            open_order_id VARCHAR,
+            PRIMARY KEY (id),
+            CHECK (is_open IN (0, 1))
+        );
+        """
+    connection.execute(sql)
+
+    # Copy data from old table to the new one
+    sql = """
+        INSERT INTO trades(id, exchange, pair, is_open, open_rate, close_rate,
+        close_profit, stake_amount, amount, open_date, close_date, open_order_id)
+        SELECT id, exchange, pair, is_open, open_rate, close_rate, close_profit,
+        btc_amount, amount, open_date, close_date, open_order_id
+        FROM trades_orig;
+        """
+    connection.execute(sql)
+
+    # Remove old table
+    sql = """
+        DROP TABLE trades_orig;
+        """
+    connection.execute(sql)
+
+    connection.commit()
+
+
+def downgrade(connection):
+    sql = """
+        ALTER TABLE trades RENAME TO trades_orig;
+        """
+    connection.execute(sql)
+
+    # Create new table
+    sql = """
+        CREATE TABLE trades (
+            id INTEGER NOT NULL,
+            exchange VARCHAR NOT NULL,
+            pair VARCHAR NOT NULL,
+            is_open BOOLEAN NOT NULL,
+            open_rate FLOAT NOT NULL,
+            close_rate FLOAT,
+            close_profit FLOAT,
+            btc_amount FLOAT NOT NULL,
+            amount FLOAT NOT NULL,
+            open_date DATETIME NOT NULL,
+            close_date DATETIME,
+            open_order_id VARCHAR,
+            PRIMARY KEY (id),
+            CHECK (is_open IN (0, 1))
+        );
+        """
+    connection.execute(sql)
+
+    # Copy data from old table to the new one
+    sql = """
+        INSERT INTO trades(id, exchange, pair, is_open, open_rate, close_rate,
+        close_profit, btc_amount, amount, open_date, close_date, open_order_id)
+        SELECT id, exchange, pair, is_open, open_rate, close_rate, close_profit,
+        stake_amount, amount, open_date, close_date, open_order_id
+        FROM trades_orig;
+        """
+    connection.execute(sql)
+
+    # Remove old table
+    sql = """
+        DROP TABLE trades_orig;
+        """
+    connection.execute(sql)
+
+    connection.commit()