From fcdbf2f55c1d47881ba34f3ee15fcb0e24329763 Mon Sep 17 00:00:00 2001 From: Kyle Drake Date: Wed, 10 Aug 2022 18:55:37 +0000 Subject: [PATCH] Fix hyphen on normal screenshot urls, remove dupe code --- Rakefile | 17 +---------------- 1 file changed, 1 insertion(+), 16 deletions(-) diff --git a/Rakefile b/Rakefile index 9717e5e5..718793f6 100644 --- a/Rakefile +++ b/Rakefile @@ -477,7 +477,7 @@ end desc 'ml_screenshots_list_dump' task :ml_screenshots_list_dump => [:environment] do ['phishing', 'spam', 'ham', nil].each do |classifier| - File.open("./files/screenshot-urls-#{classifier.to_s}.txt", 'w') do |fp| + File.open("./files/screenshot-urls#{classifier.nil? ? '' : '-'+classifier.to_s}.txt", 'w') do |fp| SiteFile.where(classifier: classifier).where(path: 'index.html').each do |site_file| begin fp.write "#{site_file.site.screenshot_url('index.html', Site::SCREENSHOT_RESOLUTIONS.first)}\n" @@ -589,19 +589,4 @@ task :generate_sitemap => [:environment] do end gz.write %{} end - - desc 'ml_screenshots_list_dump' - task :ml_screenshots_list_dump => [:environment] do - ['phishing', 'spam', 'ham', nil].each do |classifier| - File.open("./files/screenshot-urls-#{classifier.to_s}.txt", 'w') do |fp| - SiteFile.where(classifier: classifier).where(path: 'index.html').each do |site_file| - begin - fp.write "#{site_file.site.screenshot_url('index.html', Site::SCREENSHOT_RESOLUTIONS.first)}\n" - rescue NoMethodError - end - end - end - end - end - end