tariff_fetch 0.1__tar.gz

tariff_fetch-0.1/.devcontainer/devcontainer.json

@@ -0,0 +1,56 @@
+{
+  "name": "tariff_fetch",
+  "image": "mcr.microsoft.com/devcontainers/base:ubuntu-24.04",
+  "features": {
+    "ghcr.io/guiyomh/features/just:0.1.0": { "version": "1.42.4" },
+    "ghcr.io/devcontainers/features/python:1": { "version": "os-provided" },
+    "ghcr.io/kreemer/features/chrometesting:1": {},
+    "ghcr.io/nils-geistmann/devcontainers-features/zsh:0": {
+      "plugins": "git colored-man-pages colorize history zsh-autosuggestions fast-syntax-highlighting zsh-autocomplete",
+      "theme": "bira"
+    }
+  },
+  "postCreateCommand": "bash -c '.devcontainer/install-ohmyzsh-plugins.sh && .devcontainer/install-python-deps.sh && .devcontainer/install-prek.sh && .devcontainer/install-prek-deps.sh'",
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "tombi-toml.tombi",
+        "nefrob.vscode-just-syntax",
+        "davidanson.vscode-markdownlint",
+        "christian-kohler.path-intellisense",
+        "ms-python.python",
+        "charliermarsh.ruff",
+        "astral-sh.ty",
+        "tekumara.typos-vscode",
+        "foxundermoon.shell-format",
+        "dprint.dprint"
+      ],
+      "settings": {
+        "python.defaultInterpreterPath": "/workspaces/tariff_fetch/.venv/bin/python",
+        "python.terminal.activateEnvironment": false,
+        "python.testing.pytestArgs": ["tests"],
+        "python.testing.pytestPath": "/workspaces/tariff_fetch/.venv/bin/pytest",
+        "python.testing.unittestEnabled": false,
+        "python.testing.pytestEnabled": true,
+        "ruff.nativeServer": "on",
+        "[python]": {
+          "editor.defaultFormatter": "charliermarsh.ruff",
+          "editor.formatOnSave": true,
+          "editor.codeActionsOnSave": {
+            "source.fixAll": "explicit",
+            "source.organizeImports": "explicit"
+          }
+        },
+        "shellformat.flag": "-i 2",
+        "[shellscript]": {
+          "editor.defaultFormatter": "foxundermoon.shell-format",
+          "editor.formatOnSave": true
+        },
+        "[markdown]": {
+          "editor.defaultFormatter": "dprint.dprint",
+          "editor.formatOnSave": true
+        }
+      }
+    }
+  }
+}

tariff_fetch-0.1/.devcontainer/install-ohmyzsh-plugins.sh

