def __init__(self, resource, playbook_path, req_id, model):
super(MyPlaybook, self).__init__(resource)
self.results_raw = None
self.playbook_path = playbook_path
self.req = req_id
self.model = model
self.count = 0
self.session = DBSession()
def post(self):
self.set_header('Content-Type', 'application/json')
try:
print self.request.body
data = json.loads(self.request.body)
except ValueError as e:
self.set_status(400)
return self.write(
json.dumps({"message": "username and password are required!"}))
username = data.get("username", None)
password = data.get("password", None)
if username and password:
session = DBSession()
try:
user = session.query(User).filter(
User.username == username).one()
except Exception as e:
logger.error("query db failed, %s" % e.message)
user = None
self.set_status(403)
self.write(
json.dumps({"message": "user %s not exist!" % username}))
finally:
session.close()
if user:
if hashlib.sha256(password).hexdigest() == user.password:
payload = {
'id':
user.id,
'username':
username,
'exp':
datetime.datetime.utcnow() +
datetime.timedelta(seconds=43200),
}
token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')
response = {'token': token}
self.write(response)
else:
self.set_status(403)
self.write(
json.dumps({"message": 'authentication failed!'}))
else:
self.set_status(400)
self.write(
json.dumps({"message": 'username and password are required.'}))
def __init__(self, *args, **kwargs):
self.session = DBSession()
super(SSHPrivateKey, self).__init__(*args, **kwargs)
def __init__(self, *args, **kwargs):
self.session = DBSession()
super(TaskAsyncHandler, self).__init__(*args, **kwargs)
def __init__(self, *args, **kwargs):
super(SysUserHandler, self).__init__(*args, **kwargs)
self.validate_data = Dict()
self.session = DBSession()
