Ahenk/usr/share/ahenk/plugins/network-manager/delete_host.py

71 lines
2.7 KiB
Python
Raw Permalink Normal View History

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author:Mine DOGAN <mine.dogan@agem.com.tr>
import re
from base.plugin.abstract_plugin import AbstractPlugin
class DeleteHost(AbstractPlugin):
def __init__(self, task, context):
super(DeleteHost, self).__init__()
self.task = task
self.context = context
self.logger = self.get_logger()
self.message_code = self.get_message_code()
self.hosts_file = '/etc/hosts'
self.ip = self.task['ip']
self.hostname = self.task['hostname']
self.is_active = self.task['is_active']
self.logger.debug('Parameters were initialized.')
def handle_task(self):
try:
lines = self.read_file_by_line(self.hosts_file)
f = open(self.hosts_file, "w")
for line in lines:
line = str(line).strip(" ")
# to remove multiple spaces
line = re.sub(' +', ' ', line)
if self.is_active is True:
if line != '{0} {1}\n'.format(self.ip, self.hostname):
self.logger.debug(
'Writing a line to hosts file... Line: {}'.format(line))
f.write(line)
else:
self.logger.debug(
'Line has been deleted from hosts file. Line: {}'.format(line))
else:
if line != '#{0} {1}\n'.format(self.ip, self.hostname) and line != '# {0} {1}\n'.format(self.ip,
self.hostname):
self.logger.debug(
'Writing a line to hosts file... Line: {}'.format(line))
f.write(line)
else:
self.logger.debug(
'Line has been deleted from hosts file. Line: {}'.format(line))
f.close()
self.logger.info('NETWORK-MANAGER - DELETE_HOST task is handled successfully.')
self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
message='Sunucu bilgisi başarıyla silindi.')
except Exception as e:
self.logger.error(
'A problem occured while handling NETWORK-MANAGER task: {0}'.format(str(e)))
self.context.create_response(code=self.message_code.TASK_ERROR.value,
message='NETWORK-MANAGER görevi uygulanırken bir hata oluştu.')
def handle_task(task, context):
host = DeleteHost(task, context)
host.handle_task()