ari_fetch 0.0.4 → 0.0.5

data/lib/ari_fetch/ari_read_file.rb

@@ -2,7 +2,7 @@ require "nokogiri"
 
 module AriFetch
   class AriReadFile < ActiveRecord::Base
-    attr_accessible :name
+    attr_accessible :name, :data_read
 
     attr_accessor :content, :vehicles
 
@@ -11,7 +11,7 @@ module AriFetch
       @content = parse!(File.open(name, "r"))
       FileUtils.rm name
       filter_cancel_vehicles! unless include_cancel
-      result
+      result.tap { data_reading_done! }
     end
 
     private
@@ -36,5 +36,9 @@ module AriFetch
       vehicles.map { |vehicle| vehicle.result }
     end
 
+    def data_reading_done!
+      self.update_attributes(data_read: true)
+    end
+
   end
 end

data/lib/ari_fetch/files.rb

@@ -8,7 +8,7 @@ module AriFetch
     end
 
     def filter!
-      map_to_file_name.select{ |e| !useless_files.include?(e) && e.match(/\.xml\Z/) }
+      @files = map_to_file_name.select{ |e| !useless_files.include?(e) && e.match(/\.xml\Z/) }
       self
     end
 

data/lib/ari_fetch/version.rb

@@ -1,3 +1,3 @@
 module AriFetch
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

data/lib/ari_fetch.rb

@@ -28,7 +28,7 @@ module AriFetch
     end
 
     def fetch_vehicles_from_files(how_many=1)
-      unread_files.first(how_many).map {|file| AriFetch::AriReadFile.create(name: file).fetch_data(ftp_instance, include_cancel?) }.flatten!
+      unread_files.first(how_many).map {|file| AriFetch::AriReadFile.create(name: file, data_read: false).fetch_data(ftp_instance, include_cancel?) }.flatten!
     end
 
     private

data/lib/generators/ari_fetch/templates/migration.rb

@@ -2,6 +2,7 @@ class CreateAriReadFiles < ActiveRecord::Migration
   def self.up
     create_table :ari_read_files do |t|
       t.string :name
+      t.boolean :data_read
 
       t.timestamps
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ari_fetch
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors: