如何在Python中重新使用初始化的类?

时间:2019-11-05 12:33:24

标签: python dataframe

我正在尝试从其他模块访问主应用程序中的已初始化类,但不知道如何执行。 背景:我想在主应用程序的整个执行过程中用数据更新数据框。

我必须遵循以下应用程序结构(这是我的应用程序中代码的简化版本):

constraints
- test_function.py (separate module which should be able to update the initialized class in the main app)
functions
- helper.py (the class which contains the dataframe logic)
main.py (my main application code)

main.py:

import functions.helper
gotstats = functions.helper.GotStats()

gotstats.add(solver_stat='This is a test')
gotstats.add(type='This is a test Type!')

print(gotstats.result())

import constraints.test_function
constraints.test_function.test_function()

helper.py:

class GotStats(object):
    def __init__(self):
        print('init() called')
        import pandas as pd
        self.df_got_statistieken = pd.DataFrame(columns=['SOLVER_STAT','TYPE','WAARDE','WAARDE_TEKST','LOWER_BOUND','UPPER_BOUND','OPTIMALISATIE_ID','GUROBI_ID'])

    def add(self,solver_stat=None,type=None,waarde=None,waarde_tekst=None,lower_bound=None,upper_bound=None,optimalisatie_id=None,gurobi_id=None):
        print('add() called')
        self.df_got_statistieken = self.df_got_statistieken.append({'SOLVER_STAT': solver_stat,'TYPE': type, 'WAARDE': waarde, 'OPTIMALISATIE_ID': optimalisatie_id, 'GUROBI_ID': gurobi_id}, ignore_index=True)

    def result(self):
        print('result() called')
        df_got_statistieken = self.df_got_statistieken
        return df_got_statistieken

test_function.py:

import sys, os
sys.path.append(os.getcwd())

def test_function():
    import functions.helper
    gotstats = functions.helper.GotStats()
    gotstats.add(solver_stat='This is a test from the seperate module')
    gotstats.add(type='This is a test type from the seperate module!')
    print(gotstats.result())

if __name__ == "__main__":
    test_function()

主要,我使用“ gotstats = functions.helper.GotStats()”初始化该类。之后,我可以正确使用其功能,并使用添加功能添加数据框行。 我希望test_function()能够向同一对象添加数据帧行,但我不知道如何执行此操作(在当前代码中,test_function.py只是在其本地命名空间中创建了一个我不想要的新类)。我是否需要使用一个函数扩展类对象以获取活动对象(例如logging.getLogger( name ))?

在正确方向上的任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

让您的test_function将该实例作为参数,并在调用它时将其传递给函数:

main.py:

import functions.helper
from constraints.test_function import test_function


gotstats = functions.helper.GotStats()
gotstats.add(solver_stat='This is a test')
gotstats.add(type='This is a test Type!')

print(gotstats.result())

test_function(gotstats)

test_function.py:

import sys, os
import functions.helper

sys.path.append(os.getcwd())


def test_function(gotstats=None):
    if gotstats is None:
        gotstats = functions.helper.GotStats()

    gotstats.add(solver_stat='This is a test from the seperate module')
    gotstats.add(type='This is a test type from the seperate module!')
    print(gotstats.result())