Update tasks

lib/tasks/export_character.rake

@@ -22,20 +22,22 @@ namespace :granblue do
 
       # Set up filepath
       dir = "#{Rails.root}/export/"
+      filename = "#{dir}/character-#{size}.txt"
       FileUtils.mkdir(dir) unless Dir.exist?(dir)
 
-      unless File.exist?("#{dir}/character-#{size}.txt")
+      # Write to file
-        File.open("#{dir}/character-#{size}.txt", 'w') do |f|
+      File.open(filename, 'w+') do |f|
-          Character.all.each do |c|
+        Character.all.each do |c|
-            f.write("#{build_chara_url("#{c.granblue_id}_01", size)} \n")
+          f.write("#{build_chara_url("#{c.granblue_id}_01", size)} \n")
-            f.write("#{build_chara_url("#{c.granblue_id}_02", size)} \n")
+          f.write("#{build_chara_url("#{c.granblue_id}_02", size)} \n")
-            f.write("#{build_chara_url("#{c.granblue_id}_03", size)} \n") if c.flb
+          f.write("#{build_chara_url("#{c.granblue_id}_03", size)} \n") if c.flb
-            f.write("#{build_chara_url("#{c.granblue_id}_04", size)} \n") if c.ulb
+          f.write("#{build_chara_url("#{c.granblue_id}_04", size)} \n") if c.ulb
-          end
         end
       end
 
-      puts "Wrote #{Character.count} character URLs for \"#{size}\" size"
+      # CLI Output
+      count = `wc -l #{filename}`.split.first.to_i
+      puts "Wrote #{count} character URLs for \"#{size}\" size"
     end
   end
 end

lib/tasks/export_summon.rake

@@ -22,18 +22,26 @@ namespace :granblue do
 
       # Set up filepath
       dir = "#{Rails.root}/export/"
+      filename = "#{dir}/summon-#{size}.txt"
       FileUtils.mkdir(dir) unless Dir.exist?(dir)
 
-      unless File.exist?("#{dir}/summon-#{size}.txt")
+      # Write to file
-        File.open("#{dir}/summon-#{size}.txt", 'w') do |f|
+      File.open(filename, 'w') do |f|
-          Summon.all.each do |s|
+        Summon.all.each do |s|
-            f.write("#{build_summon_url("#{s.granblue_id}_01", size)} \n")
+          series = s.series.to_i
-            f.write("#{build_summon_url("#{s.granblue_id}_02", size)} \n") if (s.series == 3 || s.series == 0) && s.ulb
+          f.write("#{build_summon_url(s.granblue_id.to_s, size)} \n")
+
+          # Download second images only for Providence ULBs and Primal summons
+          if series == 3 || (series == 0 && s.ulb)
+            f.write("#{build_summon_url("#{s.granblue_id}_02",
+                                        size)} \n")
           end
         end
       end
 
-      puts "Wrote #{Summon.count} summon URLs for \"#{size}\" size"
+      # CLI output
+      count = `wc -l #{filename}`.split.first.to_i
+      puts "Wrote #{count} summon URLs for \"#{size}\" size"
     end
   end
 end

lib/tasks/export_weapon.rake

@@ -22,17 +22,19 @@ namespace :granblue do
 
       # Set up filepath
       dir = "#{Rails.root}/export/"
+      filename = "#{dir}/weapon-#{size}.txt"
       FileUtils.mkdir(dir) unless Dir.exist?(dir)
 
-      unless File.exist?("#{dir}/weapon-#{size}.txt")
+      # Write to file
-        File.open("#{dir}/weapon-#{size}.txt", 'w') do |f|
+      File.open(filename, 'w') do |f|
-          Weapon.all.each do |w|
+        Weapon.all.each do |w|
-            f.write("#{build_weapon_url(w.granblue_id.to_s, size)} \n")
+          f.write("#{build_weapon_url(w.granblue_id.to_s, size)} \n")
-          end
         end
       end
 
-      puts "Wrote #{Weapon.count} weapon URLs for \"#{size}\" size"
+      # CLI output
+      count = `wc -l #{filename}`.split.first.to_i
+      puts "Wrote #{count} weapon URLs for \"#{size}\" size"
     end
   end
 end