Spaces:
Runtime error
Runtime error
File size: 3,299 Bytes
32f7b3e da23173 32f7b3e da23173 32f7b3e da23173 32f7b3e da23173 32f7b3e da23173 32f7b3e da23173 32f7b3e da23173 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import os
from typing import List
from urllib.parse import urljoin
from settings import (
DEFAULT_HUGGINGFACE_MODELS_API_BASE_URL,
STATIC_PATH,
)
def masked(value: str, n_shown: int, length: int = None) -> str:
"""Returns a string with the first and last n_shown characters
and the middle of the string replaced with '*'
Args:
value (str): The string to mask
n_shown (int): The number of characters to show at the beginning and end of the string
length (int, optional): The length of the string. If not given, it will be calculated as the length of the value. Defaults to None.
Returns:
str: The masked string
"""
l = length or len(value)
return value[0:n_shown] + '*'*(length-2*n_shown) + value[-n_shown:]
def ofuscated(value: str) -> str:
"""Returns a string with the first and last 4 characters
and the middle of the string replaced with '*'
Args:
value (str): The string to mask
Returns:
str: The masked string
"""
return masked(value, 4, len(value)//2)
def preview(label:str, value: str, ofuscate=False):
"""Print the variable name and its value in a nice way.
If ofuscate is True, it will ofuscate the value
Args:
variable_name (str): The name of the variable to print
ofuscate (bool, optional): If True, it will ofuscate the value. Defaults to False.
"""
str_value = ofuscated(str(value)) if ofuscate else str(value)
print(f"{label} = {str_value}")
def get_url_from_env_or_default_path(env_name: str, api_path: str) -> str:
"""Takes an url from the env variable (given the env name)
or combines with urljoin the default models base url
with the default path (given the path name)
Args:
env_name (str): The name of the environment variable to check
api_path (str): The default path to use if the environment variable is not set
Returns:
str: The url to use
"""
return os.environ.get(env_name) or urljoin(
DEFAULT_HUGGINGFACE_MODELS_API_BASE_URL, api_path
)
def get_file_as_string(file_name, path=STATIC_PATH) -> str:
"""Loads the content of a file given its name
and returns all of its lines as a single string
if a file path is given, it will be used
instead of the default static path (from settings)
Args:
file_name (_type_): The name of the file to load.
path (str, optional): The path to the file. Defaults to the current directory.
Returns:
str: The content of the file as a single string
"""
with open(os.path.join(path, file_name), mode='r', encoding='UTF-8') as f:
return f.read()
def get_sections(string: str, delimiter: str, up_to: int = None) -> List[str]:
"""Splits a string into sections given a delimiter
Args:
string (str): The string to split
delimiter (str): The delimiter to use
up_to (int, optional): The maximum number of sections to return.
Defaults to None (which means all sections)
Returns:
List[str]: The list of sections (up to the given limit, if any provided)
"""
return [section.strip()
for section in string.split(delimiter)
if (section and not section.isspace())][:up_to]
|