def model_query(model, *args, **kwargs): """Query helper that accounts for context's `read_deleted` field. """ session = kwargs.get('session') or get_session() query = session.query(model, *args) return query
def pickdata_update(pickdata_id, values, session=None): if session is None: session = get_session() pickdata_ref = pickdata_get(pickdata_id, session) pickdata_ref.update(values) pickdata_ref.save(session=session)
def sensor_create(values, session=None): sensor_ref = models.Sensor() sensor_ref.update(values) if session is None: session = get_session() sensor_ref.save(session=session) return sensor_ref
def pickdata_create(values, session=None): pickdata_ref = models.Pickdata() pickdata_ref.update(values) if session is None: session = get_session() pickdata_ref.save(session=session) return pickdata_ref
def batterys_update(batterys_id, values, session=None): if session is None: session = get_session() batterys_ref = batterys_get_by_id(batterys_id, session) batterys_ref.update(values) batterys_ref.save(session=session)
def btrundata_create(values, session=None): btrundata_ref = models.Btrundata() if session is None: session = get_session() btrundata_ref.update(values) btrundata_ref.save(session=session) return btrundata_ref
def battery_create(values, session=None): battery_ref = models.Battery() if session is None: session = get_session() battery_ref.update(values) battery_ref.save(session=session) return battery_ref
def battery_update(group_id, serial_num, values, session=None): if session is None: session = get_session() battery_ref = battery_get_by_groupid_and_serialnum(group_id, serial_num, session) battery_ref.update(values) battery_ref.save(session=session)
def pickdata_delete_by_time(time, session): if session is None: session = get_session() pickdata_refs = model_query(models.Pickdata, session=session).\ filter_by(models.Pickdata.PICKEDTIME_D < time).\ all() for pickdata_ref in pickdata_refs: session.delete(pickdata_ref) session.flush()
def pickdata_get(pickdata_id, session=None): if session is None: session = get_session() result = model_query(models.Pickdata, session=session).\ filter_by(RECORD_ID=pickdata_id).\ first() if not result: LOG.error('failed to find Pickdata from pickdata_id : %d', pickdata_id) return None return result
def sensor_get_by_id(sensor_id, session=None): if session is None: session = get_session() result = model_query(models.Sensor, session=session).\ filter_by(RECORD_ID=sensor_id).\ first() if not result: LOG.error('failed to find sensor from sensor_id : %d', sensor_id) return None return result
def warning_get_by_basetype(base_type, session=None): if session is None: session = get_session() result = model_query(models.Warning, session=session).\ filter_by(BASE_TYPE=base_type).\ first() if not result: LOG.error('failed to find warning from base_type : %d', base_type) return None return result
def warning_get_by_id(warning_id, session=None): if session is None: session = get_session() result = model_query(models.Warning, session=session).\ filter_by(RECORD_ID=warning_id).\ first() if not result: LOG.error('failed to find warning from warning_id : %d', warning_id) return None return result
def save(self, session=None): """Save this object.""" if not session: session = get_session() session.add(self) try: session.flush() except IntegrityError, e: if str(e).endswith('is not unique'): raise exception.Duplicate() else: raise
def dictbattery_get_by_id(dictbattery_id, session=None): if session is None: session = get_session() result = model_query(models.Dictbattery, session=session).\ filter_by(RECORD_ID = dictbattery_id).\ first() if not result: LOG.error('failed to find dictbattery from dictbattery_id : %s', dictbattery_id) return None return result
def battery_get_by_groupid_and_serialnum(group_id, serial_num, session=None): if session is None: session = get_session() result = model_query(models.Battery, session=session).\ filter_by(GROUP_V = group_id).\ filter_by(SERIAL_N = serial_num).\ first() if not result: LOG.error('failed to find battery from group_id : %s, serial_num :%s', group_id,serial_num) return None return result
def battery_get_count_by_status(group_id, status_n,session=None): if session is None: session = get_session() result = model_query(models.Battery, session=session).\ filter_by(GROUP_V = group_id).\ filter_by(STATUS_N = status_n).\ count() if not result: LOG.error('failed to find battery from group_id : %s, status_n :%s', group_id,status_n) return None return result
def batterys_get_by_id(batterys_id, session=None): if session is None: session = get_session() result = model_query(models.Batterys, session=session).\ filter_by(RECORD_ID = batterys_id).\ first() if not result: LOG.error('failed to find batterys from batterys_id : %s', batterys_id) return None return result
def cyclesetting_get_cycle(id=1, session=None): if session is None: session = get_session() cycle_ref = model_query(models.CycleSetting, session=session).\ filter_by(RECORD_ID = id).\ first() if not cycle_ref: LOG.error('failed to find cycle') return None cycle = cycle_ref.CYCLE_N return cycle
def battery_get_by_groupid_and_serialnum(group_id, serial_num, session=None): if session is None: session = get_session() result = model_query(models.Battery, session=session).\ filter_by(GROUP_V = group_id).\ filter_by(SERIAL_N = serial_num).\ first() if not result: LOG.error('failed to find battery from group_id : %s, serial_num :%s', group_id, serial_num) return None return result
def battery_get_count_by_status(group_id, status_n, session=None): if session is None: session = get_session() result = model_query(models.Battery, session=session).\ filter_by(GROUP_V = group_id).\ filter_by(STATUS_N = status_n).\ count() if not result: LOG.error('failed to find battery from group_id : %s, status_n :%s', group_id, status_n) return None return result
def sensor_get_all(session=None): if session is None: session = get_session() query = model_query(models.Sensor, session) query.all()