forked from ProgrammerAditya36/skindetect
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb_connection.py
36 lines (34 loc) · 1.18 KB
/
db_connection.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
import pymongo
import datetime
url = "mongodb+srv://admin:[email protected]/"
client = pymongo.MongoClient(url)
db = client['skindetect']
userCollection = db['users']
chatCollection = db['chats']
def addUser(formData):
user = {
"first_name": formData.get('first_name'),
"last_name": formData.get('last_name'),
"email": formData.get('email'),
"phone": formData.get('phone'),
"dob": formData.get('dob'),
"gender": formData.get('gender'),
"phone": formData.get('phone'),
"address": formData.get('address'),
"city": formData.get('city'),
"state": formData.get('state'),
"zip_code": formData.get('zip_code'),
"any_previous_diseases": formData.get('any_previous_diseases'),
}
userCollection.insert_one(user)
print("User added to MongoDB")
return user
def store_chat_message(user_id, message, sender='user'):
chat_message = {
"user_id": user_id,
"message": message,
"sender": sender, # 'user' or 'bot'
"timestamp": datetime.datetime.now()
}
chatCollection.insert_one(chat_message)
print(f"Message from {sender} stored in MongoDB")