def bypass_params(self):
""" get params from widget and bypass to filters """
if self.stocks_tab.rb_var.get() == 'All':
stocks = 'all'
else:
stocks = self.stocks_tab.ent_specified.get()
if self.date_tab.rb_var.get() == 'Today':
date = 'today'
else:
date = self.date_tab.ent_specified.get()
filters_params = []
for i in range(len(self.filters_vars)):
filter_vars = self.filters_vars[i]
if int(filter_vars[0].get()): # [0] for checkbutton
filt = self.filters_descs[i][0]
filter_params = [var.get() for var in filter_vars[1:]]
filters_params.append([filt] + filter_params)
stockfilter.process_stocks(stocks, date, filters_params)
def process_stocks_test(stock, period, filters_params, expect, show_msg_box=False, multi=False):
""" test module """
res = stockfilter.process_stocks(stock, period, filters_params,
show_msg_box=show_msg_box, multi=multi)
if multi:
if res.values.tolist() == expect:
print('pass')
else:
print('fail')
print(res.values.tolist())
print(expect)
else:
if res == expect:
print('pass')
else:
print('fail')
print(res)
print(expect)
