mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 15:23:03 +02:00
feat: simple lichess bot
This commit is contained in:
parent
545a30a01f
commit
5b5b069033
206
.gitignore
vendored
206
.gitignore
vendored
@ -41,4 +41,208 @@ testem.log
|
|||||||
.DS_Store
|
.DS_Store
|
||||||
Thumbs.db
|
Thumbs.db
|
||||||
*.mkv
|
*.mkv
|
||||||
imageviewer
|
imageviewer
|
||||||
|
|
||||||
|
# Byte-compiled / optimized / DLL files
|
||||||
|
__pycache__/
|
||||||
|
*.py[codz]
|
||||||
|
*$py.class
|
||||||
|
|
||||||
|
# C extensions
|
||||||
|
*.so
|
||||||
|
|
||||||
|
# Distribution / packaging
|
||||||
|
.Python
|
||||||
|
build/
|
||||||
|
develop-eggs/
|
||||||
|
dist/
|
||||||
|
downloads/
|
||||||
|
eggs/
|
||||||
|
.eggs/
|
||||||
|
lib/
|
||||||
|
lib64/
|
||||||
|
parts/
|
||||||
|
sdist/
|
||||||
|
var/
|
||||||
|
wheels/
|
||||||
|
share/python-wheels/
|
||||||
|
*.egg-info/
|
||||||
|
.installed.cfg
|
||||||
|
*.egg
|
||||||
|
MANIFEST
|
||||||
|
|
||||||
|
# PyInstaller
|
||||||
|
# Usually these files are written by a python script from a template
|
||||||
|
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||||
|
*.manifest
|
||||||
|
*.spec
|
||||||
|
|
||||||
|
# Installer logs
|
||||||
|
pip-log.txt
|
||||||
|
pip-delete-this-directory.txt
|
||||||
|
|
||||||
|
# Unit test / coverage reports
|
||||||
|
htmlcov/
|
||||||
|
.tox/
|
||||||
|
.nox/
|
||||||
|
.coverage
|
||||||
|
.coverage.*
|
||||||
|
.cache
|
||||||
|
nosetests.xml
|
||||||
|
coverage.xml
|
||||||
|
*.cover
|
||||||
|
*.py.cover
|
||||||
|
.hypothesis/
|
||||||
|
.pytest_cache/
|
||||||
|
cover/
|
||||||
|
|
||||||
|
# Translations
|
||||||
|
*.mo
|
||||||
|
*.pot
|
||||||
|
|
||||||
|
# Django stuff:
|
||||||
|
*.log
|
||||||
|
local_settings.py
|
||||||
|
db.sqlite3
|
||||||
|
db.sqlite3-journal
|
||||||
|
|
||||||
|
# Flask stuff:
|
||||||
|
instance/
|
||||||
|
.webassets-cache
|
||||||
|
|
||||||
|
# Scrapy stuff:
|
||||||
|
.scrapy
|
||||||
|
|
||||||
|
# Sphinx documentation
|
||||||
|
docs/_build/
|
||||||
|
|
||||||
|
# PyBuilder
|
||||||
|
.pybuilder/
|
||||||
|
target/
|
||||||
|
|
||||||
|
# Jupyter Notebook
|
||||||
|
.ipynb_checkpoints
|
||||||
|
|
||||||
|
# IPython
|
||||||
|
profile_default/
|
||||||
|
ipython_config.py
|
||||||
|
|
||||||
|
# pyenv
|
||||||
|
# For a library or package, you might want to ignore these files since the code is
|
||||||
|
# intended to run in multiple environments; otherwise, check them in:
|
||||||
|
# .python-version
|
||||||
|
|
||||||
|
# pipenv
|
||||||
|
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||||
|
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||||
|
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||||
|
# install all needed dependencies.
|
||||||
|
#Pipfile.lock
|
||||||
|
|
||||||
|
# UV
|
||||||
|
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
|
||||||
|
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
||||||
|
# commonly ignored for libraries.
|
||||||
|
#uv.lock
|
||||||
|
|
||||||
|
# poetry
|
||||||
|
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
|
||||||
|
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
||||||
|
# commonly ignored for libraries.
|
||||||
|
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
|
||||||
|
#poetry.lock
|
||||||
|
#poetry.toml
|
||||||
|
|
||||||
|
# pdm
|
||||||
|
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
|
||||||
|
# pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
|
||||||
|
# https://pdm-project.org/en/latest/usage/project/#working-with-version-control
|
||||||
|
#pdm.lock
|
||||||
|
#pdm.toml
|
||||||
|
.pdm-python
|
||||||
|
.pdm-build/
|
||||||
|
|
||||||
|
# pixi
|
||||||
|
# Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
|
||||||
|
#pixi.lock
|
||||||
|
# Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
|
||||||
|
# in the .venv directory. It is recommended not to include this directory in version control.
|
||||||
|
.pixi
|
||||||
|
|
||||||
|
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
|
||||||
|
__pypackages__/
|
||||||
|
|
||||||
|
# Celery stuff
|
||||||
|
celerybeat-schedule
|
||||||
|
celerybeat.pid
|
||||||
|
|
||||||
|
# SageMath parsed files
|
||||||
|
*.sage.py
|
||||||
|
|
||||||
|
# Environments
|
||||||
|
.env
|
||||||
|
.envrc
|
||||||
|
.venv
|
||||||
|
env/
|
||||||
|
venv/
|
||||||
|
ENV/
|
||||||
|
env.bak/
|
||||||
|
venv.bak/
|
||||||
|
|
||||||
|
# Spyder project settings
|
||||||
|
.spyderproject
|
||||||
|
.spyproject
|
||||||
|
|
||||||
|
# Rope project settings
|
||||||
|
.ropeproject
|
||||||
|
|
||||||
|
# mkdocs documentation
|
||||||
|
/site
|
||||||
|
|
||||||
|
# mypy
|
||||||
|
.mypy_cache/
|
||||||
|
.dmypy.json
|
||||||
|
dmypy.json
|
||||||
|
|
||||||
|
# Pyre type checker
|
||||||
|
.pyre/
|
||||||
|
|
||||||
|
# pytype static type analyzer
|
||||||
|
.pytype/
|
||||||
|
|
||||||
|
# Cython debug symbols
|
||||||
|
cython_debug/
|
||||||
|
|
||||||
|
# PyCharm
|
||||||
|
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
|
||||||
|
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
||||||
|
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
||||||
|
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||||
|
#.idea/
|
||||||
|
|
||||||
|
# Abstra
|
||||||
|
# Abstra is an AI-powered process automation framework.
|
||||||
|
# Ignore directories containing user credentials, local state, and settings.
|
||||||
|
# Learn more at https://abstra.io/docs
|
||||||
|
.abstra/
|
||||||
|
|
||||||
|
# Visual Studio Code
|
||||||
|
# Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore
|
||||||
|
# that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
|
||||||
|
# and can be added to the global gitignore or merged into this file. However, if you prefer,
|
||||||
|
# you could uncomment the following to ignore the entire vscode folder
|
||||||
|
# .vscode/
|
||||||
|
|
||||||
|
# Ruff stuff:
|
||||||
|
.ruff_cache/
|
||||||
|
|
||||||
|
# PyPI configuration file
|
||||||
|
.pypirc
|
||||||
|
|
||||||
|
# Marimo
|
||||||
|
marimo/_static/
|
||||||
|
marimo/_lsp/
|
||||||
|
__marimo__/
|
||||||
|
|
||||||
|
# Streamlit
|
||||||
|
.streamlit/secrets.toml
|
||||||
74
PYTHON/lichess_bot/README.md
Normal file
74
PYTHON/lichess_bot/README.md
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
# Lichess Bot (minimal)
|
||||||
|
|
||||||
|
A small Lichess BOT that accepts standard challenges and plays quick random legal moves using python-chess. It demonstrates the Lichess Board API basics with a simple, readable implementation.
|
||||||
|
|
||||||
|
## Features
|
||||||
|
|
||||||
|
- Connects to Lichess Board API via streaming NDJSON
|
||||||
|
- Accepts only standard chess challenges (bullet/blitz/rapid/classical)
|
||||||
|
- Spawns a thread per active game
|
||||||
|
- Plays random legal moves (swap in a stronger engine later)
|
||||||
|
- Simple logging and basic retries on transient network errors
|
||||||
|
|
||||||
|
## Requirements
|
||||||
|
|
||||||
|
- Python 3.9+
|
||||||
|
- A Lichess account that is activated as a BOT
|
||||||
|
- A Lichess API access token with at least the scopes:
|
||||||
|
- bot:play
|
||||||
|
- challenge:read
|
||||||
|
- challenge:write
|
||||||
|
|
||||||
|
Install dependencies:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pip install -r PYTHON/lichess_bot/requirements.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
## Activate BOT and get a token
|
||||||
|
|
||||||
|
1) Create or use an existing Lichess account for your bot.
|
||||||
|
2) Activate it as a BOT (one-time): https://lichess.org/api#tag/Bot
|
||||||
|
- If not already BOT, you need to convert the account; follow Lichess docs.
|
||||||
|
3) Create a personal API token: https://lichess.org/account/oauth/token/create
|
||||||
|
- Grant scopes: bot:play, challenge:read, challenge:write
|
||||||
|
|
||||||
|
Export the token in your shell (recommended):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
export LICHESS_TOKEN="your_bot_token_here"
|
||||||
|
```
|
||||||
|
|
||||||
|
## Run
|
||||||
|
|
||||||
|
From the repo root:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python -m PYTHON.lichess_bot.main
|
||||||
|
```
|
||||||
|
|
||||||
|
Optional flags:
|
||||||
|
|
||||||
|
- `--log-level INFO|DEBUG|WARNING|ERROR` (default: INFO)
|
||||||
|
- `--decline-correspondence` (declines correspondence challenges)
|
||||||
|
|
||||||
|
You can also use the helper script:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
bash PYTHON/lichess_bot/run.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
## Notes
|
||||||
|
|
||||||
|
- The engine is intentionally weak (random moves). Swap it with a UCI engine or implement a better search in `engine.py`.
|
||||||
|
- Network calls hit real Lichess endpoints. Keep the bot polite; respect rate limits.
|
||||||
|
|
||||||
|
## Development
|
||||||
|
|
||||||
|
- Small unit tests are in `tests/` and only cover local helpers (no network). Run:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python -m pytest PYTHON/lichess_bot/tests -q
|
||||||
|
```
|
||||||
|
|
||||||
|
If you add tests requiring third-party packages, install them in your environment first.
|
||||||
1
PYTHON/lichess_bot/__init__.py
Normal file
1
PYTHON/lichess_bot/__init__.py
Normal file
@ -0,0 +1 @@
|
|||||||
|
__all__ = []
|
||||||
17
PYTHON/lichess_bot/engine.py
Normal file
17
PYTHON/lichess_bot/engine.py
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
import random
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
import chess
|
||||||
|
|
||||||
|
|
||||||
|
class RandomEngine:
|
||||||
|
"""Picks a random legal move.
|
||||||
|
|
||||||
|
You can replace this with a UCI engine wrapper or a better search.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def choose_move(self, board: chess.Board) -> Optional[chess.Move]:
|
||||||
|
moves = list(board.legal_moves)
|
||||||
|
if not moves:
|
||||||
|
return None
|
||||||
|
return random.choice(moves)
|
||||||
121
PYTHON/lichess_bot/lichess_api.py
Normal file
121
PYTHON/lichess_bot/lichess_api.py
Normal file
@ -0,0 +1,121 @@
|
|||||||
|
import json
|
||||||
|
import logging
|
||||||
|
import time
|
||||||
|
from typing import Dict, Generator, Optional, Tuple
|
||||||
|
|
||||||
|
import requests
|
||||||
|
import chess
|
||||||
|
|
||||||
|
|
||||||
|
LICHESS_API = "https://lichess.org"
|
||||||
|
|
||||||
|
|
||||||
|
class LichessAPI:
|
||||||
|
def __init__(self, token: str, session: Optional[requests.Session] = None):
|
||||||
|
self.token = token
|
||||||
|
self.session = session or requests.Session()
|
||||||
|
self.session.headers.update({
|
||||||
|
"Authorization": f"Bearer {self.token}",
|
||||||
|
"Accept": "application/json",
|
||||||
|
"User-Agent": "minimal-lichess-bot/0.1 (+https://lichess.org)"
|
||||||
|
})
|
||||||
|
|
||||||
|
def stream_events(self) -> Generator[Dict, None, None]:
|
||||||
|
url = f"{LICHESS_API}/api/stream/event"
|
||||||
|
with self.session.get(url, stream=True, timeout=60) as r:
|
||||||
|
r.raise_for_status()
|
||||||
|
for line in r.iter_lines(decode_unicode=True):
|
||||||
|
if not line:
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
yield json.loads(line)
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
logging.debug(f"Skipping non-JSON line: {line}")
|
||||||
|
|
||||||
|
def accept_challenge(self, challenge_id: str) -> None:
|
||||||
|
url = f"{LICHESS_API}/api/challenge/{challenge_id}/accept"
|
||||||
|
r = self.session.post(url, timeout=30)
|
||||||
|
r.raise_for_status()
|
||||||
|
|
||||||
|
def decline_challenge(self, challenge_id: str, reason: str = "generic") -> None:
|
||||||
|
url = f"{LICHESS_API}/api/challenge/{challenge_id}/decline"
|
||||||
|
data = {"reason": reason}
|
||||||
|
r = self.session.post(url, data=data, timeout=30)
|
||||||
|
r.raise_for_status()
|
||||||
|
|
||||||
|
def join_game_stream(self, game_id: str, my_color: Optional[str]) -> Tuple[chess.Board, str]:
|
||||||
|
# Join board stream once to detect initial state and my color
|
||||||
|
url = f"{LICHESS_API}/api/board/game/stream/{game_id}"
|
||||||
|
board = chess.Board()
|
||||||
|
color = my_color or "white"
|
||||||
|
|
||||||
|
with self.session.get(url, stream=True, timeout=60) as r:
|
||||||
|
r.raise_for_status()
|
||||||
|
for line in r.iter_lines(decode_unicode=True):
|
||||||
|
if not line:
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
event = json.loads(line)
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
continue
|
||||||
|
|
||||||
|
t = event.get("type")
|
||||||
|
if t == "gameFull":
|
||||||
|
# set my color
|
||||||
|
white_id = event["white"].get("id")
|
||||||
|
black_id = event["black"].get("id")
|
||||||
|
me = self.get_my_user_id()
|
||||||
|
if me == white_id:
|
||||||
|
color = "white"
|
||||||
|
elif me == black_id:
|
||||||
|
color = "black"
|
||||||
|
|
||||||
|
# Load initial state
|
||||||
|
state = event.get("state", {})
|
||||||
|
moves = state.get("moves", "")
|
||||||
|
if moves:
|
||||||
|
for m in moves.split():
|
||||||
|
try:
|
||||||
|
board.push_uci(m)
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
break
|
||||||
|
elif t == "gameState":
|
||||||
|
# may see gameState first in rare cases; skip until gameFull
|
||||||
|
continue
|
||||||
|
return board, color
|
||||||
|
|
||||||
|
def make_move(self, game_id: str, move: chess.Move) -> None:
|
||||||
|
url = f"{LICHESS_API}/api/board/game/{game_id}/move/{move.uci()}"
|
||||||
|
r = self.session.post(url, timeout=30)
|
||||||
|
if r.status_code == 429:
|
||||||
|
time.sleep(0.5)
|
||||||
|
r = self.session.post(url, timeout=30)
|
||||||
|
r.raise_for_status()
|
||||||
|
|
||||||
|
def get_game_state(self, game_id: str) -> Optional[Dict]:
|
||||||
|
url = f"{LICHESS_API}/api/board/game/stream/{game_id}"
|
||||||
|
# Use a short-lived request to read a single line update
|
||||||
|
with self.session.get(url, stream=True, timeout=10) as r:
|
||||||
|
if r.status_code >= 400:
|
||||||
|
return None
|
||||||
|
for line in r.iter_lines(decode_unicode=True):
|
||||||
|
if not line:
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
event = json.loads(line)
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
continue
|
||||||
|
if event.get("type") == "gameState":
|
||||||
|
return event
|
||||||
|
if event.get("type") == "gameFull":
|
||||||
|
return event.get("state")
|
||||||
|
# If we get other events, keep looping; this request is short-lived anyway.
|
||||||
|
return None
|
||||||
|
|
||||||
|
def get_my_user_id(self) -> Optional[str]:
|
||||||
|
url = f"{LICHESS_API}/api/account"
|
||||||
|
r = self.session.get(url, timeout=30)
|
||||||
|
if r.status_code == 200:
|
||||||
|
return r.json().get("id")
|
||||||
|
return None
|
||||||
119
PYTHON/lichess_bot/main.py
Normal file
119
PYTHON/lichess_bot/main.py
Normal file
@ -0,0 +1,119 @@
|
|||||||
|
import argparse
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
from .engine import RandomEngine
|
||||||
|
from .lichess_api import LichessAPI
|
||||||
|
from .utils import backoff_sleep
|
||||||
|
|
||||||
|
|
||||||
|
def run_bot(log_level: str = "INFO", decline_correspondence: bool = False) -> None:
|
||||||
|
logging.basicConfig(
|
||||||
|
level=getattr(logging, log_level.upper(), logging.INFO),
|
||||||
|
format="[%(asctime)s] %(levelname)s %(threadName)s: %(message)s",
|
||||||
|
)
|
||||||
|
|
||||||
|
token = os.getenv("LICHESS_TOKEN")
|
||||||
|
if not token:
|
||||||
|
raise RuntimeError("LICHESS_TOKEN environment variable is required")
|
||||||
|
|
||||||
|
logging.info("Token present. Initializing client and engine...")
|
||||||
|
api = LichessAPI(token)
|
||||||
|
engine = RandomEngine()
|
||||||
|
|
||||||
|
game_threads = {}
|
||||||
|
|
||||||
|
def handle_game(game_id: str, my_color: Optional[str] = None):
|
||||||
|
logging.info(f"Starting game thread for {game_id}")
|
||||||
|
try:
|
||||||
|
board, color = api.join_game_stream(game_id, my_color)
|
||||||
|
logging.info(f"Game {game_id}: joined as {color}")
|
||||||
|
|
||||||
|
while True:
|
||||||
|
# If it's our turn, pick and play a move
|
||||||
|
if (board.turn and color == "white") or (not board.turn and color == "black"):
|
||||||
|
move = engine.choose_move(board)
|
||||||
|
if move is None:
|
||||||
|
logging.info(f"Game {game_id}: no legal moves (game likely over)")
|
||||||
|
break
|
||||||
|
api.make_move(game_id, move)
|
||||||
|
# Sleep briefly to avoid hammering the API
|
||||||
|
time.sleep(0.2)
|
||||||
|
|
||||||
|
# Poll for updates to keep board in sync
|
||||||
|
updates = api.get_game_state(game_id)
|
||||||
|
if updates is None:
|
||||||
|
continue
|
||||||
|
# Apply last move if present
|
||||||
|
last_move_uci = updates.get("lastMove")
|
||||||
|
if last_move_uci:
|
||||||
|
try:
|
||||||
|
board.push_uci(last_move_uci)
|
||||||
|
except Exception:
|
||||||
|
# It may already be applied; ignore
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Check for game end
|
||||||
|
if updates.get("status") in {"mate", "resign", "stalemate", "timeout", "draw"}:
|
||||||
|
logging.info(f"Game {game_id} finished: {updates.get('status')}")
|
||||||
|
break
|
||||||
|
except Exception as e:
|
||||||
|
logging.exception(f"Game {game_id} thread error: {e}")
|
||||||
|
finally:
|
||||||
|
logging.info(f"Ending game thread for {game_id}")
|
||||||
|
|
||||||
|
# Main event stream: challenge and game start events
|
||||||
|
logging.info("Connecting to Lichess event stream. Waiting for challenges...")
|
||||||
|
backoff = 0
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
for event in api.stream_events():
|
||||||
|
if event.get("type") == "challenge":
|
||||||
|
challenge = event["challenge"]
|
||||||
|
ch_id = challenge["id"]
|
||||||
|
variant = challenge.get("variant", {}).get("key", "standard")
|
||||||
|
speed = challenge.get("speed")
|
||||||
|
perf_ok = speed in {"bullet", "blitz", "rapid", "classical"}
|
||||||
|
not_corr = challenge.get("speed") != "correspondence" or not decline_correspondence
|
||||||
|
if variant == "standard" and perf_ok and not_corr:
|
||||||
|
logging.info(f"Accepting challenge {ch_id} ({speed})")
|
||||||
|
api.accept_challenge(ch_id)
|
||||||
|
else:
|
||||||
|
logging.info(f"Declining challenge {ch_id} (variant={variant}, speed={speed})")
|
||||||
|
api.decline_challenge(ch_id)
|
||||||
|
|
||||||
|
elif event.get("type") == "gameStart":
|
||||||
|
game_id = event["game"]["id"]
|
||||||
|
# Spin up a game thread
|
||||||
|
if game_id not in game_threads or not game_threads[game_id].is_alive():
|
||||||
|
t = threading.Thread(target=handle_game, args=(game_id,), name=f"game-{game_id}")
|
||||||
|
t.daemon = True
|
||||||
|
game_threads[game_id] = t
|
||||||
|
t.start()
|
||||||
|
|
||||||
|
elif event.get("type") == "gameFinish":
|
||||||
|
game_id = event["game"]["id"]
|
||||||
|
logging.info(f"Game finished event: {game_id}")
|
||||||
|
else:
|
||||||
|
logging.debug(f"Unhandled event: {json.dumps(event)}")
|
||||||
|
# If stream ends normally, reset backoff
|
||||||
|
backoff = 0
|
||||||
|
except Exception as e:
|
||||||
|
logging.warning(f"Event stream error: {e}")
|
||||||
|
backoff = backoff_sleep(backoff)
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(description="Run a minimal Lichess bot")
|
||||||
|
parser.add_argument("--log-level", default="INFO", help="Logging level (default: INFO)")
|
||||||
|
parser.add_argument("--decline-correspondence", action="store_true", help="Decline correspondence challenges")
|
||||||
|
args = parser.parse_args()
|
||||||
|
run_bot(args.log_level, args.decline_correspondence)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
6
PYTHON/lichess_bot/requirements.txt
Normal file
6
PYTHON/lichess_bot/requirements.txt
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
python-chess==1.999
|
||||||
|
requests==2.32.3
|
||||||
|
urllib3==2.2.3
|
||||||
|
certifi>=2024.2.2
|
||||||
|
chardet>=5.2.0
|
||||||
|
idna>=3.7
|
||||||
51
PYTHON/lichess_bot/run.sh
Executable file
51
PYTHON/lichess_bot/run.sh
Executable file
@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Resolve script directory and repo root
|
||||||
|
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
REPO_ROOT="$(cd -- "$SCRIPT_DIR/../.." && pwd)"
|
||||||
|
|
||||||
|
# Load a local .env if present
|
||||||
|
if [[ -f "$SCRIPT_DIR/.env" ]]; then
|
||||||
|
set -a
|
||||||
|
# shellcheck disable=SC1090
|
||||||
|
source "$SCRIPT_DIR/.env"
|
||||||
|
set +a
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Optional: --token <TOKEN> to set for this run
|
||||||
|
if [[ "${1:-}" == "--token" || "${1:-}" == "-t" ]]; then
|
||||||
|
if [[ "${2:-}" == "" ]]; then
|
||||||
|
echo "--token requires a value"
|
||||||
|
exit 2
|
||||||
|
fi
|
||||||
|
export LICHESS_TOKEN="$2"
|
||||||
|
shift 2
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Ask for token if not set and export it for this run
|
||||||
|
if [[ -z "${LICHESS_TOKEN:-}" ]]; then
|
||||||
|
printf "Paste your Lichess API token and press Enter: "
|
||||||
|
read -r token
|
||||||
|
export LICHESS_TOKEN="$token"
|
||||||
|
if [[ -z "$LICHESS_TOKEN" ]]; then
|
||||||
|
echo "No token provided. Aborting."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "Token received."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Choose python: prefer local venv
|
||||||
|
PY="$SCRIPT_DIR/.venv/bin/python"
|
||||||
|
if [[ ! -x "$PY" ]]; then
|
||||||
|
PY="python"
|
||||||
|
fi
|
||||||
|
|
||||||
|
cd "$REPO_ROOT"
|
||||||
|
echo "Using Python: $PY"
|
||||||
|
echo "Repository root: $REPO_ROOT"
|
||||||
|
echo "Starting Lichess bot..."
|
||||||
|
echo "Tip: Open another terminal to watch logs; press Ctrl+C here to stop."
|
||||||
|
|
||||||
|
trap 'echo; echo "Stopping bot (Ctrl+C)."' INT
|
||||||
|
"$PY" -m PYTHON.lichess_bot.main "$@"
|
||||||
8
PYTHON/lichess_bot/tests/conftest.py
Normal file
8
PYTHON/lichess_bot/tests/conftest.py
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
# Add repository root to sys.path so 'import PYTHON.*' works when running
|
||||||
|
# pytest with a subdirectory as rootdir.
|
||||||
|
ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
|
||||||
|
if ROOT not in sys.path:
|
||||||
|
sys.path.insert(0, ROOT)
|
||||||
10
PYTHON/lichess_bot/tests/test_engine.py
Normal file
10
PYTHON/lichess_bot/tests/test_engine.py
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
import chess
|
||||||
|
from PYTHON.lichess_bot.engine import RandomEngine
|
||||||
|
|
||||||
|
|
||||||
|
def test_random_engine_returns_move_on_start_position():
|
||||||
|
board = chess.Board()
|
||||||
|
eng = RandomEngine()
|
||||||
|
move = eng.choose_move(board)
|
||||||
|
assert move is not None
|
||||||
|
assert move in board.legal_moves
|
||||||
21
PYTHON/lichess_bot/tests/test_utils.py
Normal file
21
PYTHON/lichess_bot/tests/test_utils.py
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
import builtins
|
||||||
|
|
||||||
|
from PYTHON.lichess_bot.utils import backoff_sleep
|
||||||
|
|
||||||
|
|
||||||
|
def test_backoff_sleep_increments_and_caps(monkeypatch):
|
||||||
|
slept = []
|
||||||
|
|
||||||
|
def fake_sleep(sec):
|
||||||
|
slept.append(sec)
|
||||||
|
|
||||||
|
monkeypatch.setattr("time.sleep", fake_sleep)
|
||||||
|
|
||||||
|
b = 0
|
||||||
|
b = backoff_sleep(b, base=0.1, cap=0.3)
|
||||||
|
b = backoff_sleep(b, base=0.1, cap=0.3)
|
||||||
|
b = backoff_sleep(b, base=0.1, cap=0.3)
|
||||||
|
assert b >= 1
|
||||||
|
assert len(slept) == 3
|
||||||
|
# 0.1, 0.2, 0.3 (capped)
|
||||||
|
assert slept[0] == 0.1 and slept[1] == 0.2 and slept[2] == 0.3
|
||||||
15
PYTHON/lichess_bot/utils.py
Normal file
15
PYTHON/lichess_bot/utils.py
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
import logging
|
||||||
|
import time
|
||||||
|
|
||||||
|
|
||||||
|
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))
|
||||||
|
logging.info(f"Backing off for {delay:.1f}s")
|
||||||
|
time.sleep(delay)
|
||||||
|
return min(current_backoff + 1, 10)
|
||||||
Loading…
Reference in New Issue
Block a user