Exemple #1
0
    def test_hours_with_calls(self):
        start = datetime(2012, 1, 1, tzinfo=UTC)
        end = datetime(2012, 6, 30, 23, 59, 59, 999999)
        res = [
            h for h in queue_log_dao.hours_with_calls(self.session, start, end)
        ]

        self.assertEqual(res, [])

        def _insert_at(t):
            queue_log_dao.insert_entry(t, 'hours', 'queue', 'agent', 'event')

        _insert_at('2011-12-31 12:55:22.123123')
        _insert_at('2012-01-01 08:45:23.2345')
        _insert_at('2012-06-30 23:59:59.999999')
        _insert_at('2012-07-01 00:00:00.000000')

        expected = [
            datetime(2012, 1, 1, 8, tzinfo=UTC),
            datetime(2012, 6, 30, 23, tzinfo=UTC)
        ]

        res = [
            h for h in queue_log_dao.hours_with_calls(self.session, start, end)
        ]

        self.assertEqual(res, expected)
Exemple #2
0
def update_db(end_date, start_date=None):
    if start_date is None:
        try:
            start = get_start_time()
        except RuntimeError:
            return
    else:
        start = datetime.datetime.strptime(start_date, '%Y-%m-%dT%H:%M:%S')

    end = datetime.datetime.strptime(end_date, '%Y-%m-%dT%H:%M:%S')

    logger.info('Filling cache into DB')
    logger.info('Start Time: %s, End time: %s', start, end)
    try:
        dao_sess.begin()
        insert_missing_queues(start, end)
        insert_missing_agents()
        dao_sess.commit()

        dao_sess.begin()
        queue.remove_between(dao_sess, start, end)
        agent.remove_after_start(dao_sess, start)
        queue.fill_simple_calls(dao_sess, start, end)
        dao_sess.commit()

        agent.insert_periodic_stat(dao_sess, start, end)

        for period_start in queue_log_dao.hours_with_calls(dao_sess, start, end):
            period_end = period_start + datetime.timedelta(hours=1) - datetime.timedelta(microseconds=1)
            queue.fill_calls(dao_sess, period_start, period_end)
            queue.insert_periodic_stat(dao_sess, period_start, period_end)
    except (IntegrityError, KeyboardInterrupt):
        _clean_up_after_error()
    def test_hours_with_calls(self):
        start = datetime(2012, 01, 01)
        end = datetime(2012, 6, 30, 23, 59, 59, 999999)
        res = [h for h in queue_log_dao.hours_with_calls(self.session, start, end)]

        self.assertEqual(res, [])

        def _insert_at(t):
            queue_log_dao.insert_entry(t, 'hours', 'queue', 'agent', 'event')

        _insert_at('2011-12-31 12:55:22.123123')
        _insert_at('2012-01-01 08:45:23.2345')
        _insert_at('2012-06-30 23:59:59.999999')
        _insert_at('2012-07-01 00:00:00.000000')

        expected = [
            datetime(2012, 1, 1, 8),
            datetime(2012, 6, 30, 23)
        ]

        res = [h for h in queue_log_dao.hours_with_calls(self.session, start, end)]

        self.assertEqual(res, expected)
Exemple #4
0
def update_db(config, end_date, start_date=None):
    if start_date is None:
        try:
            with session_scope() as dao_sess:
                start = get_start_time(dao_sess)
        except RuntimeError:
            return
    else:
        start = datetime.datetime.strptime(start_date, '%Y-%m-%dT%H:%M:%S%z')

    end = datetime.datetime.strptime(end_date, '%Y-%m-%dT%H:%M:%S%z')

    auth_client = AuthClient(**config['auth'])
    token_data = auth_client.token.new(expiration=300)
    confd_client = ConfdClient(**config['confd'])
    confd_client.set_token(token_data['token'])

    logger.info('Getting objects from wazo-confd...')
    confd_queues = confd_client.queues.list(recurse=True)
    confd_agents = confd_client.agents.list(recurse=True)
    master_tenant = token_data['metadata']['tenant_uuid']
    logger.info('Filling cache into DB')
    logger.info('Start Time: %s, End time: %s', start, end)
    with session_scope() as dao_sess:
        insert_missing_queues(dao_sess, start, end, confd_queues['items'],
                              master_tenant)
        insert_missing_agents(dao_sess, confd_agents['items'])
        dao_sess.flush()

        queue.remove_between(dao_sess, start, end)
        agent.remove_after_start(dao_sess, start)
        queue.fill_simple_calls(dao_sess, start, end)
        dao_sess.flush()

        logger.info('Inserting agent periodic stat')
        agent.insert_periodic_stat(dao_sess, start, end)

        logger.info('Inserting queue periodic stat')
        for period_start in queue_log_dao.hours_with_calls(
                dao_sess, start, end):
            period_end = (period_start + datetime.timedelta(hours=1) -
                          datetime.timedelta(microseconds=1))
            queue.fill_calls(dao_sess, period_start, period_end)
            queue.insert_periodic_stat(dao_sess, period_start, period_end)