Ahenk/usr/share/ahenk/plugins/service/service_list.py

102 lines
4.6 KiB
Python
Raw Normal View History

#!/usr/bin/python
# -*- coding: utf-8 -*-
# Author: Cemre ALPSOY <cemre.alpsoy@agem.com.tr>
import json
from base.plugin.abstract_plugin import AbstractPlugin
class ServiceList(AbstractPlugin):
def __init__(self, data, context):
super(AbstractPlugin, self).__init__()
self.data = data
self.context = context
self.logger = self.get_logger()
self.message_code = self.get_message_code()
def start_stop_service(self, item, service_action):
service_name=str(item['serviceName'])
(result_code, p_out, p_err) = self.execute('systemctl {0} {1}'.format(service_action,service_name))
if result_code == 0:
message = 'Service:{0} , Status:{1} '.format(service_name,service_action)
else:
message = '{0} service action was unsuccessful: {0}, return code {1}'.format(service_name, service_action, str(result_code))
self.logger.debug(message)
return result_code, message, item
def set_startup_service(self, service_name, action):
(result_code, p_out, p_err) = self.execute('update-rc.d {0} {1}'.format(service_name, action))
if result_code == 0:
message = 'Service startup action was successful: {}'.format(service_name)
else:
message = 'Service action was unsuccessful: {0}, return code {1}'.format(service_name, str(result_code))
self.logger.debug(message)
return result_code, message
def get_service_status(self, service_item):
service_name=str(service_item['serviceName'])
result, p_out, err = self.execute('systemctl status {0}'.format(service_name))
if 'not-found' in p_out:
service_item["serviceStatus"] = 'Service Not Found'
elif 'running' in p_out:
service_item["serviceStatus"] = 'Running'
elif ('inactive' in p_out) or ('failed' in p_out):
service_item["serviceStatus"] = 'Stopped'
return service_item
def handle_task(self):
self.logger.debug('Handling Service Task')
try:
items = (self.data)['serviceRequestParameters']
resultMessage = ""
for item in items:
try:
if item['serviceStatus'] is not None and (
str(item['serviceStatus']) == 'Başlat' or str(item['serviceStatus']) == 'Start' or str(item['serviceStatus']) == 'START' ):
resultcode, message, item = self.start_stop_service(item, "start")
resultMessage += message
if item['serviceStatus'] is not None and (
str(item['serviceStatus']) == 'Durdur' or str(item['serviceStatus']) == 'Stop' or str(item['serviceStatus']) == 'STOP' ):
resultcode, message, item= self.start_stop_service(item, "stop")
resultMessage += message
if item['startAuto'] is not None and (
str(item['startAuto']) == 'Başlat' or str(item['startAuto']) == 'Start' or str(item['startAuto']) == 'START'):
resultcode, message = self.set_startup_service(item, "defaults")
resultMessage += message
if item['startAuto'] is not None and (
str(item['startAuto']) == 'Durdur' or str(item['startAuto']) == 'Stop' or str(item['startAuto']) == 'STOP' ):
resultcode, message = self.set_startup_service(item, "remove")
resultMessage += message
item=self.get_service_status(item)
except Exception as e:
resultMessage += '{0} servisinin isteklerini gerçekleştirirken hata ile karşılaşıldı. Hdata : {1}\r\n'.format(
str(item['serviceName']), str(e))
self.logger.debug(resultMessage)
data = {'ResultMessage': resultMessage, 'service_list': items }
self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
message='Servis istekleri gerçekleştirildi',
data=json.dumps(data),
content_type=self.get_content_type().APPLICATION_JSON.value)
except Exception as e:
self.logger.debug('Service List Exception :' + str(e))
self.context.create_response(code=self.message_code.TASK_ERROR.value,
message='Servis istekleri gerçekleştirilirken beklenmedik hata!')
def handle_task(task, context):
plugin = ServiceList(task, context)
plugin.handle_task()