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

#!/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('systemctl {0} {1}'.format(action, service_name))
        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"] = 'active'

        elif ('inactive' in p_out) or ('failed' in p_out):
            service_item["serviceStatus"] = 'inactive'

        result, out, err = self.execute("systemctl is-enabled {0}".format(service_name))
        auto = 'disabled'
        if 'enabled' in out:
            auto = 'enabled'
        service_item["startAuto"] = auto
        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']) == 'start' or str(item['serviceStatus']) == 'active' 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']) == 'stop' or str(item['serviceStatus']) == 'inactive' 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']) == 'enabled' or str(item['startAuto']) == 'Start' or str(item['startAuto']) == 'START'):
                        resultcode, message = self.set_startup_service(item['serviceName'], "enable")
                        resultMessage += message
                    if item['startAuto'] is not None and (
                            str(item['startAuto']) == 'disabled' or str(item['startAuto']) == 'Stop' or str(item['startAuto']) == 'STOP'):
                        resultcode, message = self.set_startup_service(item['serviceName'], "disable")
                        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()