CODEWITHSUNDEEP
python
Jurudocs
Jurudocs

Reputation: 9175

How to format output of methods inside of a class in an efficient way?

I want to write a class with much stuff inside. Right now I want to have certain options for the output to show data in a disired format. Im my example a function inside a class should format the output (a former dictionary) as a list. But i really can't figure out how to do it...

here my try:

class TestClass(object):
    def __init__(self):
        pass

    def return_values_as_list(self,otherfunction):
        data_as_list=[ i for i in otherfunction.values()]
        return data_as_list

    def get_data1_from_db(self):
        self.data= {'1_testkey':'1_testvalue',"1_testkey2":"1_testvalue2"}

    def get_data2_from_db(self):
        self.data= {'2_testkey':'2_testvalue',"2_testkey2":"2_testvalue2"}
        return self.data

What i want to have at the end is something like

['1_testvalue','1_testvalue2']

when instantiation looks like the following:

testcase = TestClass()
testcase.get_data1_from_db.return_values_as_list()

Any help with that? I thought also of hooks...but i dont really know how to do that...

Upvotes: 2

Views: 93

Answers (2)

martineau
martineau

Reputation: 123483

I think your class get... methods should be property attributes.

class TestClass(object):
    def __init__(self):
        pass

    @property
    def get_data1_from_db(self):
        data= {'1_testkey':'1_testvalue',"1_testkey2":"1_testvalue2"}
        return data.values()

    @property
    def get_data2_from_db(self):
        data= {'2_testkey':'2_testvalue',"2_testkey2":"2_testvalue2"}
        return data.values()

testcase = TestClass()
print testcase.get_data1_from_db  # ['1_testvalue', '1_testvalue2']
print testcase.get_data2_from_db  # ['2_testvalue', '2_testvalue2']

Upvotes: 1

Steve Mayne
Steve Mayne

Reputation: 22818

def return_values_as_list(self,otherfunction):
    data_as_list= otherfunction().values()
    return data_as_list

You were almost there - you just needed to call the function (add parentheses)

Instantiate with:

testcase.return_values_as_list(testcase.get_data1_from_db)

Upvotes: 1

Related Questions