-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdatabase.py
61 lines (48 loc) · 1.51 KB
/
database.py
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
import datetime
from pymongo import MongoClient
connectionString="mongodb+srv://admin:[email protected]/<dbname>?retryWrites=true&w=majority"
# localConnection=""
client = MongoClient(connectionString)
db=client.scraped
authors = db.authors
def queryDatabase(search_param):
result = authors.find_one({ 'full_name' : search_param }, {'_id':False} )
if(result is not None):
return result
else:
return None
def insertData(name, data):
if(queryDatabase(name) is not None):
authors.replace_one({ 'full_name' : name }, {
'full_name' : name,
'date' : datetime.datetime.now(),
'research_gate' : data['research_gate'],
'google' : data['google']
})
else:
authors.insert_one({
'full_name' : name,
'date' : datetime.datetime.now(),
'research_gate' : data['research_gate'],
'google' : data['google']
})
insertedData = queryDatabase(name)
return insertedData
def insertTest(name):
authors.insert_one({
'full_name' : name,
'date' : datetime.datetime.now(),
'research_gate' : 'No data',
'google' : 'No data'
})
inserted = queryDatabase(name)
return inserted
def replaceTest(name, newname):
authors.replace_one({ 'full_name' : name }, {
'full_name' : newname,
'date' : datetime.datetime.now(),
'research_gate' : 'No data',
'google' : 'No data'
})
replaced = queryDatabase(newname)
return replaced