def run(self):
        start = int(time.time())
        end = start + self.interval
        now = start

        while now < end:
            # Sleep for short periods in an attempt to not get too skewed
            time.sleep(30)
            now = int(time.time())

        stoneridge.enqueue(srid=self.srid, nightly=self.nightly,
                           ldap=self.ldap, sha=self.sha,
                           netconfigs=self.netconfigs,
                           operating_systems=self.operating_systems,
                           attempt=self.attempt)
示例#2
0
def main():
    parser = stoneridge.ArgumentParser()
    parser.parse_args()

    root = stoneridge.get_config('enqueuer', 'root')
    username = stoneridge.get_config('enqueuer', 'username')
    password = stoneridge.get_config('enqueuer', 'password')

    try:
        res = requests.get(root + '/list_unhandled', auth=(username, password))
    except:
        # For some reason, we sometimes get a requests failure here, even
        # though everything seems to be working fine. Ignore that, and try
        # again later.
        logging.exception('Error listing unhandled pushes')
        return

    try:
        queue = json.loads(res.text)
    except:
        # Yet another failure mode. Yay.
        logging.exception('Error demarshalling result %s' % (res.text,))
        return

    for entry in queue:
        try:
            requests.post(root + '/mark_handled', data={'id': entry['pushid']},
                          auth=(username, password))
        except:
            # If we fail to mark this as handled, wait until the next try so we
            # don't run the same thing more than once. It's not the end of the
            # world ot have to wait...
            logging.exception('Error marking entry %s as handled' % (entry,))
            return

        logging.debug('Equeuing entry %s' % (entry,))
        stoneridge.enqueue(nightly=False, ldap=entry['ldap'], sha=entry['sha'],
                           netconfigs=entry['netconfigs'],
                           operating_systems=entry['operating_systems'],
                           srid=entry['srid'])