Add partial `get_db` support