@@ -0,0 +1,98 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# install-ohmyzsh-plugins.sh - Install custom oh-my-zsh plugins
+#
+# This script clones custom oh-my-zsh plugins for the current user.
+
+echo "===================================================================="
+echo "🔌 Installing custom oh-my-zsh plugins"
+echo "===================================================================="
+echo
+
+PLUGINS_DIR="${HOME}/.oh-my-zsh/custom/plugins"
+
+# Initialize counters for installed and existing plugins
+INSTALLED_COUNT=0
+EXISTING_COUNT=0
+
+# Verify oh-my-zsh is installed
+if [[ -d "${HOME}/.oh-my-zsh" ]]; then
+  # Get oh-my-zsh version from git
+  if [[ -d "${HOME}/.oh-my-zsh/.git" ]]; then
+    # Try different methods to get version info
+    if OMZ_VERSION=$(cd "${HOME}/.oh-my-zsh" && git describe --tags 2>/dev/null); then
+      echo "✅ Found oh-my-zsh version: ${OMZ_VERSION}"
+    elif OMZ_COMMIT=$(cd "${HOME}/.oh-my-zsh" && git rev-parse --short HEAD 2>/dev/null); then
+      echo "✅ Found oh-my-zsh commit: ${OMZ_COMMIT}"
+    else
+      echo "✅ Found oh-my-zsh at ${HOME}/.oh-my-zsh"
+    fi
+  else
+    echo "✅ Found oh-my-zsh at ${HOME}/.oh-my-zsh"
+  fi
+else
+  echo "❌ ERROR: oh-my-zsh not found at ${HOME}/.oh-my-zsh" >&2
+  echo "The zsh feature must be installed first" >&2
+  exit 1
+fi
+echo
+
+# Create plugins directory if it doesn't exist
+if [[ -d "${PLUGINS_DIR}" ]]; then
+  echo "✅ Found plugins directory: ${PLUGINS_DIR}"
+else
+  echo "📁 Creating plugins directory: ${PLUGINS_DIR}"
+  mkdir -p "${PLUGINS_DIR}"
+fi
+echo
+
+# Clone zsh-autosuggestions
+if [[ -d "${PLUGINS_DIR}/zsh-autosuggestions" ]]; then
+  echo "✅ zsh-autosuggestions already installed"
+  EXISTING_COUNT=$((EXISTING_COUNT + 1))
+else
+  echo "📦 Cloning zsh-autosuggestions..."
+  git clone --quiet https://github.com/zsh-users/zsh-autosuggestions.git \
+    "${PLUGINS_DIR}/zsh-autosuggestions"
+  INSTALLED_COUNT=$((INSTALLED_COUNT + 1))
+fi
+
+# Clone fast-syntax-highlighting
+if [[ -d "${PLUGINS_DIR}/fast-syntax-highlighting" ]]; then
+  echo "✅ fast-syntax-highlighting already installed"
+  EXISTING_COUNT=$((EXISTING_COUNT + 1))
+else
+  echo "📦 Cloning fast-syntax-highlighting..."
+  git clone --quiet https://github.com/zdharma-continuum/fast-syntax-highlighting.git \
+    "${PLUGINS_DIR}/fast-syntax-highlighting"
+  INSTALLED_COUNT=$((INSTALLED_COUNT + 1))
+fi
+
+# Clone zsh-autocomplete
+if [[ -d "${PLUGINS_DIR}/zsh-autocomplete" ]]; then
+  echo "✅ zsh-autocomplete already installed"
+  EXISTING_COUNT=$((EXISTING_COUNT + 1))
+else
+  echo "📦 Cloning zsh-autocomplete..."
+  git clone --quiet --depth 1 https://github.com/marlonrichert/zsh-autocomplete.git \
+    "${PLUGINS_DIR}/zsh-autocomplete"
+  INSTALLED_COUNT=$((INSTALLED_COUNT + 1))
+fi
+
+echo
+
+# Display summary
+if [[ $INSTALLED_COUNT -gt 0 ]] && [[ $EXISTING_COUNT -gt 0 ]]; then
+  echo "✅ Installed ${INSTALLED_COUNT} plugin(s), ${EXISTING_COUNT} already installed"
+elif [[ $INSTALLED_COUNT -gt 0 ]]; then
+  echo "✅ Installed ${INSTALLED_COUNT} plugin(s)"
+else
+  echo "✅ ${EXISTING_COUNT} plugin(s) already installed"
+fi
+echo
+echo "===================================================================="
+echo "✨ oh-my-zsh plugins installed successfully!"
+echo "===================================================================="
+echo
+echo

tariff_fetch-0.1/.devcontainer/install-prek-deps.sh

