Post Black Linting
This commit is contained in:
+57
-22
@@ -28,8 +28,6 @@ import pandas as pd
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
|
||||
|
||||
def import_to_dataframe(file_path: str) -> pd.DataFrame:
|
||||
df = pd.DataFrame()
|
||||
|
||||
@@ -96,17 +94,17 @@ def choose_file(initial_directory=None, required_substring=None):
|
||||
return file_path
|
||||
|
||||
|
||||
|
||||
|
||||
def get_sanitized_input(prompt: str) -> str:
|
||||
while True:
|
||||
user_input = input(prompt)
|
||||
if user_input.strip() == "":
|
||||
return user_input # Allow blank lines
|
||||
if re.match(r'^[a-zA-Z0-9_\- .]+$', user_input.strip()):
|
||||
if re.match(r"^[a-zA-Z0-9_\- .]+$", user_input.strip()):
|
||||
return user_input
|
||||
else:
|
||||
print("Invalid input. Only letters, numbers, underscores, spaces, hyphens, and periods are allowed.")
|
||||
print(
|
||||
"Invalid input. Only letters, numbers, underscores, spaces, hyphens, and periods are allowed."
|
||||
)
|
||||
|
||||
|
||||
def regulator(paths, case_insensitive=True):
|
||||
@@ -119,8 +117,10 @@ def regulator(paths, case_insensitive=True):
|
||||
pattern = "(?i)" + pattern # Add inline case-insensitive flag
|
||||
print(f"Regulator is providing: {pattern}")
|
||||
return pattern
|
||||
|
||||
|
||||
def irtang():
|
||||
print(
|
||||
print(
|
||||
colorText(
|
||||
r"""
|
||||
███
|
||||
@@ -149,6 +149,8 @@ def irtang():
|
||||
"yellow",
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def displayIntro():
|
||||
|
||||
print(
|
||||
@@ -164,6 +166,8 @@ def displayIntro():
|
||||
"cyan",
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def welcome():
|
||||
print(
|
||||
colorText(
|
||||
@@ -184,11 +188,21 @@ def welcome():
|
||||
)
|
||||
)
|
||||
|
||||
def section_header(title):
|
||||
print(colorText("\n --------------------------------------------------------------------", "cyan"))
|
||||
print(colorText(f" ------------- {title} -------------", "cyan"))
|
||||
print(colorText(" --------------------------------------------------------------------", "cyan"))
|
||||
|
||||
def section_header(title):
|
||||
print(
|
||||
colorText(
|
||||
"\n --------------------------------------------------------------------",
|
||||
"cyan",
|
||||
)
|
||||
)
|
||||
print(colorText(f" ------------- {title} -------------", "cyan"))
|
||||
print(
|
||||
colorText(
|
||||
" --------------------------------------------------------------------",
|
||||
"cyan",
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def areYouSure():
|
||||
@@ -348,7 +362,11 @@ def printDeviceEnforceChecklist():
|
||||
"cyan",
|
||||
)
|
||||
)
|
||||
print(colorText(" When complete, save the csv file to the directory 'approved'", "cyan"))
|
||||
print(
|
||||
colorText(
|
||||
" When complete, save the csv file to the directory 'approved'", "cyan"
|
||||
)
|
||||
)
|
||||
print(
|
||||
colorText(
|
||||
" Do the same process with the list of publishers forthe same directories",
|
||||
@@ -357,20 +375,38 @@ def printDeviceEnforceChecklist():
|
||||
)
|
||||
print(colorText(" Preflight Lists will be generated", "cyan"))
|
||||
|
||||
print(colorText("5. Choose the destination policy and parent and child allow list", "cyan"))
|
||||
print(
|
||||
colorText(
|
||||
"5. Choose the destination policy and parent and child allow list", "cyan"
|
||||
)
|
||||
)
|
||||
|
||||
print(colorText("6. Test ------------------------------------------------------", "cyan"))
|
||||
print(
|
||||
colorText(
|
||||
"6. Test ------------------------------------------------------", "cyan"
|
||||
)
|
||||
)
|
||||
print(colorText(" Print rather than apply selected data.", "cyan"))
|
||||
|
||||
print(colorText("7. Liftoff ------------------------------------------------------", "cyan"))
|
||||
print(
|
||||
colorText(
|
||||
"7. Liftoff ------------------------------------------------------", "cyan"
|
||||
)
|
||||
)
|
||||
print(
|
||||
colorText(
|
||||
" Apply path exclusions according to allowed and approved paths",
|
||||
"cyan",
|
||||
)
|
||||
)
|
||||
print(colorText(" Apply signed or attested hashes to Parent Allow List", "cyan"))
|
||||
print(colorText(" Apply approved, but unsigned hashes to the Child Allow List", "cyan"))
|
||||
print(
|
||||
colorText(" Apply signed or attested hashes to Parent Allow List", "cyan")
|
||||
)
|
||||
print(
|
||||
colorText(
|
||||
" Apply approved, but unsigned hashes to the Child Allow List", "cyan"
|
||||
)
|
||||
)
|
||||
|
||||
print(
|
||||
colorText(
|
||||
@@ -523,10 +559,9 @@ def formatHTML(df, output_html_path=None, overwrite=True):
|
||||
return styled_html
|
||||
|
||||
|
||||
|
||||
def open_directory(path):
|
||||
system = platform.system()
|
||||
|
||||
|
||||
if system == "Windows":
|
||||
os.startfile(path)
|
||||
elif system == "Linux":
|
||||
@@ -537,9 +572,9 @@ def open_directory(path):
|
||||
|
||||
def print_x_wide(items: list, width: int):
|
||||
for i in range(0, len(items), width):
|
||||
row = items[i:i+width]
|
||||
row = items[i : i + width]
|
||||
print(" | ".join(row))
|
||||
|
||||
|
||||
|
||||
def clear_screen():
|
||||
os.system('cls' if os.name == 'nt' else 'clear')
|
||||
os.system("cls" if os.name == "nt" else "clear")
|
||||
|
||||
Reference in New Issue
Block a user