查询MongoDB中相关数据,并且写入相对应文件

Server字段中有IIS,Apache等服务名称,查询Server字段存在不为空的value并且按按照分类写入txt中
from libs.Conn_scan import Mongo

f1 = open('Result/Server/Apache.txt', 'a')
f2 = open('Result/Server/Nginx.txt', 'a')
f3 = open('Result/Server/vWebServer.txt', 'a')
f4 = open('Result/Server/OpebResty.txt', 'a')
f5 = open('Result/Server/Tengine.txt', 'a')
f6 = open('Result/Server/APUSIC.txt', 'a')
f7 = open('Result/Server/365GCD.txt', 'a')
f8 = open('Result/Server/TWebAP.txt', 'a')
f9 = open('Result/Server/Marco.txt', 'a')

s = Mongo.coll['Result'].find({'short_server' : {"$exists" : True}})
print Mongo.coll['Result'].count()
for i in s:
    print i['URL']
    if 'Apache' in i['short_server'] :
        f1.write(i['URL'] + '\n')
        
    if 'Nginx' in i['short_server'] :
        f2.write(i['URL'] + '\n')
           
    if 'vWebServer' in i['short_server'] :
        f3.write(i['URL'] + '\n')
         
    if 'OpebResty' in i['short_server'] :
        f4.write(i['URL'] + '\n')
            
    if 'Tengine' in i['short_server'] :
        f5.write(i['URL'] + '\n')

    if 'APUSIC' in i['short_server'] :
        f6.write(i['URL'] + '\n') 
        
    if '365GCD' in i['short_server'] :
        f7.write(i['URL'] + '\n') 
        
    if 'TWebAP' in i['short_server'] :
        f8.write(i['URL'] + '\n') 
        
    if 'Marco' in i['short_server'] :
        f9.write(i['URL'] + '\n') 
        
        
f2.close()
f1.close()
f3.close()
f4.close()
f5.close()
f6.close()
f7.close()
f8.close()
f9.close()
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容