Change the county constructor back and add overload it for districts.

nysenate · Sep 25, 2012 · 95f8c47 · 95f8c47
1 parent 0b1bed9
commit 95f8c47
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/main/java/gov/nysenate/sage/api/methods/BluebirdMethod.java b/src/main/java/gov/nysenate/sage/api/methods/BluebirdMethod.java
@@ -94,7 +94,7 @@ else if (!result.status_code.equals("0"))
         }
         dr.setAssembly(new Assembly(address.assembly_code+""));
         dr.setCongressional(new Congressional(address.congressional_code+""));
-        dr.setCounty(new County(address.county_code+""));
+        dr.setCounty(new County(null,address.county_code+""));
         dr.setElection(new Election(address.election_code+""));
         dr.setLat(address.latitude);
         dr.setLon(address.longitude);

diff --git a/src/main/java/gov/nysenate/sage/model/districts/County.java b/src/main/java/gov/nysenate/sage/model/districts/County.java
@@ -11,7 +11,12 @@ public County() {
 
     }
 
-    public County(String district) {
+    public County(String countyName) {
+        this.countyName = countyName;
+    }
+
+    public County(String countyName, String district) {
+        this.countyName = countyName;
         this.district = district;
     }