mirror of
https://github.com/kuhyx/testsAndMisc-archive.git
synced 2026-07-04 17:23:09 +02:00
Krzysztof kuhy Rudnicki
2ffa1cc169
- Add _logger = logging.getLogger(__name__) to all modules - Replace logging.X() calls with _logger.X() calls - Remove logging.basicConfig() from module level (keep in run_bot()) - Add G004 to global ignores (f-strings in logging are more readable) - Remove LOG015 and G004 per-file ignores from pyproject.toml - Fix pytest_ignore_collect hook signature in conftest.py
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
"""Utility functions for the Lichess bot."""
|
|
|
|
import logging
|
|
import os
|
|
import time
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _version_file_path() -> str:
|
|
"""Return the path to the persistent bot version file.
|
|
|
|
Stored alongside this module as a simple text file containing an integer.
|
|
"""
|
|
override = os.getenv("LICHESS_BOT_VERSION_FILE")
|
|
if override:
|
|
return override
|
|
return os.path.join(os.path.dirname(__file__), ".bot_version")
|
|
|
|
|
|
def get_and_increment_version() -> int:
|
|
"""Read the current bot version, increment it, persist, and return the new version.
|
|
|
|
If the version file doesn't exist or is invalid, starts from 0, then sets to 1.
|
|
"""
|
|
path = _version_file_path()
|
|
current = 0
|
|
try:
|
|
with open(path) as f:
|
|
raw = f.read().strip()
|
|
if raw:
|
|
current = int(raw)
|
|
except (OSError, ValueError):
|
|
# Missing or unreadable file -> treat as version 0
|
|
current = 0
|
|
|
|
new_version = current + 1
|
|
try:
|
|
tmp_path = path + ".tmp"
|
|
with open(tmp_path, "w") as f:
|
|
f.write(str(new_version))
|
|
os.replace(tmp_path, path)
|
|
except OSError:
|
|
# As a fallback, try a direct write; failure is non-fatal to bot operation
|
|
try:
|
|
with open(path, "w") as f:
|
|
f.write(str(new_version))
|
|
except OSError:
|
|
_logger.debug("Could not persist bot version to %s", path)
|
|
|
|
return new_version
|
|
|
|
|
|
def backoff_sleep(current_backoff: int, base: float = 0.5, cap: float = 8.0) -> int:
|
|
"""Sleep with exponential backoff. Returns the next backoff step.
|
|
|
|
- current_backoff: number of consecutive failures
|
|
- base: base delay in seconds
|
|
- cap: maximum delay in seconds
|
|
"""
|
|
delay = min(cap, base * (2**current_backoff))
|
|
_logger.info("Backing off for %.1fs", delay)
|
|
time.sleep(delay)
|
|
return min(current_backoff + 1, 10)
|