Python files

Author: snehal1797

Windows_host.py

@@ -0,0 +1,74 @@
+import abc
+import socket
+import psutil
+import sys
+import json
+
+
+class HostInfo:
+    __metaclass__ = abc.ABCMeta
+
+    def __init__(self):
+        self.hostname = None
+        self.memory = None
+        self.cpu = None
+        self.ip = None
+        self.disk_size = None
+
+    @abc.abstractmethod
+    def get_hardware_info(self):
+        pass
+
+    def update_param(self, hostname, memory, cpu, ip, disk_size):
+        self.hostname = hostname
+        self.memory = memory
+        self.cpu = cpu
+        self.ip = ip
+        self.disk_size = disk_size
+
+    def display_hardware_info(self):
+        dict_data = {
+            "hostname": self.hostname,
+            "memory": self.memory,
+            "cpu": self.cpu,
+            "ip": self.ip,
+            "disk_size": self.disk_size
+        }
+
+        json_data = json.dumps(dict_data)
+        print(json_data)
+
+
+class LinuxHost(HostInfo):
+    def get_hardware_info(self):
+        hostname = socket.gethostname()
+        memory = str(psutil.virtual_memory()[0]) + " GB"
+        cpu = psutil.cpu_count()
+        ip = socket.gethostbyname(hostname)
+        disk_size = str(psutil.disk_usage("/")[0] // (2**30)) + " GB"
+        self.update_param(hostname, memory, cpu, ip, disk_size)
+
+
+class WindowsHost(HostInfo):
+    def get_hardware_info(self):
+        hostname = socket.gethostname()
+        memory = str(psutil.virtual_memory()[0]) + " GB"
+        cpu = psutil.cpu_count()
+        ip = socket.gethostbyname(hostname)
+        disk_size = str(psutil.disk_usage("/")[0] // (2 ** 30)) + " GB"
+        self.update_param(hostname, memory, cpu, ip, disk_size)
+
+
+is_windows = sys.platform.startswith('win')
+if is_windows:
+    obj = WindowsHost()
+    obj.get_hardware_info()
+    obj.display_hardware_info()
+
+
+is_linux = sys.platform.startswith('lin')
+if is_linux:
+    obj = LinuxHost()
+    obj.get_hardware_info()
+    obj.display_hardware_info()
+

json_text.py

@@ -0,0 +1,55 @@
+import json
+
+
+def get_input():
+    inp = input("Enter Memory/CPU/Linux/windows: ").upper()
+
+    pattern = ['MEMORY', 'CPU', 'LINUX', 'WINDOWS']
+
+    if inp not in pattern:
+        print("Exception: Invalid Choice")
+
+    return inp
+
+
+def get_sys_memory_cpu(inp):
+    inp = inp.lower()
+    with open('inventory.json', 'r', encoding='UTF-8') as file:
+        data = file.read()
+    data_list=[]
+    data_dict = json.loads(data)
+    for i in data_dict["inventory"]:
+        (data_list.append(i))
+
+
+
+
+
+
+
+
+
+def get_win_linux(inp):
+    inp = inp.lower().capitalize()
+    with open('inventory.json', 'r', encoding='UTF-8') as file:
+        data = file.read()
+    data_dict = json.loads(data)
+    all_sys = []
+    for key, val in data_dict.items():
+        for ind, dict_ele in enumerate(val):
+            if (ind + 1) % 2 == 0:
+                win_linux = dict_ele.get('os')
+                if inp == win_linux:
+                    all_sys.append(dict_ele)
+    return all_sys
+
+
+inp = get_input()
+
+if inp in ['MEMORY']:
+    data_dict = get_sys_memory_cpu(inp)
+    print(data_dict)
+
+if inp in ['LINUX', 'WINDOWS']:
+    all_sys = get_win_linux(inp)
+    print(all_sys)

log_parser_new.py

@@ -0,0 +1,57 @@
+import sys
+import os
+
+
+class Parser:
+    def __init__(self):
+        self.file_name = None
+        self.lines = None
+        self.filter_input = None
+        self.n_lines_data = None
+        self.filter_input_modified = []
+
+    def read_file_data(self):
+        with open(self.file_name, 'r', encoding="UTF-8") as file:
+            data = [next(file) for x in range(self.lines)]
+        self.n_lines_data = data
+
+    def get_input(self):
+        file_name = input("Enter log file name: ")
+        if not os.path.exists(file_name):
+            print("Enter Correct file-name")
+            sys.exit()
+        try:
+            num_lines = int(input("Enter number of lines you want to read: "))
+        except:
+            num_lines = 10
+        filter_input = (input("Types of log messages 'ERROR','DEBUG','INFO','WARNING': ") or "ERROR").split(",")
+
+        for i in filter_input:
+            i = i.upper().replace(" ", "")
+            if i not in ['ERROR', 'DEBUG', 'INFO', 'WARNING']:
+                print("invalid input: ", i)
+                sys.exit()
+            else:
+                self.filter_input_modified.append('[' + i + ']')
+
+        self.file_name = file_name
+        self.lines = num_lines
+        self.filter_input = filter_input
+        # print(self.filter_input_modified)
+
+    def filter_data_display(self):
+        output_list = []
+
+        for i in self.filter_input_modified:
+            for j in self.n_lines_data:
+                if i in j.split(" "):
+                    output_list.append(j)
+
+        return output_list
+
+
+obj = Parser()
+obj.get_input()
+obj.read_file_data()
+output_list = obj.filter_data_display()
+print(output_list)

os_commands.py

@@ -0,0 +1,55 @@
+import subprocess
+
+
+def get_cmd():
+    cmd = input("Enter CMD or type 'b' to break: ")
+    return cmd
+
+
+def cmd_exec(cmd):
+    p = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
+    (output, err) = p.communicate()
+    status = 'Success'
+    error = 'None'
+    if len(output) == 0:
+        output = ""
+        status = "Failed"
+        error = f"{cmd} is not recognized as an internal or external command,operable program or batch file."
+    else:
+        output = str(output)
+    return output, error, status
+
+
+res_li = []
+
+while True:
+    cmd = get_cmd()
+    if cmd.lower() == 'b':
+        break
+    try:
+        output, error, status = cmd_exec(cmd)
+    except:
+        output = ""
+        status = "Failed"
+        error = "None"
+    res_dict = {
+        cmd: {
+            "output": output,
+            "status": status,
+            "error": error
+
+        }
+    }
+
+    res_li.append(res_dict)
+key_list = []
+value_list = []
+for i in res_li:
+    for k, v in i.items():
+        if k not in key_list:
+            key_list.append(k)
+            value_list.append(v)
+dic = {}
+for i in range(len(key_list)):
+    dic[key_list[i]]=value_list[i]
+print(dic)