def updateWeightsByJson(userId, categoryWeights):
print("invoked")
catWeights = json.loads(categoryWeights)
#print(catWeights['nature'])
incStr = {
"$inc": {
"categories.landmark": catWeights['landmark'],
"categories.nature": catWeights['nature'],
"categories.shopping": catWeights['shopping'],
"categories.theatre": catWeights['theatre'],
"categories.restaurant": catWeights['restaurant']
}
}
#incStr = '''{"$inc":{"'''
#for key,value in catWeights.items():
#print (key,value)
#incStr += "categories."+key+":"+value
#incStr +='''}'''
#print(incStr)
conn = MongoConnection()
user_coll = conn.getUsersCollection()
row = user_coll.update_one({"userId": userId}, incStr)
if (row.acknowledged):
print("Weights updated successfully")
return row.acknowledged
def fetchUser(userId):
query = {"userId": userId}
conn = MongoConnection()
user_coll = conn.getUsersCollection()
row = user_coll.find_one(query)
#print (row)
return row
def insertUser(name,
age,
gender,
budget=0,
duration=0,
visited={},
prefCategories=""):
landmark = nature = shopping = theatre = restaurant = 0
for aCat in prefCategories.split(","):
#print(aCat)
if (aCat == "landmark"):
landmark = 1
elif (aCat == "nature"):
nature = 1
elif (aCat == "shopping"):
shopping = 1
elif (aCat == "theatre"):
theatre = 1
else:
restaurant = 1
'''print("landmark",landmark)
print("nature",nature)
print("shopping",shopping)
print("theatre",theatre)
print("restaurant",restaurant)'''
userId = randint(1000, 9999)
categories = {
"landmark": landmark,
"nature": nature,
"shopping": shopping,
"theatre": theatre,
"restaurant": restaurant
}
userDetails = {
"userId": userId,
"name": name,
"age": age,
"gender": gender,
"avgBudget": budget,
"avgDuration": duration,
"visited": visited,
"categories": categories
}
conn = MongoConnection()
print(conn.dummy())
user_coll = conn.getUsersCollection()
row = user_coll.insert_one(userDetails)
print(row.inserted_id)
if (row.acknowledged):
return userId
def updateWeights(userId, landmarkWeight, natureWeight, shopWeight,
theatreWeight, restoWeight):
print("categories.landmark", landmarkWeight, "categories.natureWeight",
natureWeight, "categories.shopWeight", shopWeight,
"categories.theatreWeight", theatreWeight, "categories.restoWeight",
restoWeight)
incStr = {
"$inc": {
"categories.landmark": landmarkWeight,
"categories.nature": natureWeight,
"categories.shopping": shopWeight,
"categories.theatre": theatreWeight,
"categories.restaurant": restoWeight
}
}
conn = MongoConnection()
user_coll = conn.getUsersCollection()
user_coll.update_one({"userId": userId}, incStr)