Ahenk/usr/share/ahenk/plugins/network-manager/delete_host.py
2020-04-06 18:13:10 +03:00

70 lines
2.7 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/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()