Exclude non-residential communities

1 files changed, 9 insertions, 4 deletions

diff --git a/bin/community b/bin/community
index a95b2c2..dafa075 100755
--- a/bin/community
+++ b/bin/community
@@ -185,6 +185,8 @@ class Community
 end
 
 class Census
+  include Enumerable
+
   def initialize(path)
     @path = path
   end
@@ -196,8 +198,11 @@ class Census
   end
 end
 
-Census.new("data/Census_by_Community_2019_20240524.csv").each do |community|
-  community.create_directory_in("yyc")
-  community.create_readme
-end
+Census
+  .new("data/Census_by_Community_2019_20240524.csv")
+  .find_all { |x| x[:class].include?("Residential") }
+  .each do |community|
+    community.create_directory_in("yyc")
+    community.create_readme
+  end