aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJoshua Peek <josh@joshpeek.com>2012-07-23 11:21:55 -0500
committerJoshua Peek <josh@joshpeek.com>2012-07-23 11:21:55 -0500
commitdb88e143ba96f21502942ad7f39c19307b9a6e02 (patch)
tree470eec7d6d9216b8dea8ba7510448072ef438036
parent817e3cb946bb8f69af94e5117e98c4a4c18f96c0 (diff)
Dump classifier as plain hash
-rw-r--r--lib/linguist/classifier.rb17
-rw-r--r--lib/linguist/classifier.yml1
2 files changed, 8 insertions, 10 deletions
diff --git a/lib/linguist/classifier.rb b/lib/linguist/classifier.rb
index 3c518c2..7a348aa 100644
--- a/lib/linguist/classifier.rb
+++ b/lib/linguist/classifier.rb
@@ -17,16 +17,16 @@ module Linguist
#
# Returns Classifier.
def self.instance
- @instance ||= YAML.load_file(PATH)
+ @instance ||= new(YAML.load_file(PATH))
end
# Public: Initialize a Classifier.
- def initialize
- @tokens_total = 0
- @languages_total = 0
- @tokens = {}
- @language_tokens = {}
- @languages = {}
+ def initialize(attrs = {})
+ @tokens_total = attrs['tokens_total'] || 0
+ @languages_total = attrs['languages_total'] || 0
+ @tokens = attrs['tokens'] || {}
+ @language_tokens = attrs['language_tokens'] || {}
+ @languages = attrs['languages'] || {}
end
# Public: Compare Classifier objects.
@@ -152,8 +152,7 @@ module Linguist
#
# Returns nothing.
def to_yaml(io)
- data = "--- !ruby/object:Linguist::Classifier\n"
-
+ data = ""
escape = lambda { |s| s.inspect.gsub(/\\#/, "\#") }
data << "languages_total: #{@languages_total}\n"
diff --git a/lib/linguist/classifier.yml b/lib/linguist/classifier.yml
index 5874b62..23b4447 100644
--- a/lib/linguist/classifier.yml
+++ b/lib/linguist/classifier.yml
@@ -1,4 +1,3 @@
---- !ruby/object:Linguist::Classifier
languages_total: 243
tokens_total: 164127
languages: