def __init__(self, *args, **kwargs): GatewayCounter.__init__(self, *args, **kwargs) # If gateway column doesnt exists create it session = Session() gateway_column = session.query(CounterTable).filter( CounterTable.provider == self.provider).first() if gateway_column == None: self._create_column()
def get_counts(self): """Get counts from sql database.""" session = Session() current_date = datetime.date.today() day_timestamp = current_date.strftime('%Y%m%d') month_timestamp = current_date.strftime('%Y%m') # Get gateway conuter data try: gateway_data = session.query(CounterTable).filter( CounterTable.provider == self.provider).first() except SQLAlchemyError, exception: raise CounterError(exception)
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()