# put/rename the file in YOUR CONFIG/qtvcp/screens/qtdragon_hd/qtdragon_hd_handler.py

import sys
import importlib
from PyQt5.QtGui import QColor
from qtvcp.core import Path

PATH = Path()

# get reference to original handler file so we can subclass it
sys.path.insert(0, PATH.SCREENDIR)
module = "{}.{}_handler".format(PATH.BASEPATH,PATH.BASEPATH)
mod = importlib.import_module(module, PATH.SCREENDIR)
sys.path.remove(PATH.SCREENDIR)
HandlerClass = mod.HandlerClass

# return our subclassed handler object
def get_handlers(halcomp, widgets, paths):
    return [UserHandlerClass(halcomp, widgets, paths)]

class UserHandlerClass(HandlerClass):
    print('\nCustom subclassed handler for custom gcode cursor color - loaded\n')

    def init_widgets(self):
        # call the original function:
        super().init_widgets()
        self.w.gcode_viewer.editor.setCaretForegroundColor(QColor("white"))
        self.w.gcode_editor.editor.setCaretForegroundColor(QColor("white"))