Python win_unicode_console.enable方法代碼示例

# 需要導入模塊: import win_unicode_console [as 別名]
# 或者: from win_unicode_console import enable [as 別名]
def set_colors():
	global G, Y, B, R, W , M , C , end ,Bold,underline
	if os.name=="nt":
		try:
			import win_unicode_console , colorama
			win_unicode_console.enable()
			colorama.init()
			#green - yellow - blue - red - white - magenta - cyan - reset
			G,Y,B,R,W,M,C,end= '\033[92m','\033[93m','\033[94m','\033[91m','\x1b[37m','\x1b[35m','\x1b[36m','\033[0m'
			Bold = "\033[1m"
			underline = "\033[4m"
		except:
			G = Y = B = R = W = G = Y = B = R = Bold = underline = ''
	else:
		#import colorama
		#colorama.init()
		#green - yellow - blue - red - white - magenta - cyan - reset
		G,Y,B,R,W,M,C,end= '\033[92m','\033[93m','\033[94m','\033[91m','\x1b[37m','\x1b[35m','\x1b[36m','\033[0m'
		Bold = "\033[1m"
		underline = "\033[4m" 

# 需要導入模塊: import win_unicode_console [as 別名]
# 或者: from win_unicode_console import enable [as 別名]
def init_output():
    import colorama
    win_unicode_console.enable()
    colorama.init() 

# 需要導入模塊: import win_unicode_console [as 別名]
# 或者: from win_unicode_console import enable [as 別名]
def set_colors():
	global G, Y, B, R, W , M , C , end
	if sys.platform.startswith('win'):
		# Windows deserve coloring too :D
		try:
			import win_unicode_console , colorama
			win_unicode_console.enable()
			colorama.init()
			#Now the unicode will work ^_^
			G = '\033[92m'  # green
			Y = '\033[93m'  # yellow
			B = '\033[94m'  # blue
			R = '\033[91m'  # red
			W = '\033[0m'   # white
			M = '\x1b[35m'  # magenta
			C = '\x1b[36m'  # cyan
			end = '\33[97m'
		except:
			#print("[!] Error: Coloring libraries not installed ,no coloring will be used [Check the readme]")
			G = Y = B = R = W = G = Y = B = R = W = ''
	else:
		G = '\033[92m'  # green
		Y = '\033[93m'  # yellow
		B = '\033[94m'  # blue
		R = '\033[91m'  # red
		W = '\033[0m'   # white
		M = '\x1b[35m'  # magenta
		C = '\x1b[36m'  # cyan
		end = '\33[97m' 

# 需要導入模塊: import win_unicode_console [as 別名]
# 或者: from win_unicode_console import enable [as 別名]
def init():
    os.environ['version'] = get_version()

    win_unicode_console.enable()
    colorama.init()
    description()

    args = get_args()

    levels = ['NOTSET', 'WARN', 'INFO', 'DEBUG']
    level = os.getenv('LOG_LEVEL')
    if not level:
        level = levels[args.log] if args.log else 'NOTSET'
    # end if
    if level != 'NOTSET':
        os.environ['debug_mode'] = 'yes'
        logging.basicConfig(
            level=logging.getLevelName(level),
            format=Fore.CYAN + '%(asctime)s '
            + Fore.RED + '[%(levelname)s] '
            + Fore.YELLOW + '(%(name)s)\n'
            + Fore.WHITE + '%(message)s' + Fore.RESET,
        )
        debug_mode(level)
    # end if

    if args.suppress:
        input_suppression()
        print(args)
    # end if

    if args.bot:
        os.environ['BOT'] = args.bot
    # end if

    for key, val in args.extra.items():
        os.environ[key] = val[0]
    # end for

    # requests.urllib3.disable_warnings(
    #     requests.urllib3.exceptions.InsecureRequestWarning)
    # # end if
# end def 

# 需要導入模塊: import win_unicode_console [as 別名]
# 或者: from win_unicode_console import enable [as 別名]
def processProject(vbaParser, args, output_file = sys.stdout):
    try:
        vbaProjects = vbaParser.find_vba_projects()
        if vbaProjects is None:
            return
        if output_file.isatty() and WIN_UNICODE_CONSOLE:
            win_unicode_console.enable()
        for vbaRoot, _, dirPath in vbaProjects:
            print('=' * 79, file=output_file)
            if not args.disasmOnly:
                print('dir stream: {}'.format(dirPath), file=output_file)
            dirData, codeModules, is64bit = processDir(vbaParser, dirPath, args, output_file=output_file)
            vbaProjectPath = vbaRoot + 'VBA/_VBA_PROJECT'
            vbaProjectData = process_VBA_PROJECT(vbaParser, vbaProjectPath, args, output_file=output_file)
            identifiers = getTheIdentifiers(vbaProjectData)
            if not args.disasmOnly:
                print('Identifiers:', file=output_file)
                print('', file=output_file)
                i = 0
                for identifier in identifiers:
                    print('{:04X}: {}'.format(i, identifier), file=output_file)
                    i += 1
                print('', file=output_file)
                print('_VBA_PROJECT parsing done.', file=output_file)
                print('-' * 79, file=output_file)
            print('Module streams:', file=output_file)
            for module in codeModules:
                modulePath = vbaRoot + 'VBA/' + module
                # make sure it is unicode, because that is what vbaParser expects:
                if PYTHON2:
                    # modulePath is UTF8 bytes (see processDir)
                    modulePath_unicode = modulePath.decode('utf8', errors='replace')
                else:
                    # modulePath is already unicode
                    modulePath_unicode = modulePath
                moduleData = vbaParser.ole_file.openstream(modulePath_unicode).read()
                print ('{} - {:d} bytes'.format(modulePath, len(moduleData)), file=output_file)
                pcodeDump(moduleData, vbaProjectData, dirData, identifiers, is64bit, args, output_file=output_file)
        if output_file.isatty() and WIN_UNICODE_CONSOLE:
            win_unicode_console.disable()
    except Exception as e:
        print('Error: {}.'.format(e), file=sys.stderr) 

# 需要導入模塊: import win_unicode_console [as 別名]
# 或者: from win_unicode_console import enable [as 別名]
def main():
    """
    Run main script
    """
    global R, B, C, W, Y, url, spinner
    sys.path.append(os.path.dirname(os.path.realpath(__file__)))
    import spin
    R, B, Y, C, W = '\033[1;31m', '\033[1;37m', '\033[93m', '\033[1;30m', '\033[0m'
    args = vars(parse_args())
    
    
    # OS Compatibility : Coloring
   
    if sys.platform.startswith('win'):
        try:
            import win_unicode_console, colorama
            win_unicode_console.enable()
            colorama.init()
        except:
            print('[+] Error: Coloring libraries not installed')
            R, B, Y, C, W = '', '', '', '', ''
    
    if len(sys.argv) < 2 :
        header()
        print('{}Usage: r3con1z3r -d domain.com\n'.format(Y, C))
        print('{}Example: r3con1z3r -d google.com\n'.format(Y, C))
        print('{}[!] Please specify a domain'.format(Y, C))
        sys.exit()
    else:
        if args['about']:
            about()
        else: 
            url = args['domain']
            if  url[0:8] == 'https://':
                url = url.replace("https://","")
            elif url[0:7] == 'http://':
                url = url.replace("http://","")
            else:
                url = url
                
                # declare global spinner
                spinner = spin.create_spinner(before="Generating Report")
                gaphy()