general_use.py 文件源码

def check_distribution():
    '''
        This function checks which distribution the user is running and returns that distribution type
    '''

    distro = ''
    distribution = (platform.linux_distribution())[0]

    if 'Kali' == distribution or 'kali' == distribution:
        distro = 'kali'
    elif 'Debian' == distribution or 'debian' == distribution:
        distro = 'debian'
    elif 'Ubuntu' == distribution or 'ubuntu' == distribution:
        distro = 'ubuntu'
    elif 'Red Hat' == distribution or 'red hat' == distribution:
        distro = 'red hat'

    return distro