From: Tzu-Mainn Chen <[email protected]>

---
 ...25170745_update_credential_definition_labels.rb |   32 ++++++++++++++++++++
 1 files changed, 32 insertions(+), 0 deletions(-)
 create mode 100644 
src/db/migrate/20121025170745_update_credential_definition_labels.rb

diff --git 
a/src/db/migrate/20121025170745_update_credential_definition_labels.rb 
b/src/db/migrate/20121025170745_update_credential_definition_labels.rb
new file mode 100644
index 0000000..a4606d7
--- /dev/null
+++ b/src/db/migrate/20121025170745_update_credential_definition_labels.rb
@@ -0,0 +1,32 @@
+class UpdateCredentialDefinitionLabels < ActiveRecord::Migration
+  def self.up
+    CredentialDefinition.all.each do |cred|
+      if name_mapping.has_key? cred.label
+        cred.label = name_mapping[cred.label]
+        cred.save!
+      end
+    end
+  end
+
+  def self.down
+    reverse_mapping = name_mapping.invert
+    CredentialDefinition.all.each do |cred|
+      if reverse_mapping.has_key? cred.label
+        cred.label = reverse_mapping[cred.label]
+        cred.save!
+      end
+    end
+  end
+
+  def self.name_mapping
+    {
+      "Access Key" => "access_key",
+      "Secret Access Key" => "secret_access_key",
+      "Account Number" => "account_number",
+      "Username" => "username",
+      "Key" => "key",
+      "Certificate" => "certificate",
+      "Password" => "password"
+    }
+  end
+end
-- 
1.7.6.5

Reply via email to