diff --git a/src/db.jl b/src/db.jl index 197e234..8364583 100644 --- 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}}()