@@ -0,0 +1,103 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# install-prek-deps.sh - Install prek pre-commit hooks
+#
+# Requirements:
+#   - prek must be installed
+#   - .pre-commit-config.yaml file must exist
+
+echo "===================================================================="
+echo "🪝 Installing prek pre-commit hooks from .pre-commit-config.yaml"
+echo "===================================================================="
+echo
+
+# Find the git repository root and cd there
+# This works in both devcontainers and on laptops
+REPO_ROOT=$(git rev-parse --show-toplevel 2>&1) || {
+  # Git may fail with "dubious ownership" in containers where mounted files
+  # are owned by a different user than the running process. This happens because:
+  #   - Docker Desktop on Mac maps host user UIDs to container UIDs (often UID 1000)
+  #   - The container runs as root (UID 0)
+  #   - Git sees the mismatch and refuses to operate (security feature from CVE-2022-24765)
+  # The safe.directory config tells git to trust this specific directory.
+  # We only add it when the dubious ownership error occurs, not unconditionally.
+  if echo "$REPO_ROOT" | grep -q "dubious ownership"; then
+    git config --global --add safe.directory "$(pwd)"
+    REPO_ROOT=$(git rev-parse --show-toplevel)
+  else
+    echo "❌ ERROR: Not in a git repository" >&2
+    exit 1
+  fi
+}
+cd "${REPO_ROOT}"
+echo "📁 Working in repository: ${REPO_ROOT}"
+echo
+
+# Check if .pre-commit-config.yaml exists
+PRECOMMIT_CONFIG=".pre-commit-config.yaml"
+if [ -f "${PRECOMMIT_CONFIG}" ]; then
+  echo "✅ Found .pre-commit-config.yaml: $(realpath "${PRECOMMIT_CONFIG}")"
+else
+  echo "❌ ERROR: .pre-commit-config.yaml is necessary but not found in $(pwd)" >&2
+  exit 1
+fi
+echo
+
+# Check if prek is installed
+echo "📦 Checking for prek..."
+echo
+if command -v prek >/dev/null 2>&1; then
+  PREK_VERSION=$(prek --version 2>&1)
+  echo "✅ Using: ${PREK_VERSION}"
+else
+  echo "❌ ERROR: prek is not installed" >&2
+  echo "Expected location: ${HOME}/.local/bin/prek" >&2
+  echo "" >&2
+  echo "Install prek first by running: .devcontainer/install-prek.sh" >&2
+  exit 1
+fi
+echo
+
+# Check if pre-commit hooks are already installed
+PRECOMMIT_HOOK=".git/hooks/pre-commit"
+HOOKS_ALREADY_INSTALLED=false
+
+if [ -f "${PRECOMMIT_HOOK}" ]; then
+  # Check if it's a prek-managed hook
+  if grep -q "prek" "${PRECOMMIT_HOOK}" 2>/dev/null; then
+    HOOKS_ALREADY_INSTALLED=true
+  fi
+fi
+
+# Count total hooks from .pre-commit-config.yaml (look for "- id:" pattern)
+TOTAL_HOOKS=$(grep -c "^\s*- id:" .pre-commit-config.yaml 2>/dev/null || echo "0")
+# Ensure it's a single integer
+TOTAL_HOOKS=$(echo "$TOTAL_HOOKS" | head -1 | tr -d ' ')
+
+# Install prek pre-commit hooks
+if [ "$HOOKS_ALREADY_INSTALLED" = true ]; then
+  if [ "$TOTAL_HOOKS" -gt 0 ] 2>/dev/null; then
+    echo "✅ ${TOTAL_HOOKS} pre-commit hooks already installed"
+  else
+    echo "✅ Pre-commit hooks already installed"
+  fi
+else
+  echo "📥 Installing prek pre-commit hooks..."
+  PREK_OUTPUT=$(prek install --install-hooks 2>&1)
+  echo "$PREK_OUTPUT"
+  echo
+
+  if [ "$TOTAL_HOOKS" -gt 0 ] 2>/dev/null; then
+    echo "✅ Installed ${TOTAL_HOOKS} pre-commit hooks"
+  else
+    echo "✅ Pre-commit hooks installed"
+  fi
+fi
+echo
+
+echo "===================================================================="
+echo "✨ prek pre-commit hooks installed successfully!"
+echo "===================================================================="
+echo
+echo

tariff_fetch-0.1/.devcontainer/install-prek.sh

