from glob import glob
import os
def _get_malware_dir(fp):
    """[get list of directory of all malwares]
    /datasets/dsc180a-wi20-public/Malware/amd_data_smali
    Arguments:
        fp {[string]} -- [the root directory of amd_data_smali]
    """    
    malware_dir = glob(os.path.join(fp, '*'))
    malware_dir = [glob(i+'/*') for i in malware_dir]
    malware_dir = [item for sublist in malware_dir for item in sublist]
    malware_dir = [glob(i + '/*') for i in malware_dir]
    malware_dir = [item for sublist in malware_dir for item in sublist]
    return malware_dir