Skip to main content

A constraint solver for generating schedules

Project description

Course Constraint Scheduler

A powerful constraint satisfaction solver for generating academic course schedules using the Z3 theorem prover.

Overview

The Course Constraint Scheduler is designed to solve complex academic scheduling problems by modeling them as constraint satisfaction problems. It can handle:

  • Faculty Constraints: Availability, credit limits, course preferences
  • Room Constraints: Room assignments, lab requirements, capacity limits
  • Time Constraints: Time slot conflicts, meeting patterns, duration requirements
  • Course Constraints: Prerequisites, conflicts, section limits
  • Optimization: Multiple optimization strategies for better schedules

Features

  • Z3 Integration: Uses Microsoft's Z3 theorem prover for efficient constraint solving
  • Flexible Configuration: JSON-based configuration with comprehensive validation
  • Multiple Output Formats: JSON and CSV output support with type-safe serialization
  • REST API: Full HTTP API for integration with web applications
  • Asynchronous Processing: Background schedule generation for large problems
  • Session Management: Persistent sessions for iterative schedule generation
  • Optimization Flags: Configurable optimization strategies
  • Type Safety: Comprehensive TypedDict definitions for all JSON structures
  • Enhanced Validation: Cross-reference validation and business logic constraints
  • Improved Error Handling: Detailed error messages for configuration debugging
  • Strict Type Validation: Pydantic models with strict validation and custom type definitions
  • Computed Fields: Automatic serialization with computed fields for clean JSON output

Quick Start

Requires a minimum version of Python 3.12

Installation

pip install course-constraint-scheduler

Command Line Usage

# Generate schedules from configuration file
scheduler example.json --limit 10 --format json --output schedules

# Interactive mode
scheduler example.json --limit 5

Python API

from scheduler import (
    CombinedConfig,
    Scheduler,
    load_config_from_file,
)

# Load configuration
config = load_config_from_file(CombinedConfig, "example.json")

# Create scheduler
scheduler = Scheduler(config)

# Generate schedules
for schedule in scheduler.get_models():
    print("Schedule:")
    for course in schedule:
        print(f"{course.as_csv()}")

REST API

# Start the server with custom options
scheduler-server --port 8000 --host 0.0.0.0 --log-level info --workers 16

# Submit a schedule request
curl -X POST "http://localhost:8000/submit" \
  -H "Content-Type: application/json" \
  -d @example.json

# Get the next schedule
curl -X POST "http://localhost:8000/schedules/{schedule_id}/next"

# Check generation progress
curl -X GET "http://localhost:8000/schedules/{schedule_id}/count"

Documentation

Configuration

The scheduler uses a JSON configuration file that defines:

  • Rooms and Labs: Available facilities and their constraints
  • Courses: Course requirements, conflicts, and faculty assignments
  • Faculty: Availability, preferences, and teaching constraints
  • Time Slots: Available time blocks and class patterns
  • Optimization: Flags for different optimization strategies

Example configuration:

{
  "config": {
    "rooms": ["Room A", "Room B"],
    "labs": ["Lab 1"],
    "courses": [
      {
        "course_id": "CS101",
        "credits": 3,
        "room": ["Room A"],
        "lab": ["Lab 1"],
        "conflicts": [],
        "faculty": ["Dr. Smith"]
      }
    ],
    "faculty": [
      {
        "name": "Dr. Smith",
        "maximum_credits": 12,
        "minimum_credits": 6,
        "unique_course_limit": 3,
        "times": {
          "MON": ["09:00-17:00"],
          "TUE": ["09:00-17:00"],
          "WED": ["09:00-17:00"],
          "THU": ["09:00-17:00"],
          "FRI": ["09:00-17:00"]
        }
      }
    ]
  },
  "time_slot_config": {
    "times": {
      "MON": [
        {
          "start": "09:00",
          "spacing": 60,
          "end": "17:00"
        }
      ]
    },
    "classes": [
      {
        "credits": 3,
        "meetings": [
          {
            "day": "MON",
            "duration": 150,
            "lab": false
          }
        ]
      }
    ]
  },
  "limit": 10,
  "optimizer_flags": ["faculty_course", "pack_rooms"]
}

Architecture