@@ -0,0 +1,42 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# install-prek.sh - Install prek CLI for pre-commit hooks
+#
+# This script installs prek for the current user.
+# The actual hook installation (prek install --install-hooks) happens
+# after the container starts, when .git/ is mounted from the host.
+
+echo "===================================================================="
+echo "🪝 Installing prek pre-commit hook framework"
+echo "===================================================================="
+echo
+
+# Install prek
+echo "📦 Installing prek"
+curl --proto '=https' --tlsv1.2 -LsSf \
+  https://github.com/j178/prek/releases/download/v0.2.11/prek-installer.sh | sh
+echo
+
+# Add prek to PATH for this session
+# The installer adds prek to ~/.local/bin, so we add that to PATH
+export PATH="${HOME}/.local/bin:${PATH}"
+
+# Verify installation
+if ! PREK_VERSION=$(prek --version 2>&1); then
+  echo "❌ ERROR: prek installation failed or prek command not found" >&2
+  echo "Expected location: ${HOME}/.local/bin/prek" >&2
+  exit 1
+fi
+
+echo "✅ Installed: ${PREK_VERSION}"
+
+echo
+echo "💡 To install pre-commit hooks:"
+echo "   prek install --install-hooks"
+echo
+echo "===================================================================="
+echo "✨ prek installed successfully!"
+echo "===================================================================="
+echo
+echo

tariff_fetch-0.1/.devcontainer/install-python-deps.sh

@@ -0,0 +1,101 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# install-python-deps.sh - Install Python dependencies using uv
+#
+# Usage:
+#   ./install-python-deps.sh [path/to/workspace]
+#
+# This script:
+# 1. Checks if Python is installed
+# 2. Installs uv package manager
+# 3. Installs Python dependencies from pyproject.toml + uv.lock
+#
+# Requirements:
+#   - Python must be installed
+#   - pyproject.toml and uv.lock must exist in the workspace directory
+
+WORKSPACE_PATH="${1:-.}"
+
+echo "===================================================================="
+echo "🐍 Installing Python dependencies from pyproject.toml + uv.lock"
+echo "===================================================================="
+echo
+
+# Check if Python is installed
+if command -v python3 >/dev/null 2>&1; then
+  PYTHON_VERSION=$(python3 --version)
+  echo "✅ Using: ${PYTHON_VERSION}"
+else
+  echo "❌ ERROR: Python is not installed" >&2
+  echo "" >&2
+  echo "Please install Python before running this script:" >&2
+  echo "  • Ubuntu/Debian: sudo apt-get install python3" >&2
+  echo "  • macOS: brew install python3" >&2
+  exit 1
+fi
+echo
+
+# Check if pyproject.toml exists
+PYPROJECT_PATH="${WORKSPACE_PATH}/pyproject.toml"
+if [ -f "${PYPROJECT_PATH}" ]; then
+  echo "✅ Found pyproject.toml: $(realpath "${PYPROJECT_PATH}")"
+else
+  echo "❌ ERROR: pyproject.toml is necessary but not found in ${WORKSPACE_PATH}" >&2
+  exit 1
+fi
+echo
+
+# Check if uv is already installed
+# Add common uv installation paths to PATH for checking
+export PATH="${HOME}/.local/bin:${PATH}"
+
+if command -v uv >/dev/null 2>&1; then
+  UV_VERSION=$(uv --version 2>&1)
+  echo "✅ uv already installed: ${UV_VERSION}"
+else
+  # Install uv package manager
+  echo "📦 Installing uv package manager..."
+  curl -LsSf https://astral.sh/uv/install.sh | sh
+
+  # Add uv to PATH for this session so command can be used
+  # The installer adds uv to ~/.local/bin
+  export PATH="${HOME}/.local/bin:${PATH}"
+
+  # Verify installation
+  if UV_VERSION=$(uv --version 2>&1); then
+    echo "✅ Installed: ${UV_VERSION}"
+  else
+    echo "❌ ERROR: uv installation failed or uv command not found" >&2
+    echo "Expected location: ${HOME}/.local/bin/uv" >&2
+    exit 1
+  fi
+fi
+echo
+
+# Change to workspace directory
+cd "${WORKSPACE_PATH}"
+
+# Install Python dependencies
+echo "📥 Installing Python dependencies from pyproject.toml + uv.lock..."
+UV_OUTPUT=$(uv sync --group dev 2>&1)
+echo "$UV_OUTPUT"
+echo
+
+# Parse uv output to determine what happened
+if echo "$UV_OUTPUT" | grep -q "Installed [0-9]"; then
+  PACKAGE_COUNT=$(echo "$UV_OUTPUT" | grep -o "Installed [0-9]\+ package" | grep -o "[0-9]\+")
+  echo "✅ Installed ${PACKAGE_COUNT} Python packages"
+elif echo "$UV_OUTPUT" | grep -q "Audited [0-9]"; then
+  PACKAGE_COUNT=$(echo "$UV_OUTPUT" | grep -o "Audited [0-9]\+ package" | grep -o "[0-9]\+")
+  echo "✅ ${PACKAGE_COUNT} Python packages already installed"
+else
+  echo "✅ Python packages ready"
+fi
+echo
+
+echo "===================================================================="
+echo "✨ Python dependencies installed successfully!"
+echo "===================================================================="
+echo
+echo

