def set_counts(self, day_amount_change, month_amount_change,
day_trans_change, month_trans_change):
"""Set counts in sql database."""
session = Session()
# Get gateway conuter data
gateway_data = session.query(CounterTable).filter(
CounterTable.provider == self.provider).first()
# Increase counts
gateway_data.day_amount_count += day_amount_change
gateway_data.month_amount_count += month_amount_change
gateway_data.day_trans_count += day_trans_change
gateway_data.month_trans_count += month_trans_change
session.commit()
session.close()
def _create_column(self):
"""Create required column for gateway counters. This is called if
gateway column doesn't exists in database."""
current_date = datetime.date.today()
session = Session()
# Create authorize net column
authorize_net = CounterTable(self.provider)
authorize_net.day_amount_count = 0.0
authorize_net.month_amount_count = 0.0
authorize_net.day_trans_count = 0
authorize_net.month_trans_count = 0
authorize_net.day_count_timestamp = current_date.strftime('%Y%m%d')
authorize_net.month_count_timestamp = current_date.strftime('%Y%m')
session.add(authorize_net)
session.commit()
