Add support for classification of non-fiction books.

[quanlib.git] / book.rb

diff --git a/book.rb b/book.rb
index e0ccc841eed4c86600d56e18b0373fc14a138ab2..5b698e7a474e2909433c0b6c893dc13af802c193 100644 (file)
--- a/book.rb
+++ b/book.rb
@@ -11,6 +11,7 @@ class Book
 
   def initialize(store)
     @author = nil
+    @classification_id = nil
     @cover = nil
     @description = nil
     @path = nil
@@ -20,7 +21,7 @@ class Book
     @volume = nil
   end
 
-  def load_from_file(fileName)
+  def load_from_file!(fileName)
     @path = fileName
     parse_file_name!(fileName)
   end
@@ -52,6 +53,14 @@ class Book
     @author = value
   end
 
+  def classification_id
+    @classification_id
+  end
+
+  def classification_id=(value)
+    @classification_id = value
+  end
+
   def cover
     return @cover
   end
@@ -146,6 +155,14 @@ class Book
     @title = value
   end
 
+  def title_grouping
+    if nil == @path
+      return nil
+    end
+
+    return File.basename(@path, '.*')
+  end
+
   def volume
     @volume
   end
@@ -225,6 +242,8 @@ class Book
     elsif lc_file_name.end_with?(".pdf")
       scan_pdf!(file_name)
     end
+
+    @classification_id = @store.find_classification(@author.grouping, File.basename(file_name, '.*'))
   end
 
   protected