offlinesec-client 1.1.44__tar.gz → 1.1.45__tar.gz

{offlinesec_client-1.1.44 → offlinesec_client-1.1.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: offlinesec_client
-Version: 1.1.44
+Version: 1.1.45
 Summary: Offline Security Client
 Home-page: https://offlinesec.com
 Author: Offline Security

offlinesec_client-1.1.45/offlinesec_client/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.1.45"

{offlinesec_client-1.1.44 → offlinesec_client-1.1.45}/offlinesec_client/resolve_report.py

@@ -4,6 +4,7 @@ import os
 import argparse
 from pathlib import Path
 
+from build.lib.offlinesec_client.masking import ROLE_MASK
 from offlinesec_client.const import SUBDIR, FILE
 from offlinesec_client.masking import Masking, SAPSID_MASK, USER_MASK,HOST_MASK, PATH_MASK, RFCDEST_MASK
 import offlinesec_client.func
@@ -99,7 +100,61 @@ def do_rfc_transform(file_name, sheet_name="Parameters"):
     wb.save(fullFilename=file_name)
     wb.close()
 
+def do_roles_transform(file_name, sheet_name="Parameters"):
+    wb = editpyxl.Workbook()
+    source_filename = file_name
+
+    wb.open(source_filename)
 
+    if not sheet_name in wb.sheetnames:
+        wb.close()
+        return
+
+    ws = wb[sheet_name]
+
+    masking = {
+        SAPSID_MASK: Masking(SAPSID_MASK),
+        ROLE_MASK: Masking(ROLE_MASK)
+    }
+
+    column_map = {
+        "System Name": SAPSID_MASK,
+        "Role Name": ROLE_MASK
+    }
+
+    max_masking = 20
+    columns = list()
+    for i in range(1, max_masking + 1):
+        try:
+            column_name = ws.cell(row=1, column=i).value
+            if column_name is None or column_name.strip() == "":
+                break
+            if column_name in column_map:
+                columns.append(column_map[column_name])
+            else:
+                columns.append(None)
+                continue
+        except IndexError as err:
+            break
+
+    max_lines = 10000
+    for i, column in enumerate(columns):
+        for k in range(2, max_lines + 1):
+            if column is not None:
+                try:
+                    cell = ws.cell(row=k, column=i + 1).value
+
+                except IndexError as err:
+                    break
+                else:
+                    if cell is not None and cell.strip() != "":
+                        new_value = masking[column].do_unmask(cell)
+                        ws.cell(row=k, column=i + 1).value = new_value
+                    else:
+                        break
+
+    wb.save(fullFilename=file_name)
+    wb.close()
 
 def do_secnotes_transform(file_name):
     wb = editpyxl.Workbook()
@@ -129,35 +184,16 @@ def do_secnotes_transform(file_name):
     wb.save(fullFilename=file_name)
     wb.close()
 
-
 def read_file(file):
-    flag = False
-    if os.path.basename(file).startswith("roles_"):
-        flag = True
-    elif os.path.basename(file).startswith("rfc_") and os.path.basename(file).endswith(".xlsx"):
-        do_rfc_transform(file)
-    elif os.path.basename(file).startswith("secnotes_") and os.path.basename(file).endswith(".xlsx"):
-        do_secnotes_transform(file)
-    elif os.path.basename(file).startswith("params_") and os.path.basename(file).endswith(".xlsx"):
-        do_secnotes_transform(file)
-
-    if not flag:
-        return
-    wb = openpyxl.load_workbook(file)
-    ws = wb.worksheets[0]
-    title = ws['B1'].value
-
-    if title.startswith("Critical Authorizations"):
-        roles = read_dict_file()
-        resolve_roles_page1(ws, roles)
-        ws = wb.worksheets[1]
-        resolve_roles_page2(ws, roles)
-
-
-        wb.save(file)
-        print(" * Roles in file %s have been converted" % (os.path.basename(file),))
-
-
+    if os.path.basename(file).endswith(".xlsx"):
+        if os.path.basename(file).startswith("roles_"):
+            do_roles_transform(file)
+        elif os.path.basename(file).startswith("rfc_"):
+            do_rfc_transform(file)
+        elif os.path.basename(file).startswith("secnotes_"):
+            do_secnotes_transform(file)
+        elif os.path.basename(file).startswith("params_"):
+            do_secnotes_transform(file)
 
 def resolve_roles_page1(ws, roles):
     max_row = ws.max_row

{offlinesec_client-1.1.44 → offlinesec_client-1.1.45}/offlinesec_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: offlinesec-client
-Version: 1.1.44
+Version: 1.1.45
 Summary: Offline Security Client
 Home-page: https://offlinesec.com
 Author: Offline Security

offlinesec_client-1.1.44/offlinesec_client/__init__.py

@@ -1 +0,0 @@
-__version__ = "1.1.44"