python查詢mongodb內數據
阿新 • • 發佈：2017-11-03
%d mongod 運行時間 from 一個 imp item ocl .cn
初學者，寫的不好請指出。
#第一步以insertTime為條件查詢時間段內的數據
#第二部步可以選擇是否再以通話Id為條件篩選第一步所查詢出來的數據
#因為使用的是配置文件，所以首先在代碼當前目錄下創建一個配置文件，db.conf
代碼：
1 import configparser 2 import pymongo 3 from pymongo import MongoClient 4 import time 5 6 print(‘以insertTime為條件查詢；可以按通話id篩選數據‘) 7 count = 0 8 f = open(‘獲取數據.txt‘, ‘w‘,encoding=‘utf-8‘) #在寫入文件夾是會報編碼格式錯誤，所以增加encoding=‘utf-8‘解決 9 cf =configparser.ConfigParser() 10 #讀取當前目錄下的配置文件db.conf 11 cf.read(‘db.conf‘) 12 get_start_time = cf.getint(‘mongodb‘,‘start_time‘) 13 get_end_time = cf.getint(‘mongodb‘,‘end_time‘) 14 get_call_id = cf.get(‘mongodb‘,‘call_id‘)15 #將時間戳毫秒轉換為秒，db內的時間戳為毫秒單位，python的time為秒單位，所以除以1000 16 s_local_time = time.localtime(get_start_time/1000) 17 e_local_time = time.localtime(get_end_time/1000) 18 #將秒轉換為日期格式2017-01-01 12:00:00 19 s_time = time.strftime(‘%Y-%m-%d %H:%M:%S‘,s_local_time) 20 e_time = time.strftime(‘%Y-%m-%d %H:%M:%S‘,e_local_time) 2122 print(‘數據庫IP:‘,cf.get(‘mongodb‘,‘db_ip‘),‘\n數據庫Port:‘,cf.getint(‘mongodb‘,‘db_port‘)) 23 print(‘查詢開始時間:‘,s_time,‘\n查詢結束時間:‘,e_time) 24 print(‘通話ID:‘,get_call_id) 25 client = MongoClient(cf.get(‘mongodb‘,‘db_ip‘),cf.getint(‘mongodb‘,‘db_port‘)) 26 print(‘****************************************************‘) 27 28 #異常處理，連接失敗走except pymongo.errors.ServerSelectionTimeoutError並退出，成功走else路徑。 29 try: 30 print(‘正在連接...‘) 31 client.database_names() #檢查是否能獲取到數據庫列表,獲取不到說明連接失敗。 32 print(‘連接成功...‘) 33 print(‘****************************************************‘) 34 except pymongo.errors.ServerSelectionTimeoutError: 35 print(‘無法連接,請查看數據庫是否啟成功，或數據庫IP和Port是否配置正確。‘) 36 print(‘****************************************************‘) 37 else: 38 # 輸出數據庫列表,循環輸出 39 print(‘數據庫名稱列表:‘) 40 database_names_num = 1 #為數據庫編號，從1開始，循環一次+1 41 for d_list in client.database_names(): 42 print(database_names_num,‘:‘,d_list) 43 database_names_num += 1 44 print(‘****************************************************‘) 45 while True: 46 #異常處理，如輸入非整數走下一步except ValueError，重新開始循環。否則走else路徑。 47 try: 48 num_1 = int(input(‘請以編號選擇數據庫：‘)) 49 except ValueError: 50 print(‘輸入類型錯誤，請重新輸入。‘) 51 print(‘****************************************************‘) 52 else : 53 #判斷輸入的編號是否大於數據庫的數量，或小於1 54 if num_1 > len(client.database_names()) or num_1 < 1 : 55 print(‘沒有找到這個數據庫，請重新輸入。‘) 56 print(‘****************************************************‘) 57 else: 58 #因數據庫實際全部存在為一個列表內，索引從0開始，我們為數據庫設置的編號為從1開始，所以需要-1 59 db = client[client.database_names()[num_1-1]] #數據庫名稱 60 print(‘****************************************************‘) 61 #print(db) 62 63 #列出數據庫內集合列表，循環輸出 64 print(‘數據庫內集合列表：‘) 65 collection_names_num = 1 #為集合編號，從1開始，循環一次+1 66 for c_list in db.collection_names(): 67 print(collection_names_num, ‘:‘,c_list ) 68 collection_names_num += 1 69 print(‘****************************************************‘) 70 71 while True: 72 # 異常處理，如輸入非整數走下一步except ValueError，重新開始循環。否則走else路徑。 73 try: 74 num_2 = int(input(‘請以編號選擇數據庫集合：‘)) 75 print(‘****************************************************‘) 76 except ValueError: 77 print(‘輸入類型錯誤，請重新輸入。‘) 78 print(‘****************************************************‘) 79 else: 80 # 判斷輸入的編號是否大於數據庫內集合的數量，或小於1 81 if num_2 > int(len(db.collection_names())) or num_2 < 1: 82 print(‘沒有找到這個集合，請重新輸入。‘) 83 print(‘****************************************************‘) 84 #print(db.collection_names()) 85 else: 86 # 因數據庫集合實際全部存在為一個列表內，索引從0開始，我們為數據庫集合設置的編號為從1開始，所以需要-1 87 my_db = db[db.collection_names()[num_2-1]] 88 #print(my_db) 89 n = input(‘是否需要篩選通話id,輸入“Y”或“N”：‘) 90 print(‘****************************************************‘) 91 print(‘開始運行...‘) 92 93 #這個循環分為3條路徑，輸入n，y個一條，然後輸入別的任何字符走else路徑，並重新循環 94 while True: 95 if n == ‘n‘ or n == ‘N‘: 96 s__time = time.clock() # 本地開始查詢時間 97 #find查詢條件，循環遍歷集合，查詢到符合條件的數據存到變量data 98 for data in my_db.find({"insertTime": {‘$gte‘: get_start_time, ‘$lte‘: get_end_time}}): 99 #print(data) 100 f.write(str(data) + ‘\n‘) #寫入文件內 101 count += 1 #統計共輸出數據的條數，初始為0，循環一次+1 102 e__time = time.clock() #本地結束查詢時間 103 #耗時等於結束時間減去開始時間 104 print(‘運行結束，共查詢到‘, count, ‘條數據，共耗時‘,int(e__time - s__time), ‘秒。‘) 105 break 106 107 elif n == ‘y‘ or n ==‘Y‘: 108 # 開始運行時間 109 s__time = time.clock() # 本地開始查詢時間 110 # find查詢條件，循環遍歷集合，查詢到符合條件的數據存到變量data 111 for data in my_db.find({"insertTime": {‘$gte‘: get_start_time, ‘$lte‘: get_end_time}}): 112 #k為key,v為value,循環遍歷data，查詢value內含有通話ID的數據寫入到文件 113 for k, v in data.items(): 114 if get_call_id in str(v): 115 #print(data) 116 f.write(str(data) + ‘\n‘) 117 count += 1 #統計共輸出數據的條數，初始為0，循環一次+1 118 e__time = time.clock() #本地結束查詢時間 119 # 耗時等於結束時間減去開始時間 120 print(‘運行結束，共查詢到‘, count, ‘條數據，共耗時‘, int(e__time - s__time), ‘秒。‘) 121 break 122 break 123 break 124 input(‘按回車鍵結束...‘)
python查詢mongodb內數據