states.rb in looky-lu-0.0.2

- old
+ new
@@ -1,61 +1,23 @@
 module LookyLu
   class States
 
-    def self.united_states
-      [
-        {name: 'Alabama', abbreviation: 'AL'},
-        {name: 'Alaska', abbreviation: 'AK'},
-        {name: 'Arizona', abbreviation: 'AZ'},
-        {name: 'Arkansas', abbreviation: 'AR'},
-        {name: 'California', abbreviation: 'CA'},
-        {name: 'Colorado', abbreviation: 'CO'},
-        {name: 'Connecticut', abbreviation: 'CT'},
-        {name: 'Delaware', abbreviation: 'DE'},
-        {name: 'District of Columbia', abbreviation: 'DC'},
-        {name: 'Florida', abbreviation: 'FL'},
-        {name: 'Georgia', abbreviation: 'GA'},
-        {name: 'Hawaii', abbreviation: 'HI'},
-        {name: 'Idaho', abbreviation: 'ID'},
-        {name: 'Illinois', abbreviation: 'IL'},
-        {name: 'Indiana', abbreviation: 'IN'},
-        {name: 'Iowa', abbreviation: 'IA'},
-        {name: 'Kansas', abbreviation: 'KS'},
-        {name: 'Kentucky', abbreviation: 'KY'},
-        {name: 'Louisiana', abbreviation: 'LA'},
-        {name: 'Maine', abbreviation: 'ME'},
-        {name: 'Maryland', abbreviation: 'MD'},
-        {name: 'Massachusetts', abbreviation: 'MA'},
-        {name: 'Michigan', abbreviation: 'MI'},
-        {name: 'Minnesota', abbreviation: 'MN'},
-        {name: 'Mississippi', abbreviation: 'MS'},
-        {name: 'Missouri', abbreviation: 'MO'},
-        {name: 'Montana', abbreviation: 'MT'},
-        {name: 'Nebraska', abbreviation: 'NE'},
-        {name: 'Nevada', abbreviation: 'NV'},
-        {name: 'New Hampshire', abbreviation: 'NH'},
-        {name: 'New Jersey', abbreviation: 'NJ'},
-        {name: 'New Mexico', abbreviation: 'NM'},
-        {name: 'New York', abbreviation: 'NY'},
-        {name: 'North Carolina', abbreviation: 'NC'},
-        {name: 'North Dakota', abbreviation: 'ND'},
-        {name: 'Ohio', abbreviation: 'OH'},
-        {name: 'Oklahoma', abbreviation: 'OK'},
-        {name: 'Oregon', abbreviation: 'OR'},
-        {name: 'Pennsylvania', abbreviation: 'PA'},
-        {name: 'Rhode Island', abbreviation: 'RI'},
-        {name: 'South Carolina', abbreviation: 'SC'},
-        {name: 'South Dakota', abbreviation: 'SD'},
-        {name: 'Tennessee', abbreviation: 'TN'},
-        {name: 'Texas', abbreviation: 'TX'},
-        {name: 'Utah', abbreviation: 'UT'},
-        {name: 'Vermont', abbreviation: 'VT'},
-        {name: 'Virginia', abbreviation: 'VA'},
-        {name: 'Washington', abbreviation: 'WA'},
-        {name: 'West Virginia', abbreviation: 'WV'},
-        {name: 'Wisconsin', abbreviation: 'WI'},
-        {name: 'Wyoming', abbreviation: 'WY'}
-      ]
+    def self.from_country(country_name)
+
+      yml = YAML.load_file(File.dirname(__FILE__) + "/states.yml")
+
+      if country_name == 'all'
+        states = []
+
+        yml.each do |k,v|
+          states += v
+        end
+
+      else
+        states = yml[country_name]
+      end
+
+      states
     end
 
   end
 end
\ No newline at end of file