stubber.codemod.add_method

Codemod to add methods to a classes.

Used to add methods that are documented, but are not reported by the firmware, so they are also not present in the board stubs.

Module Contents

Classes

CallFinder	Find the Pin.__call__ method and extract it from a (machine) module.
CallAdder	Add a __call__ method to a class if it is missing.

class stubber.codemod.add_method.CallFinder

Bases: libcst.matchers.MatcherDecoratableTransformer

Find the Pin.__call__ method and extract it from a (machine) module.

class_name: str = 'Pin'

method_name: str = '__call__'

detect_call(node: libcst.FunctionDef) → None

find the __call__ method and store it.

Parameters:

node (libcst.FunctionDef) –

Return type:

None

class stubber.codemod.add_method.CallAdder(call_meth: libcst.FunctionDef)

Bases: libcst.matchers.MatcherDecoratableTransformer

Add a __call__ method to a class if it is missing.

Parameters:

call_meth (libcst.FunctionDef) –

class_name = 'Pin'

has_call = 0

detect_call(node: libcst.FunctionDef) → None

Detect if the class already has a __call__ method.

Parameters:

node (libcst.FunctionDef) –

Return type:

None

add_call(original_node: libcst.ClassDef, updated_node: libcst.ClassDef) → libcst.ClassDef

Add the __call__ method to the class if it is not already there.

Parameters:

• original_node (libcst.ClassDef) –

• updated_node (libcst.ClassDef) –

Return type:

libcst.ClassDef