def verify_token(req,resp,resource,params):
admin=session.query(Admin).get(1)
token=req.get_header("token")
if not (admin and token==admin.token and admin.validate_token()):
msg={"message":"InvalidToken"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
def on_delete(self,req,resp):
data=req.params
if not data:
msg={"message":"NoData"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
else:
emp=session.query(Employee).get(int(data["id"]))
session.delete(emp)
session.commit()
msg={"message":"EmployeeDeleted"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
def on_put(self,req,resp):
data=req.media
if not data:
msg={"message":"NoData"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
else:
emp=session.query(Employee).get(int(data["id"]))
if "ed" in data: emp.ed=data["ed"]
if "role" in data: emp.role=data["role"]
session.commit()
msg={"message":"EmployeeUpdated"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
def on_get(self,req,resp):
emps=session.query(Employee).all()
if emps:
result=[]
att_list=["id","name","age","ed","role"]
for emp in emps:
result.append(dict(zip(att_list,
[emp.id,emp.name,emp.age,emp.ed,emp.role])))
resp.body=json.dumps(result)
resp.status=falcon.HTTP_OK
else:
msg={"message":"EmployeeDBEmpty"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
def on_get(self, req, resp):
data = req.params
if not data:
msg = {"message": "NoData"}
resp.body = json.dumps(msg)
resp.status = falcon.HTTP_OK
else:
admin = session.query(Admin).filter_by(
username=data["username"]).first()
if admin and admin.check_password(data['password']):
token = admin.get_token()
session.commit()
self.admin = admin
msg = {"token": token, "message": "ValidUser"}
resp.body = json.dumps(msg)
resp.status = falcon.HTTP_OK
else:
msg = {"message": "InvalidUser"}
resp.body = json.dumps(msg)
resp.status = falcon.HTTP_OK
def on_post(self,req,resp):
admin=session.query(Admin).all()
if admin:
msg={"message":"AdminExists"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
else:
data=req.media
if not data:
msg={"message":"NoData"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK
else:
un=data['username']
admin=Admin(username=un)
admin.set_password(data['password'])
session.add(admin)
session.commit()
msg={"message":"AdminAdded"}
resp.body=json.dumps(msg)
resp.status=falcon.HTTP_OK