tariff_fetch-0.1/.github/ISSUE_TEMPLATE/issue_template.yml

@@ -0,0 +1,41 @@
+name: Data science task
+description: Help describe what you're building and why
+title: "Short title that captures the 'What'"
+labels: []
+assignees: []
+body:
+  - type: textarea
+    id: what
+    attributes:
+      label: What
+      description: Describe what you're building in 1-2 clear sentences. Focus on the end result, not the implementation details.
+      placeholder: "Example: A new residential heat pump model that accounts for variable-speed compressor operation and ambient temperature derating effects."
+    validations:
+      required: true
+
+  - type: textarea
+    id: why
+    attributes:
+      label: Why
+      description: Explain the problem this solves or the value it creates. Include context about who benefits and why this matters now.
+      placeholder: "Example: Current OCHRE heat pump models assume fixed-speed operation, leading to 15-20% overestimation of energy consumption in moderate climates. This enhancement will improve accuracy for policy analyses evaluating electrification incentives and grid impact studies."
+    validations:
+      required: true
+
+  - type: textarea
+    id: how
+    attributes:
+      label: How
+      description: Outline your approach or key steps. This doesn't need to be exhaustive - just enough to show you have a plan.
+      placeholder: "Example:\n1. Implement variable-speed compressor performance curves from AHRI standards\n2. Add ambient temperature correction factors based on manufacturer data\n3. Integrate with existing OCHRE thermal model architecture\n4. Validate against lab test data and field measurements"
+    validations:
+      required: false
+
+  - type: textarea
+    id: deliverables
+    attributes:
+      label: Deliverables
+      description: List the specific outputs others can expect. Be concrete about what will be delivered.
+      placeholder: "- New HeatPumpVarSpeed class extending base OCHRE equipment\n- Performance curve parameter database\n- Unit tests with validation data\n- Documentation with physics equations and policy use cases"
+    validations:
+      required: true

tariff_fetch-0.1/.github/actions/setup-python-env/action.yml

@@ -0,0 +1,31 @@
+name: "Setup Python Environment"
+description: "Set up Python environment for the given Python version"
+
+inputs:
+  python-version:
+    description: "Python version to use"
+    required: true
+    default: "3.13"
+  uv-version:
+    description: "uv version to use"
+    required: true
+    default: "0.8.22"
+
+runs:
+  using: "composite"
+  steps:
+    - name: Install Python
+      uses: actions/setup-python@v6
+      with:
+        python-version: ${{ inputs.python-version }}
+
+    - name: Install uv
+      uses: astral-sh/setup-uv@v6
+      with:
+        version: ${{ inputs.uv-version }}
+        enable-cache: 'true'
+        cache-suffix: ${{ matrix.python-version }}
+
+    - name: Install Python dependencies
+      run: uv sync --frozen
+      shell: bash

