Custom python context manager definition & its usage

custom-context-manager-example.py

#  Context manager definition
class FileOpen(object):
    def __init__(self, file_name, method):
        self.file_object = open(file_name, method)

    def __enter__(self):
        return self.file_object

    def __exit__(self, value, type, traceback):
        self.file_object.close()


# Example usage
with FileOpen('discot.txt', 'w') as opened_file:
    opened_file.write("I am a disco dancer")