Metadata-Version: 2.4
Name: ChessPuzzleKit
Version: 1.3.0
Summary: A Python library for retrieving and filtering chess puzzles from the Lichess database.
Author-email: Jack Scallan <breezechess99@gmail.com>
License: MIT
Project-URL: Homepage, https://github.com/JackScallan02/chess-puzzle-kit
Project-URL: Repository, https://github.com/JackScallan02/chess-puzzle-kit
Requires-Python: >=3.7
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: requests
Requires-Dist: pandas
Requires-Dist: psycopg2
Provides-Extra: dev
Requires-Dist: pytest; extra == "dev"
Requires-Dist: black; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Dynamic: license-file

# ChessPuzzleKit
[![PyPI version](https://img.shields.io/pypi/v/ChessPuzzleKit.svg)](https://pypi.org/project/ChessPuzzleKit/)

**ChessPuzzleKit** is a Python library for accessing and working with nearly 5 million chess puzzles from the [Lichess puzzle database](https://database.lichess.org/#puzzles). It provides functionality to retrieve unique puzzles by theme types, rating, popularity, and more, all with zero setup.

## Features

- Filter puzzles by:
  - Theme (e.g. `fork`, `pin`, `mateIn2`, etc.)
  - Rating or popularity range
  - Other attributes
- Automatically downloads and caches a Lichess puzzle database (almost 5 million puzzles, ~900 MB)
- Supports custom PostgreSQL or SQLite database connections

## Installation
You can install this package from [PyPI](https://pypi.org/project/ChessPuzzleKit/):
```bash
pip install ChessPuzzleKit
```

## Usage
```py
import chesspuzzlekit as cpk

# Initializes an SQLite database if none provided
cpk.initialize_connection()

# Optional: use custom database path or connection string
cpk.initialize_connection('/path/to/database')

# Get puzzles and optionally filter
puzzles = cpk.get_puzzle(themes=['fork'], ratingRange=[2000, 2200], count=3)
for p in puzzles:
    print(p['fen'], p['moves'], p['rating'])

# Retrieve all possible themes
themes = cpk.get_all_themes()
print(themes)

# Close any database connections
cpk.close_all_connections()
```

### Supported Puzzle Themes

```text
attackingF2F7      queensideAttack      kingsideAttack      middlegame
quietMove          advancedPawn         promotion           underPromotion
enPassant          interference         deflection          intermezzo
clearance          attraction           discoveredAttack    xRayAttack
skewer             fork                 pin                 doubleCheck
sacrifice          trappedPiece         hangingPiece        defensiveMove
equality           endgame              pawnEndgame         rookEndgame
bishopEndgame      knightEndgame        queenEndgame        queenRookEndgame
capturingDefender  zugzwang             mateIn1             mateIn2
mateIn3            mateIn4              mateIn5             mate
backRankMate       smotheredMate        bodenMate           anastasiaMate
doubleBishopMate   arabianMate          hookMate            killBoxMate
vukovicMate        dovetailMate         exposedKing         crushing
veryLong           long                 short               oneMove
master             superGM              masterVsMaster      advantage
opening            castling
```

### Available Puzzle Attributes

| Attribute         | Description (editable)                              |
|------------------|------------------------------------------------------|
| `GameUrl`         | Link to the full Lichess game                       |
| `FEN`             | Board state in Forsyth–Edwards Notation             |
| `Moves`           | Solution moves for the puzzle                       |
| `OpeningTags`     | Opening tags or names                               |
| `PuzzleId`        | Unique identifier for the puzzle                    |
| `NbPlays`         | Number of times the puzzle has been played          |
| `Popularity`      | Popularity score of the puzzle (-89 to 100)         |
| `Rating`          | Difficulty rating of the puzzle (339 to 3352)       |
| `RatingDeviation` | Uncertainty in the puzzle's rating                  |
| `Themes`          | List of themes (e.g., fork, pin, mateIn2)           |

### Contributions
Bug reports, bug fixes, documentation improvements, enhancements, and ideas are all welcome - please submit an issue or start a discussion in this repository. For anything else, please send inquiries to `breezechess99@gmail.com`.
