| UP_SQL = """ | |
| ALTER TABLE afe_shards ADD COLUMN hostname VARCHAR(255) NOT NULL; | |
| ALTER TABLE afe_hosts ADD COLUMN shard_id INT NULL; | |
| ALTER TABLE afe_hosts ADD CONSTRAINT hosts_to_shard_ibfk | |
| FOREIGN KEY (shard_id) REFERENCES afe_shards(id); | |
| """ | |
| DOWN_SQL = """ | |
| ALTER TABLE afe_hosts DROP FOREIGN KEY hosts_to_shard_ibfk; | |
| ALTER TABLE afe_hosts DROP COLUMN shard_id; | |
| ALTER TABLE afe_shards DROP COLUMN hostname; | |
| """ |