The scheduler is built with a modular architecture:

  • Core Solver: Z3-based constraint satisfaction engine
  • Configuration Management: Pydantic-based configuration validation with comprehensive error handling
  • Model Classes: Enhanced data structures for courses, faculty, and time slots with improved serialization
  • JSON Types: Comprehensive TypedDict definitions for type-safe JSON handling
  • Output Writers: JSON and CSV output formatters with context manager support
  • REST Server: FastAPI-based HTTP API with asynchronous processing and session management
  • Session Management: Persistent session handling for large problems with background task support

Performance

  • Small Problems (< 10 courses): Near-instantaneous solving
  • Medium Problems (10-50 courses): Seconds to minutes
  • Large Problems (50+ courses): May take several minutes
  • Optimization: Use appropriate optimizer flags to reduce solving time

Development

Setup

# Clone the repository
git clone <repository-url>
cd course-constraint-scheduler

# Install development dependencies
pip install -e ".[dev]"

# Run tests
pytest

# Run linting
ruff check src/

Project Structure

src/scheduler/
├── __init__.py              # Main package exports with all types
├── config.py                # Configuration models with strict validation and type definitions
├── json_types.py            # TypedDict definitions for JSON structures
├── logging.py               # Logging setup
├── main.py                  # Command-line interface
├── scheduler.py             # Core scheduling logic with Z3 integration
├── server.py                # REST API server with session management
├── time_slot_generator.py   # Utility for generating valid time slots
├── models/                  # Enhanced data models
│   ├── __init__.py          # Model exports
│   ├── course.py            # Course and instance models with computed fields
│   ├── day.py               # Day enumeration (IntEnum)
│   ├── time_slot.py         # Time-related models with comprehensive methods
│   └── identifiable.py      # Base identifiable class
└── writers/                 # Output formatters
    ├── __init__.py          # Writer exports
    ├── csv_writer.py        # CSV output with context manager support
    └── json_writer.py       # JSON output with context manager support

Contributing

  1. Fork the repository
  2. Create a feature branch
  3. Make your changes
  4. Add tests for new functionality
  5. Ensure all tests pass
  6. Submit a pull request

License

This project is licensed under the MIT License - see the LICENSE file for details.

Support

For questions, issues, or feature requests:

  • Check the documentation
  • Review existing issues
  • Create a new issue with detailed information
  • Include configuration examples and error messages

Recent Updates

Enhanced Configuration Validation

  • Comprehensive cross-reference validation ensures all IDs exist
  • Business logic validation prevents impossible constraints
  • Detailed error messages for easier debugging
  • Support for preference scores (0-10) with improved validation

Improved Type Safety

  • New json_types.py module with comprehensive TypedDict definitions
  • Type-safe JSON handling throughout the application
  • Enhanced serialization with computed fields
  • Better integration with modern Python type checking

Enhanced REST API

  • Improved session management with background task support
  • Better error handling and status codes
  • Enhanced command-line options for server configuration
  • Comprehensive API documentation with examples

Model Improvements

  • Enhanced Course and TimeSlot models with better methods
  • Improved serialization with computed fields
  • Better time handling with IntEnum for days
  • Context manager support for writers

Roadmap

  • Web-based configuration interface
  • Schedule visualization tools
  • Multi-objective optimization support

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

course_constraint_scheduler-2.2.1.tar.gz (98.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

course_constraint_scheduler-2.2.1-py3-none-any.whl (29.7 kB view details)

Uploaded Python 3

File details

Details for the file course_constraint_scheduler-2.2.1.tar.gz.

File metadata

File hashes

Hashes for course_constraint_scheduler-2.2.1.tar.gz
Algorithm Hash digest
SHA256 a8a6585a6e0e75e87bf753052b5e4ed2356019297381018833cc52f49354b07b
MD5 2072bed64d0dd6ad73230dd85e23c3ef
BLAKE2b-256 e13aedc58896be8c1efe2782347f9ff9618324a9eabb427000c5fc5f97b56bf8

See more details on using hashes here.

File details

Details for the file course_constraint_scheduler-2.2.1-py3-none-any.whl.

File metadata

File hashes

Hashes for course_constraint_scheduler-2.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 15af7f3e5775396450d3b29fb7adf36659d914da82f1ffa58fdab1f6eeac5987
MD5 718762ef0fc1376c75f69f172124dd16
BLAKE2b-256 b4cb4e7b872b4951755844dec7671996170eb24cd3dc2ff3976d1a396e47303e

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page