set background in jupyter cells

set_background.py

from IPython.core.magic import register_cell_magic
from IPython.display import HTML, display

def set_background(color):    
    script = (
        "var cell = this.closest('.jp-CodeCell');"
        "var editor = cell.querySelector('.jp-Editor');"
        "editor.style.background='{}';"
        "this.parentNode.removeChild(this)"
    ).format(color)

    display(HTML('<img src onerror="{}">'.format(script)))
@register_cell_magic
def background(color, cell):
    set_background(color)
    
    
    
# found at https://stackoverflow.com/a/50824920/4296772