2016-04-27 332 views
1

以下代码在秒组中查找0.0.0.0/0,并在松弛状态下打印。如何从输出中去掉unicode'u'? Python

def getTag(connection, instanceId): 
    reservations=connection.get_all_instances(filters={'instance_id':instanceId}) 
    for res in reservations: 
     for instance in res.instances: 
     if instance.state == "running": 
      return instance.tags['Name'],instance.private_ip_address,instance.region 
     else: 
      return None 

#regions = ['us-east-1','us-west-1','us-west-2','eu-west-1','sa-east-1','ap-southeast-1','ap-southeast-2','ap-northeast-1'] 
#regions = ['us-west-2','eu-central-1','ap-southeast-1'] ##Prod VPC 
regions = ['us-west-2'] 
data = [] 
header = ["Port","Open For","Security group","Instance Details"] 

for region in regions: 
    connection=ec2.connect_to_region(region) 
    sg = connection.get_all_security_groups() 
    try: 
     for securityGroup in sg: 
      for rule in securityGroup.rules: 
       if '0.0.0.0/0' in str(rule.grants): 
        for instanceid in securityGroup.instances(): 
         instanceId=str(instanceid) 
        # print instanceId 
         tag = getTag(connection, instanceId.split(':')[1]) 
         if tag is not None: 
         # print tab    
         tab =[str(rule.to_port),"0.0.0.0/0",str(securityGroup.name),str(getTag(connection, instanceId.split(':')[1]))] 
         data.append(tab) 
    except Exception,e: 
     pass 

print tabulate(data, headers=header, tablefmt='simple') 
cool_stuff = tabulate(data, headers=header, tablefmt='simple') 
slack.chat.post_message('#scanbot','```'+cool_stuff+'```',username='AWS_security_group_scanner') 

的问题是,这个打印:

Port Open For Security group      Instance Details 
------ ---------- --------------------------------- ---------------------------------------------------------------------------------- 
    1433 0.0.0.0/0 default       (u'ABCcorp', u'10.17.9.22', RegionInfo:us-west-2) 
    80 0.0.0.0/0 pdx01-ms-dev-elbs     (u'dev-lb05', u'10.17.5.249', RegionInfo:us-west-2) 
    80 0.0.0.0/0 pdx01-ms-dev-elbs     (u'dev-sdl-lb02', u'10.17.5.63', RegionInfo:us-west-2) 
    80 0.0.0.0/0 pdx01-ms-dev-elbs     (u'lbCasio', u'10.17.5.11', RegionInfo:us-west-2) 
    80 0.0.0.0/0 pdx01-ms-dev-elbs     (u'lb01', u'10.17.4.196', RegionInfo:us-west-2) 
    80 0.0.0.0/0 pdx01-ms-dev-elbs     (u'lb01', u'10.17.5.133', RegionInfo:us-west-2) 
    80 0.0.0.0/0 pdx01-ms-dev-elbs     (u'gehc-lb01', u'10.17.5.222', RegionInfo:us-west-2) 
    443 0.0.0.0/0 pdx01-ms-dev-elbs     (u'lb05', u'10.17.5.249', RegionInfo:us-west-2) 

我想因为它们与松弛的东西搞乱摆脱的“U”。有人能帮我吗 ?

+0

尝试打印字符串,而不是一个元组 –

+0

使用'印刷海峡的(东西)',而不是'print something' – EbraHim

+0

正如链接问题所写,为什么不使用Python 3呢? – hyst329

回答

2

更改此

tab = [str(rule.to_port), "0.0.0.0/0", str(securityGroup.name), str(getTag(connection, instanceId.split(':')[1]))] 

tab = [str(rule.to_port), "0.0.0.0/0", str(securityGroup.name), tuple(list(i.encode('UTF8') for i in getTag(connection, instanceId.split(':')[1])[0:2]) + [getTag(connection, instanceId.split(':')[1])[2]])] 
+0

这工作正常,但最后'RegionInfo'丢失 –

+0

我以为你不会想这(无需更改)被转换成任何类型。 – AceLearn

+0

我的意思是我只是需要摆脱'u',你的代码看起来很好,但是'RegionInfo'没有在输出中打印出来。其余的都没问题.. –

2

希望这个作品

import unicodedata 
for region in regions: 
    connection=ec2.connect_to_region(region) 
    sg = connection.get_all_security_groups() 
    try: 
     for securityGroup in sg: 
      for rule in securityGroup.rules: 
       if '0.0.0.0/0' in str(rule.grants): 
        for instanceid in securityGroup.instances(): 
         instanceId=str(instanceid) 
        # print instanceId 
         tag = getTag(connection, instanceId.split(':')[1]) 
         if tag is not None: 
         # print tab    
         tab =[str(rule.to_port),"0.0.0.0/0",str(securityGroup.name),unicodedata.normalize('NFKD', getTag(connection, instanceId.split(':')[1])).encode('ascii','ignore')] 
         data.append(tab) 
    except Exception,e: 
     pass 
+0

这是不按预期工作,只是突破了第一个循环..没有数据没有任何东西 –