def delete(self, user_id):
user = User.query.filter_by(id=user_id).first()
if not user:
return response.errorView('User not found', 404)
db.session.delete(user)
db.session.commit()
return response.view('User deleted', 200)
def get(self, user_id):
#notes = mongo.db.notes.find({'user_id': user_id},{'_id': 1, 'user_id': 1,'name':1})
notes = mongo.db.notes.find({'user_id': int(user_id)})
notes_list = []
for doc in notes:
encodeObjectID(doc)
notes_list.append(doc)
return response.view(notes_list, 200)
def post(self, user_id):
createData = {}
try:
createData = request.get_json()
createData["user_id"] = user_id
mongo.db.notes.insert(createData)
return response.view(encodeObjectID(createData), 200)
except Exception as ex:
return response.errorView(str(ex), 400)
def get(self):
users = User.query.all()
users_dict = []
for user in users:
users_dict.append({
'id': user.id,
'username': user.username,
'email': user.email,
'phone': user.phone,
'created_at': user.created_at.__str__()
})
return response.view(users_dict, 200)
def put(self, user_id):
user = User.query.filter_by(id=user_id).first()
if not user:
return response.errorView('User not found', 404)
args = get_request_args()
if args:
for key, value in args.items():
if not value:
continue
setattr(user, key, value)
db.session.commit()
return response.view(getUserJSONFormat(user), 201)
def get(self, user_id, note_id):
try:
notes = mongo.db.notes.find({
'user_id': int(user_id),
"_id": ObjectId(note_id)
})
notes_list = []
for doc in notes:
encodeObjectID(doc)
notes_list.append(doc)
return response.view(notes_list, 200)
except:
return response.errorView('Provided note_id is invalid', 400)
def put(self, user_id, note_id):
try:
noteItem = mongo.db.notes.find_one({'_id': ObjectId(note_id)},
{'_id': 1})
if noteItem != None:
updateValues = {"$set": request.get_json()}
mongo.db.notes.update_one(
{
'user_id': int(user_id),
"_id": ObjectId(note_id)
}, updateValues)
return response.view("Successfully updated", 200)
else:
return deleteErrorMessage()
except Exception as ex:
return customExceptionHandler(ex)
def delete(self, user_id, note_id):
try:
noteItem = mongo.db.notes.find_one(
{
'user_id': int(user_id),
'_id': ObjectId(note_id)
}, {'_id': 1})
if noteItem != None:
mongo.db.notes.delete_one({
'user_id': int(user_id),
"_id": ObjectId(note_id)
})
return response.view('Note was successfully deleted!', 200)
else:
return deleteErrorMessage()
except Exception as ex:
return customExceptionHandler(ex)
def register_user():
args = get_request_args()
if not args['username']:
return response.errorView("Username should be provided",400)
elif not args['email']:
return response.errorView("Email should be provided",400)
elif not args['password']:
return response.errorView("Password should be provided",400)
else:
user = userAuth(args['username'],
hashlib.md5(args['password'].encode('utf-8')).hexdigest(),
args['email'],
args['phone'],
datetime.datetime.now()
)
db.session.add(user)
db.session.commit()
return response.view({'status': 'Successfully created!'},200)
def migrateTables():
try:
db.create_all()
return response.view({'status': True}, 200)
except IntegrityError:
return json.dumps({'status': False})
def index():
return response.view({
'info': "Flask REST API",
}, 200)
def get(self, user_id):
user = User.query.filter_by(id=user_id).first()
if not user:
return response.errorView('User not found', 404)
data = getUserJSONFormat(user)
return response.view(data, 200)