def push_msg(cls, msg):
'''
msg:
{
type: log|stat
msg: msg
}
'''
WSHandler.update_logs(msg)
msg = json.dumps(msg)
for client in cls.clients:
try:
client.write_message(msg)
except Exception as e:
utils.log('Error sending message: %s' % str(e))
def post(self):
global webhook_cnt, webhook_last, webhook_repo, WEBHOOKIT_CONFIGURE
# gitosc: hook, github(application/x-www-form-urlencoded): payload
data = self.get_argument('hook', self.get_argument('payload', None))
if data is None:
# gitlab and github(application/json)
data = self.request.body or None
try:
data = json.loads(data) # webhook data
# webhook configs
config = WEBHOOKIT_CONFIGURE or {}
repo_name = parser.get_repo_name(data) or ''
repo_branch = parser.get_repo_branch(data) or ''
webhook_key = '%s/%s' % (repo_name, repo_branch)
# 需要出发操作的服务器 server 数组
servers = config.get(webhook_key, [])
if servers and len(servers) > 0:
# 存在 server，需要执行 shell 脚本
cnt = 0
for s in servers:
# 遍历执行
utils.log('Starting to execute %s' % s.get('SCRIPT', ''))
utils.do_webhook_shell(s, data)
webhook_cnt += 1
cnt += 1
# 更新最后执行的时间
webhook_last = utils.current_date()
# 更新最后执行的 git 仓库
webhook_repo = webhook_key
msg = [webhook_cnt, webhook_last, webhook_repo]
WSHandler.push_msg({'type': 'stat', 'msg': msg})
t = 'Processed in thread, total %s threads.' % cnt
self.write(utils.standard_response(True, t))
else:
t = ('Not match the repo and branch '
'or the webhook servers is not exist: %s') % webhook_key
utils.log(t)
self.write(utils.standard_response(False, t))
except Exception as e:
t = ('Request trigger traceback: %s') % str(e)
utils.log(t)
utils.log('data is %s.' % (data or None))
self.write(utils.standard_response(False, t))
def webhookit():
global webhook_cnt, webhook_last, webhook_repo
data = utils.get_parameter('hook', None)
if data is None:
data = request.data
try:
data = json.loads(data) # webhook data
# webhook configs
config = flask_instance.config.get('WEBHOOKIT_CONFIGURE', None) or {}
repo_name = parser.get_repo_name(data) or ''
repo_branch = parser.get_repo_branch(data) or ''
webhook_key = '%s/%s' % (repo_name, repo_branch)
# 需要出发操作的服务器 server 数组
servers = config.get(webhook_key, [])
if servers and len(servers) > 0:
# 存在 server，需要执行 shell 脚本
cnt = 0
for s in servers:
# 遍历执行
utils.log('Starting to execute %s' % s.get('SCRIPT', ''))
utils.do_webhook_shell(s, data)
webhook_cnt += 1
cnt += 1
# 更新最后执行的时间
webhook_last = utils.current_date()
# 更新最后执行的 git 仓库
webhook_repo = webhook_key
t = 'Processed in thread, total %s threads.' % cnt
return utils.standard_response(True, t)
else:
t = ('Not match the repo and branch '
'or the webhook servers is not exist: %s') % webhook_key
utils.log(t)
return utils.standard_response(False, t)
except Exception as e:
t = ('Request trigger traceback: %s') % str(e)
utils.log(t)
utils.log(data)
return utils.standard_response(False, t)