implement haskey & get for DB

JuliaText · Feb 7, 2018 · d6df620 · d6df620
1 parent 49b38ae
commit d6df620
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/db.jl b/src/db.jl
@@ -87,3 +87,9 @@ end
 function path_to_index_file(base_dir, pos)
     joinpath(base_dir, "dict", "index.$(SYNSET_TYPES[pos])")
 end
+
+Base.haskey(db::DB, pos::Char) = haskey(db.lemmas, pos)
+Base.haskey(db::DB, pos::Char, word::AbstractString) = haskey(db, pos) ? haskey(db.lemmas[pos], word) : false
+
+Base.get(db::DB, pos::Char, word::AbstractString, default) = haskey(db, pos, word) ? db.lemmas[pos][word] : default
+Base.get(db::DB, word::AbstractString, pos::Char, default) = get(db, pos, word, default)