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()