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