How to make the “return statement-related” code to become a parameter value of a function?

Question

Here is the working code with some lines omitted for brevity.

class DataGenerator():
    def __init__(self):
        self.unique_id_list = []
        self.dem