Add convenient getindex accessor for lemmas

JuliaText · Sep 22, 2015 · c64c0c8 · c64c0c8
1 parent 9489fe2
commit c64c0c8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/db.jl b/src/db.jl
@@ -11,6 +11,12 @@ end
 
 Base.show(io::IO, db::DB) = print(io, "Wordnet.DB")
 
+function Base.getindex(db::DB, pos::Char, word::String) 
+    db.lemmas[pos][lowercase(word)]
+end
+
+Base.getindex(db::DB, word::String, pos::Char) = db[pos, word]
+
 function load_lemmas(base_dir)
     lemmas = Dict{Char, Dict{String, Lemma}}()