Custom sqlline Prompt

It'd be nice to have a way to set/define a custom sqlline prompt in Drill to display, say, the current workspace it is connected to.

For example:

In Hive one could do,
set hive.cli.print.current.db=true

And in MySQL in /etc/my.cnf,
[mysql]
prompt=\u@\h:[\d]>_