#!/usr/bin/python3
# -*- coding: utf-8 -*-


import sys
from pyeole.bareos import run_bareos_cmd, bareos_query
from creole.client import CreoleClient


def update_needed():
    count_volumes = bareos_query('count_volumes')
    if not count_volumes:
        print('impossible de compter le nombre de volume, le serveur bareos est peut être défaillant')
        sys.exit(1)
    #attention la locale est en_US avec séparateur virgule pour les milliers
    volume_count = int(bareos_query('count_volumes')[0]['Volumes'].replace(',', ''))
    if volume_count > 0:
        return True
    return False


def update_volume_size():
    stdin = "@output /dev/null\nupdate volume\n14\nexit\n"
    code, stdout, stderr = run_bareos_cmd(stdin)
    if code != 0:
        sys.exit(code)


if __name__ == '__main__':
    client = CreoleClient()
    if client.get_creole('activer_bareos_dir', 'non') == 'oui' and update_needed():
        update_volume_size()
    sys.exit(0)
