OCD.py

from pyocd.core.helpers import ConnectHelper
from pyocd.flash.loader import FlashLoader
from pyocd.flash.eraser import FlashEraser
from pyocd.flash.file_programmer import FileProgrammer

class ClientOCD():
    board = None
    target = None
    session = None
   
    def __init__(self):
        self.session = ConnectHelper.session_with_chosen_probe(blocking=False)  
        if None != self.session:
            self.session.open()

    def __del__(self):
        if None != self.session:
            self.session.close()
                  
    def connect(self):
        status = False
        if None != self.session:
            status = True
            self.board = self.session.board
            self.target = self.board.target
        return status        

    def write_target(self, file, address, format):
        programmer = FileProgrammer(self.session)
        programmer.program(file, file_format=format, base_address=address)

    def erase_target(self):
        eraser = FlashEraser(self.session, FlashEraser.Mode.CHIP)
        eraser.erase()
    
    def reset(self):
        self.target.reset_and_halt()
    
    def resume(self):
        self.target.resume()

    def reset_resume(self):
        self.reset()
        self.resume()