In Gauge [python] custom_screenshot_writer not working when driver is none

[KrishnaAstra]

Discussed in #2428

^{Originally posted by KrishnaAstra November 8, 2023}
I use Gauge and have created the following Custom screenshot method to get full screen screenshots, It works fine.. However when the script fails before driver instantiation I had returned None but it fails...

Custom method:
@custom_screenshot_writer
def take_screenshot():
"""Custom Utility for Screenshot capture as per the called drivers

Returns:
    screenshot path
"""
screenshot_name = "screenshot-{0}.png".format(uuid1().int)

try:
    if (DriverMgr().driver or DriverMgr().driver != None:
        driver: webdriver= DriverMgr().driver
            
        screenshot_as_png = driver.get_screenshot_as_png()
        screenshot_file_name = os.path.join(os.getenv("gauge_screenshots_dir"), screenshot_name)        
        screenshot_file = open(screenshot_file_name, "wb")
        screenshot_file.write(screenshot_as_png)
        screenshot_file.close
        return os.path.basename(screenshot_file_name)
except Exception as exception:
        return None

Error Message: Exception calling application: expected str, bytes or os.PathLike object, not NoneType

I have some precondition and db verification steps which does not need driver to be present and where I dont have to have screesnhots as well. So what should I return so that Gauge does not throw me error