-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpush_data.py
More file actions
77 lines (48 loc) · 1.84 KB
/
Copy pathpush_data.py
File metadata and controls
77 lines (48 loc) · 1.84 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import os
import sys
import json
from dotenv import load_dotenv
load_dotenv()
Mongo_url = os.getenv("MONGODB_URL")
print(Mongo_url)
import certifi
ca = certifi.where() #retirevs a list of CA's ( Certificate Authorities )
import pandas as pd
import numpy as np
import pymongo
from Networksecurity.exception.exception import NetwrorkSecurityException
#from Networksecurity.Logging.logger import logging
class NetworkDataExtract():
def __init__(self):
try :
pass
except Exception as e :
raise NetwrorkSecurityException(e,sys)
def csv_to_json_convertor(self,file_path):
try :
data = pd.read_csv(file_path)
data.reset_index(drop=True,inplace=True)
records=list(json.loads(data.T.to_json()).values())
return records
except Exception as e :
raise NetwrorkSecurityException(e,sys)
def insert_data_mongdb(self,records,database,collection):
try :
self.database = database
self.collection = collection
self.records = records
self.mongoclient = pymongo.MongoClient(Mongo_url)
self.database = self.mongoclient[self.database]
self.collection=self.database[self.collection]
self.collection.insert_many(self.records)
return(len(self.records))
except Exception as e :
raise NetwrorkSecurityException(e,sys)
if __name__=='__main__':
FILE_PATH = r'Network_Data\phisingData.csv'
DATABASE = "Sufyaan"
Collection = "NetworkData"
networkobj = NetworkDataExtract()
records = networkobj.csv_to_json_convertor(file_path=FILE_PATH)
no_of_records = networkobj.insert_data_mongdb(records,DATABASE,Collection)
print(no_of_records)