tariff_fetch-0.1/.github/workflows/on-release-main.yml

@@ -0,0 +1,65 @@
+name: release-main
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  set-version:
+    runs-on: ubuntu-24.04
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Export tag
+        id: vars
+        run: echo tag=${GITHUB_REF#refs/tags/} >> $GITHUB_OUTPUT
+        if: ${{ github.event_name == 'release' }}
+
+      - name: Update project version
+        run: |
+          sed -i "s/^version = \".*\"/version = \"$RELEASE_VERSION\"/" pyproject.toml
+        env:
+          RELEASE_VERSION: ${{ steps.vars.outputs.tag }}
+        if: ${{ github.event_name == 'release' }}
+
+      - name: Upload updated pyproject.toml
+        uses: actions/upload-artifact@v4
+        with:
+          name: pyproject-toml
+          path: pyproject.toml
+
+  publish:
+    runs-on: ubuntu-latest
+    needs: [set-version]
+    steps:
+      - name: Check out
+        uses: actions/checkout@v4
+
+      - name: Set up the environment
+        uses: ./.github/actions/setup-python-env
+
+      - name: Download updated pyproject.toml
+        uses: actions/download-artifact@v4
+        with:
+          name: pyproject-toml
+
+      - name: Build package
+        run: uv build
+
+      - name: Publish package
+        run: uv publish
+        env:
+          UV_PUBLISH_TOKEN: ${{ secrets.PYPI_TOKEN }}
+
+  deploy-docs:
+    needs: publish
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out
+        uses: actions/checkout@v4
+
+      - name: Set up the environment
+        uses: ./.github/actions/setup-python-env
+
+      - name: Deploy documentation
+        run: uv run mkdocs gh-deploy --force

tariff_fetch-0.1/.github/workflows/python-package-main.yml

@@ -0,0 +1,72 @@
+name: Main
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    types: [opened, synchronize, reopened, ready_for_review]
+
+jobs:
+  quality:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out
+        uses: actions/checkout@v5
+
+      - name: Set up the environment
+        uses: ./.github/actions/setup-python-env
+
+      - name: Run ruff, ty and pre-commit hooks
+        uses: j178/prek-action@v1
+
+  tests-and-type-check:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+      fail-fast: false
+    defaults:
+      run:
+        shell: bash
+    steps:
+      - name: Check out
+        uses: actions/checkout@v5
+
+      - name: Set up the environment
+        uses: ./.github/actions/setup-python-env
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Run tests
+        run: uv run python -m pytest tests
+
+      - name: Check typing
+        run: uv run ty check
+
+  check-docs:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out
+        uses: actions/checkout@v5
+
+      - name: Set up the environment
+        uses: ./.github/actions/setup-python-env
+
+      - name: Check if documentation can be built
+        run: uv run mkdocs build -s
+
+  deploy-docs:
+    if: github.event_name == 'push'
+    needs: [check-docs]
+    runs-on: ubuntu-latest
+    steps:
+
+      - name: Check out
+        uses: actions/checkout@v5
+
+      - name: Set up the environment
+        uses: ./.github/actions/setup-python-env
+
+      - name: Deploy documentation
+        run: uv run mkdocs gh-deploy --force

tariff_fetch-0.1/.gitignore

@@ -0,0 +1,73 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Virtual environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+
+
+
+# IDEs
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db
+
+# Project specific
+site/
+tmp/
+main.py
+selenium_error.png

tariff_fetch-0.1/.markdownlint.json

@@ -0,0 +1,8 @@
+{
+  "// Disable rules that overlap with dprint formatter": "",
+  "MD009": false,
+  "MD010": false,
+  "MD012": false,
+  "MD013": false,
+  "MD047": false
+}