From 76a9e2d17c2130829d5772c9c8d55072ab40ab58 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20G=C3=BCtlein?= <martin.guetlein@gmail.com>
Date: Wed, 17 Feb 2010 10:01:06 +0100
Subject: migrate when delete_all, dataset prediction fixed

---
 application.rb | 3 ++-
 lazar.rb       | 9 ++++++---
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/application.rb b/application.rb
index d3787a9..db5b41e 100644
--- a/application.rb
+++ b/application.rb
@@ -46,6 +46,7 @@ end
 
 
 delete '/?' do
-	Model.all.each { |d| d.destroy! }
+	#Model.all.each { |d| d.destroy! }
+  Model.auto_migrate!
 	"All Models deleted."
 end
diff --git a/lazar.rb b/lazar.rb
index 4c10c04..9014d3b 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -3,9 +3,11 @@ class Lazar < Model
 	attr_accessor :dataset, :predictions
 
 	def classify(compound_uri)
-
-		@dataset = OpenTox::Dataset.new
-		@predictions = {}
+  
+    unless @dataset
+		  @dataset = OpenTox::Dataset.new
+		  @predictions = {}
+    end
 		lazar = YAML.load yaml
 		compound = OpenTox::Compound.new(:uri => compound_uri)
 		compound_matches = compound.match lazar[:features]
@@ -50,6 +52,7 @@ class Lazar < Model
 					:features => compound_matches
 				} } }
 		end
+    
 	end
 
 	def database_activity?(compound_uri)
-- 
cgit v1.2.3