#!/usr/local/mig_python/bin/python3
# -*- coding: utf-8 -*-
# Copyright (c) 2024 TencentOS Server

import sys
import os

lib_path = '/usr/local/migrate2tencentos/'
sys.path.insert(0, os.path.join(lib_path))

from Backup.BackupCli import BackupCli
from Cli import check

def check_disk_space(dir):
    root = os.path.abspath(dir)
    stat = os.statvfs(root)
    free_space = stat.f_frsize * stat.f_bavail
    print("disk space/GB: ", int(free_space / (1024 * 1024 * 1024)))
    if free_space > 10 * 1024 * 1024 * 1024:
        return True
    else:
        return False

if __name__ == '__main__':
    check()
    backup = BackupCli(sys.argv)
    bakdir = backup.dir
    if not check_disk_space(bakdir):
        print('disk space is less than 10G')
        sys.exit(1)
    backup.run()