def add_data(self, data):
try:
user_data = self.get_file_data()
user_data.append(data)
return self.put_file_data(user_data)
except Exception as e:
raise ApplicationException({"saved-status": "failed"})
def get_file_data(self):
try:
with open(file_config.get('name'), 'r') as json_file:
json_data = json.load(json_file)
return json_data
except Exception as e:
raise ApplicationException({"fetch-status": "failed"})
def get_user_by_id(self, id):
log.info('MongoDb get_user_by_id() ')
try:
user = self.collection.find({'_id': id})
return [{item: data[item] for item in data} for data in user]
except Exception as e:
raise ApplicationException("User not found in db")
def put_file_data(self, data):
try:
with open(file_config.get('name'), 'w') as json_file:
json_file.write(json.dumps(data))
json_file.close()
return {"status": "success", "data": data}
except Exception as e:
raise ApplicationException({"saved-status": "failed"})
def get_users(self):
log.info('MongoDb get_users() ')
user_list = []
try:
response = self.collection.find()
for x in response:
user_list.append(x)
return user_list
except Exception as e:
raise ApplicationException("User data not found in db")
def get_data(self, id):
try:
user_data = self.get_file_data()
for user in user_data:
if user.get('id') == int(id):
return {"fetch-status": "user fetched", "data": user}
else:
continue
return {"fetch-status": "user not found"}
except Exception as e:
raise ApplicationException({"fetch-status": "failed"})
def delete_user_by_id(self, id):
log.info('MongoDb delete_user_by_id() ')
try:
response = self.collection.delete_one({'_id': id})
return {
'Status':
'Successfully Deleted'
if response.deleted_count > 0 else "Deleted UnSuccessful"
}
except Exception as e:
raise ApplicationException("User not delete in db")
def update_user(self, id, user):
log.info('MongoDb update_user() ')
try:
response = self.collection.update_one({'_id': id}, {"$set": user})
return {
'Status':
'Successfully Updated'
if response.modified_count > 0 else "Updated UnSuccessful"
}
except Exception as e:
raise ApplicationException("User not updated in db")
def delete_data(self, id):
try:
user_data = self.get_file_data()
for index, user in enumerate(user_data):
if user.get('id') == int(id):
del user_data[index]
self.put_file_data(user_data)
return {"delete-status": "user deleted ", "data": user}
else:
continue
return {"update-status": "user not found for update "}
except Exception as e:
raise ApplicationException({"update-status": "failed"})
def create_user(self, user):
log.info('MongoDb create_user() ')
try:
return self.collection.insert(user)
except Exception as e:
raise ApplicationException("User not created in db")