From 7d6f4449a64c8d24b514d37393647c5afc930b08 Mon Sep 17 00:00:00 2001 From: Eryn Wells Date: Sun, 17 May 2015 19:10:27 -0700 Subject: [PATCH] Fix paths to model and archive --- bots.rb | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/bots.rb b/bots.rb index 22d21c5..d25a525 100644 --- a/bots.rb +++ b/bots.rb @@ -32,6 +32,9 @@ class ERWEbooksBot < Ebooks::Bot def top20; @top20 ||= model.keywords.take(20); end def on_startup + @model_path ||= "model/#{original}.model" + @archive_path = "model/#{original}.json" + archive! load_model! @@ -128,17 +131,15 @@ class ERWEbooksBot < Ebooks::Bot private def load_model! - @model_path ||= "model/#{original}.model" - log "Loading model #{model_path}" - @model = Ebooks::Model.load(model_path) + log "Loading model #{@model_path}" + @model = Ebooks::Model.load(@model_path) end private def archive! - archive_path = "model/#{original}.json" - log "Archiving tweets #{archive_path}" - Ebooks::Archive.new(@original, archive_path).sync - Ebooks::Model.consume(archive_path).save(model_path) + log "Archiving tweets #{@archive_path}" + Ebooks::Archive.new(@original, @archive_path).sync + Ebooks::Model.consume(@archive_path).save(@model_path) end end