rofl.py

rofl.py

import dis


class BaseA(type):
    def __init__(self, name, bases, attrs):
        foo_bytecode = [i.opcode for i in dis.Bytecode(attrs["foo"])]

        if not 125 in foo_bytecode:
            raise Exception(
                f"The foo function in class {name} must assign at least one variable."
            )


class A(metaclass=BaseA):
    def dooglederp(self):
        print("doogle")

    def foo(self):
        self.dooglederp()
        baz = "baz"


class B(metaclass=BaseA):
    def foo(self):
        pass