BIOSUtilities/common/a7z_comp.py

#!/usr/bin/env python3
#coding=utf-8

"""
Copyright (C) 2022 Plato Mavropoulos
"""

import os
import subprocess

from common.path_ops import project_root, safe_path
from common.system import get_os_ver
from common.system import printer

# Get 7z path
def get_7z_path(static=False):
    exec_name = '7z.exe' if get_os_ver()[1] else ('7zzs' if static else '7zz')
    
    return safe_path(project_root(), ['external',exec_name])

# Check if file is 7z supported
def is_7z_supported(in_path, padding=0, static=False):
    try:
        subprocess.run([get_7z_path(static), 't', in_path, '-bso0', '-bse0', '-bsp0'], check=True)
    except:
        printer(f'Error: 7-Zip could not check support for file {in_path}!', padding)
        
        return False
    
    return True

# Archive decompression via 7-Zip
def a7z_decompress(in_path, out_path, in_name, padding=0, static=False):
    if not in_name: in_name = 'archive'
    
    try:
        subprocess.run([get_7z_path(static), 'x', '-aou', '-bso0', '-bse0', '-bsp0', '-o' + out_path, in_path], check=True)
        
        if not os.path.isdir(out_path): raise Exception('EXTRACT_DIR_MISSING')
    except:
        printer(f'Error: 7-Zip could not extract {in_name} file {in_path}!', padding)
        
        return 1
    
    printer(f'Succesfull {in_name} decompression via 7-Zip!', padding)
    
    return 0