initial commit

ge-shutdown.iml

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$" isTestSource="false" />
+    </content>
+    <orderEntry type="jdk" jdkName="Python 3.8 (ge-shutdown)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

ge-shutdown.py

@@ -0,0 +1,116 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+
+import sys
+from typing import Dict, List, Union
+
+
+# check if user is in group and return True or False
+# example: is_user_in_group('sapadm', 'sapsys') -> True
+def is_user_in_group(user, group) -> bool:
+    from grp import getgrnam
+    try:
+        return user in getgrnam(group).gr_mem
+    except KeyError:
+        print(f'Group {group} not found')
+        exit(1)
+
+
+# build command to list all instances
+def build_command(command: str, user='sapadm') -> str:
+    from getpass import getuser
+    if user != 'sapadm' or not is_user_in_group(getuser(), 'sapsys'):
+        command = f'sudo -n -u {user} {command}'
+    return command
+
+
+# execute OS command and return output
+def execute_command(command: str, user=None) -> Dict[str, Union[str, int]]:
+    from subprocess import CalledProcessError, run, PIPE, STDOUT
+    if user:
+        command = build_command(command, user)
+    else:
+        command = build_command(command)
+
+    try:
+        output = run(command.split(), stdout=PIPE, stderr=STDOUT, timeout=30, check=True)
+        return {'output': output.stdout.decode('utf-8'), 'returncode': output.returncode}
+    except CalledProcessError as e:
+        if e.returncode == 1:
+            print(f'Command {command} failed due to invalid parameters')
+        elif e.returncode == 2:
+            print(f'Command {command} failed due to timeout')
+        elif e.returncode == 3 or e.returncode == 4:
+            return {'output': e.output.decode('utf-8'), 'returncode': e.returncode}
+        else:
+            print(f'Command {command} failed due to internal error')
+        sys.exit(e.returncode)
+    except TimeoutError:
+        print(f'Command {command} timed out')
+        sys.exit(1)
+
+
+# parse command output to get the instnance sid and instance number
+def parse_command_output(command_output: str) -> dict:
+    from re import search
+    sap_instances: dict = {}
+    for line in command_output.splitlines():
+        if not line.startswith(' Inst Info : '):
+            continue
+        instance_sid, instance_number = search(r'Inst Info : ([A-Z][A-Z\d][A-Z\d]) - (\d\d)', line).groups()
+        sap_instances.setdefault(instance_sid, {'nr': [], 'user': instance_sid.lower() + 'adm'})
+        sap_instances[instance_sid]['nr'].append(instance_number)
+    return sap_instances
+
+
+# parse SAP process list
+def parse_sap_process_list(command_output: str):
+    # 0 starttime: 2022 07 05 19:20:14
+    # 0 elapsedtime: 283:52:39
+    # 0 pid: 875626
+    from re import search
+    # sap_process_list: List[Dict[str, str]] = []
+    sap_process_list: list = []
+    for line in command_output.splitlines():
+        fields = search(r'\s*(\d+)\s+([a-z]+):\s+(.+)$', line)
+        if fields is None:
+            continue
+        idx, key, value = fields.groups()
+        if int(idx) not in range(-len(sap_process_list), len(sap_process_list)):
+            sap_process_list.append({})
+        else:
+            sap_process_list[int(idx)][key] = value
+    return sap_process_list
+
+
+# get process list for all instances
+def get_process_list_all(sap_instances) -> Dict[str, List[Dict[str, str]]]:
+    sap_instances_process_list = {}
+    for instance_sid in sap_instances:
+        for instance_number in [sap_instances[instance_sid]['nr'] for instance_sid in sap_instances]:
+            command = f'/usr/sap/hostctrl/exe/sapcontrol -format script -nr {instance_number} -function GetProcessList'
+            sap_process_list = parse_sap_process_list(
+                execute_command(command, sap_instances[instance_sid]['user'])['output'])
+            sap_instances_process_list[instance_sid] = sap_process_list
+    return sap_instances_process_list
+
+
+# stop SAP system
+def stop_sap_system() -> None:
+    pass
+
+
+def main():
+    # execute command
+    command_output = execute_command('/usr/sap/hostctrl/exe/saphostctrl -softtimeout 10,10,10 -function ListInstances')
+    # parse command output
+    sap_instances = parse_command_output(command_output['output'])
+    process_list_all = get_process_list_all(sap_instances)
+    for sid in process_list_all:
+        for idx, list_of_instances in enumerate(process_list_all[sid]):
+            print(f'{sid} -> {idx} -> {list_of_instances}')
+
+
+if __name__ == '__main__':
+    main()