mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 06:04:21 +01:00
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
import re
|
|
import os
|
|
import logging
|
|
from common.constants import LOCATOR_LEN_HEX
|
|
|
|
|
|
def check_sha256_hex_format(value):
|
|
"""
|
|
Checks if a given value is a 32-byte hex encoded string.
|
|
|
|
Args:
|
|
value(:mod:`str`): the value to be checked.
|
|
|
|
Returns:
|
|
:mod:`bool`: Whether or not the value matches the format.
|
|
"""
|
|
return isinstance(value, str) and re.match(r"^[0-9A-Fa-f]{64}$", value) is not None
|
|
|
|
|
|
def check_locator_format(value):
|
|
"""
|
|
Checks if a given value is a 16-byte hex encoded string.
|
|
|
|
Args:
|
|
value(:mod:`str`): the value to be checked.
|
|
|
|
Returns:
|
|
:mod:`bool`: Whether or not the value matches the format.
|
|
"""
|
|
return isinstance(value, str) and re.match(r"^[0-9A-Fa-f]{32}$", value) is not None
|
|
|
|
|
|
def compute_locator(tx_id):
|
|
"""
|
|
Computes an appointment locator given a transaction id.
|
|
Args:
|
|
tx_id (:obj:`str`): the transaction id used to compute the locator.
|
|
Returns:
|
|
(:obj:`str`): The computed locator.
|
|
"""
|
|
|
|
return tx_id[:LOCATOR_LEN_HEX]
|
|
|
|
|
|
def setup_data_folder(data_folder, logger):
|
|
if not os.path.isdir(data_folder):
|
|
logger.info("Data folder not found. Creating it")
|
|
os.makedirs(data_folder, exist_ok=True)
|
|
|
|
|
|
def check_conf_fields(conf_fields):
|
|
conf_dict = {}
|
|
|
|
for field in conf_fields:
|
|
value = conf_fields[field]["value"]
|
|
correct_type = conf_fields[field]["type"]
|
|
|
|
if (value is not None) and isinstance(value, correct_type):
|
|
conf_dict[field] = value
|
|
else:
|
|
err_msg = "{} variable in config is of the wrong type".format(field)
|
|
raise ValueError(err_msg)
|
|
|
|
return conf_dict
|
|
|
|
|
|
def extend_paths(base_path, config_fields):
|
|
for key, field in config_fields.items():
|
|
if field.get("path"):
|
|
config_fields[key]["value"] = base_path + config_fields[key]["value"]
|
|
|
|
return config_fields
|
|
|
|
|
|
def setup_logging(log_file_path, log_name_prefix):
|
|
if not isinstance(log_file_path, str):
|
|
print(log_file_path)
|
|
raise ValueError("Wrong log file path.")
|
|
|
|
if not isinstance(log_name_prefix, str):
|
|
raise ValueError("Wrong log file name.")
|
|
|
|
# Create the file logger
|
|
f_logger = logging.getLogger("{}_file_log".format(log_name_prefix))
|
|
f_logger.setLevel(logging.INFO)
|
|
|
|
fh = logging.FileHandler(log_file_path)
|
|
fh.setLevel(logging.INFO)
|
|
fh_formatter = logging.Formatter("%(message)s")
|
|
fh.setFormatter(fh_formatter)
|
|
f_logger.addHandler(fh)
|
|
|
|
# Create the console logger
|
|
c_logger = logging.getLogger("{}_console_log".format(log_name_prefix))
|
|
c_logger.setLevel(logging.INFO)
|
|
|
|
ch = logging.StreamHandler()
|
|
ch.setLevel(logging.INFO)
|
|
ch_formatter = logging.Formatter("%(message)s.", "%Y-%m-%d %H:%M:%S")
|
|
ch.setFormatter(ch_formatter)
|
|
c_logger.addHandler(ch)
|