mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 16:23:04 +02:00
Krzysztof kuhy Rudnicki
728d42b230
- Extract try-except bodies into helper functions: - download_cats: _download_single_image() - randomize_numbers: _parse_single_number() - lichess_bot/main: _apply_move_to_board(), _process_event_stream(), _run_event_loop_iteration() - Use else block for return statements after try (TRY300) - Remove PERF203 from per-file ignores in pyproject.toml
112 lines
3.6 KiB
Python
112 lines
3.6 KiB
Python
"""Randomize numbers by applying a random percentage variation."""
|
|
|
|
import contextlib
|
|
import logging
|
|
import re
|
|
import secrets
|
|
import sys
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
# Use cryptographically secure random number generator
|
|
_rng = secrets.SystemRandom()
|
|
|
|
DEFAULT_MIN_PERCENTAGE = 1
|
|
DEFAULT_MAX_PERCENTAGE = 20
|
|
|
|
|
|
def randomize_numbers(
|
|
numbers: list[float],
|
|
min_percentage: float = DEFAULT_MIN_PERCENTAGE,
|
|
max_percentage: float = DEFAULT_MAX_PERCENTAGE,
|
|
) -> list[float]:
|
|
"""Apply random percentage variation to a list of numbers."""
|
|
randomized_numbers = []
|
|
for number in numbers:
|
|
percentage = _rng.uniform(min_percentage, max_percentage) / 100
|
|
if _rng.choice([True, False]):
|
|
new_number = number + (number * percentage)
|
|
else:
|
|
new_number = number - (number * percentage)
|
|
randomized_numbers.append(new_number)
|
|
return randomized_numbers
|
|
|
|
|
|
def parse_input(input_string: str) -> tuple[list[float], list[int]]:
|
|
"""Parse a string of numbers and return floats with decimal counts."""
|
|
# Replace commas with dots and remove non-numeric characters
|
|
# except dots, commas, and digits
|
|
cleaned_input = re.sub(r"[^\d.,\s]", "", input_string).replace(",", ".")
|
|
# Split the cleaned input into individual numbers
|
|
number_strings = cleaned_input.split()
|
|
# Convert the number strings to floats
|
|
numbers: list[float] = []
|
|
decimal_counts: list[int] = []
|
|
for num in number_strings:
|
|
parsed = _parse_single_number(num)
|
|
if parsed is not None:
|
|
float_num, digits_count = parsed
|
|
numbers.append(float_num)
|
|
decimal_counts.append(digits_count)
|
|
return numbers, decimal_counts
|
|
|
|
|
|
def _parse_single_number(num: str) -> tuple[float, int] | None:
|
|
"""Parse a single number string into float and decimal count.
|
|
|
|
Args:
|
|
num: The number string to parse.
|
|
|
|
Returns:
|
|
Tuple of (float value, decimal count) or None if invalid.
|
|
"""
|
|
try:
|
|
float_num = float(num)
|
|
digits_count = len(num.split(".")[-1]) if "." in num else 0
|
|
except ValueError:
|
|
return None
|
|
else:
|
|
return float_num, digits_count
|
|
|
|
|
|
MIN_ARGS = 2
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < MIN_ARGS:
|
|
logging.info(
|
|
"Usage: python random_digits.py <number1> <number2> ... "
|
|
"[min_percentage max_percentage]"
|
|
)
|
|
sys.exit(1)
|
|
|
|
input_string = " ".join(sys.argv[1:])
|
|
numbers, decimal_counts = parse_input(input_string)
|
|
|
|
if len(numbers) == 0:
|
|
logging.error("No valid numbers provided.")
|
|
sys.exit(1)
|
|
|
|
min_percentage = DEFAULT_MIN_PERCENTAGE
|
|
max_percentage = DEFAULT_MAX_PERCENTAGE
|
|
|
|
try:
|
|
if len(sys.argv) > len(numbers) + 1:
|
|
with contextlib.suppress(ValueError):
|
|
min_percentage = float(sys.argv[len(numbers) + 1])
|
|
if len(sys.argv) > len(numbers) + 2:
|
|
with contextlib.suppress(ValueError):
|
|
max_percentage = float(sys.argv[len(numbers) + 2])
|
|
|
|
randomized_numbers = randomize_numbers(numbers, min_percentage, max_percentage)
|
|
formatted_numbers = []
|
|
for i, num in enumerate(randomized_numbers):
|
|
format_str = f".{decimal_counts[i]}f"
|
|
formatted_numbers.append(float(format(num, format_str)))
|
|
|
|
logging.info(f"Original numbers: {numbers}")
|
|
logging.info(f"Randomized numbers: {formatted_numbers}")
|
|
except ValueError:
|
|
logging.exception("Error processing numbers")
|
|
logging.exception("Please provide valid numbers and percentages.")
|
|
sys.exit(1)
|