mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-12-23 08:42:18 +03:00
105 lines
4.9 KiB
Python
105 lines
4.9 KiB
Python
#!/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()
|