131 lines
2.7 KiB
Python
131 lines
2.7 KiB
Python
from src.functions import logger, search_mapping
|
|
|
|
|
|
def setup_black_white_lists(
|
|
blacklist_library: str,
|
|
whitelist_library: str,
|
|
blacklist_library_type: str,
|
|
whitelist_library_type: str,
|
|
blacklist_users: str,
|
|
whitelist_users: str,
|
|
library_mapping=None,
|
|
user_mapping=None,
|
|
):
|
|
blacklist_library, blacklist_library_type, blacklist_users = setup_x_lists(
|
|
blacklist_library,
|
|
blacklist_library_type,
|
|
blacklist_users,
|
|
"White",
|
|
library_mapping,
|
|
user_mapping,
|
|
)
|
|
|
|
whitelist_library, whitelist_library_type, whitelist_users = setup_x_lists(
|
|
whitelist_library,
|
|
whitelist_library_type,
|
|
whitelist_users,
|
|
"Black",
|
|
library_mapping,
|
|
user_mapping,
|
|
)
|
|
|
|
return (
|
|
blacklist_library,
|
|
whitelist_library,
|
|
blacklist_library_type,
|
|
whitelist_library_type,
|
|
blacklist_users,
|
|
whitelist_users,
|
|
)
|
|
|
|
def setup_x_lists(
|
|
xlist_library,
|
|
xlist_library_type,
|
|
xlist_users,
|
|
xlist_type,
|
|
library_mapping=None,
|
|
user_mapping=None,
|
|
):
|
|
if xlist_library:
|
|
if len(xlist_library) > 0:
|
|
xlist_library = xlist_library.split(",")
|
|
xlist_library = [x.strip() for x in xlist_library]
|
|
if library_mapping:
|
|
temp_library = []
|
|
for library in xlist_library:
|
|
library_other = search_mapping(library_mapping, library)
|
|
if library_other:
|
|
temp_library.append(library_other)
|
|
|
|
xlist_library = xlist_library + temp_library
|
|
else:
|
|
xlist_library = []
|
|
logger(f"{xlist_type}list Library: {xlist_library}", 1)
|
|
|
|
if xlist_library_type:
|
|
if len(xlist_library_type) > 0:
|
|
xlist_library_type = xlist_library_type.split(",")
|
|
xlist_library_type = [x.lower().strip() for x in xlist_library_type]
|
|
else:
|
|
xlist_library_type = []
|
|
logger(f"{xlist_type}list Library Type: {xlist_library_type}", 1)
|
|
|
|
if xlist_users:
|
|
if len(xlist_users) > 0:
|
|
xlist_users = xlist_users.split(",")
|
|
xlist_users = [x.lower().strip() for x in xlist_users]
|
|
if user_mapping:
|
|
temp_users = []
|
|
for user in xlist_users:
|
|
user_other = search_mapping(user_mapping, user)
|
|
if user_other:
|
|
temp_users.append(user_other)
|
|
|
|
xlist_users = xlist_users + temp_users
|
|
else:
|
|
xlist_users = []
|
|
else:
|
|
xlist_users = []
|
|
logger(f"{xlist_type}list Users: {xlist_users}", 1)
|
|
|
|
return xlist_library, xlist_library_type, xlist_users
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|