oakscriptpy 0.1.0__tar.gz

oakscriptpy-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,26 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: astral-sh/setup-uv@v5
+      - run: uv sync --dev
+      - run: uv run pytest
+
+  publish:
+    needs: test
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write
+    steps:
+      - uses: actions/checkout@v4
+      - uses: astral-sh/setup-uv@v5
+      - run: uv build
+      - uses: pypa/gh-action-pypi-publish@release/v1

oakscriptpy-0.1.0/.gitignore

@@ -0,0 +1,7 @@
+.idea/
+__pycache__/
+*.pyc
+*.egg-info/
+dist/
+build/
+.tmp/

oakscriptpy-0.1.0/DEVLOG.md

@@ -0,0 +1,81 @@
+# DEVLOG
+
+## 2026-02-26 - Port oakscriptJS to oakscriptPy
+
+### Goal
+Port the complete oakscriptJS TypeScript library (PineScript-like API) to Python, covering all 217+ functions across 13 namespaces.
+
+### Approach Taken
+7-phase implementation following dependency order:
+1. Foundation types & utils
+2. Core computation (ta, math, array, color, str, time, matrix)
+3. Drawing objects (line, box, label, linefill, chartpoint, polyline)
+4. Series + TA-Series (lazy evaluation, operator overloading)
+5. Runtime system (context, plot, hline, inputs, adapters)
+6. Infrastructure (indicator factory, input helpers, plot helpers, zigzag library)
+7. Package integration (__init__.py, re-exports, verification)
+
+### What Worked
+- Parallel background agents for ta.py (1795 lines) and matrix.py (1136 lines) while writing smaller modules directly
+- Python operator overloading for Series (`__add__`, `__radd__`, `__gt__`, etc.) maps naturally to the JS `.add()`, `.sub()`, `.gt()` methods
+- `typing.Protocol` for adapter interfaces mirrors TS interfaces cleanly
+- Trailing underscore naming (`math_.py`, `str_.py`, `time_.py`) avoids stdlib clashes, re-exported as `math`, `str`, `time` in `__init__.py`
+- `range_`, `max_`, `min_` in ta.py avoid Python builtin clashes
+
+### Current State
+- **Done**: All 7 phases complete, 6224 lines across 28 Python files
+- **Verified**: Package imports, SMA/EMA calculations, Series operators, matrix operations, line interpolation, string functions, chart points, time functions, is_na/nz
+- **Not yet done**: Series dual-mode support in math_.py (scalar functions work, Series support deferred)
+
+### Key Decisions
+- `PineArray` = plain `list[T]` (no wrapper class)
+- `PineMatrix` = `@dataclass` with `rows`, `columns`, `data: list[list[Any]]`
+- NaN = `float('nan')`, NA = `None`, `is_na()` checks both
+- Series logical ops as methods (`and_()`, `or_()`, `not_()`) since Python reserves keywords
+- No external dependencies (pure Python)
+- Python 3.11+ required (union type syntax)
+
+## 2026-02-26 - Port Jest tests to pytest (math, color, str, new_functions)
+
+### Goal
+Port all Jest test files from oakscriptJS to Python pytest equivalents for math, color, str namespaces and new_functions.
+
+### Approach Taken
+1-to-1 file mapping from JS test files to Python test files. Adapted for Python-specific behavior differences (ValueError vs NaN for math domain errors, float formatting, Python slicing semantics, Unicode code points vs UTF-16 surrogate pairs).
+
+### What Worked
+- Direct porting of all scalar math, trig, utility, color, and string tests
+- Series-based math tests marked with `pytest.mark.skip` (56 tests)
+- Adapted JS-specific edge cases to Python equivalents (e.g., `math.sqrt(-1)` raises ValueError in Python vs returning NaN in JS)
+
+### Files Created
+- `tests/math/test_basic.py` (abs, ceil, floor, round, max, min, avg, sum)
+- `tests/math/test_algebraic.py` (sqrt, pow, exp, log, log10)
+- `tests/math/test_trigonometric.py` (sin, cos, tan, asin, acos, atan)
+- `tests/math/test_utility.py` (toradians, todegrees, random, sign)
+- `tests/math/test_series_support.py` (skipped - Series not implemented)
+- `tests/math/test_sum_series.py` (skipped - Series not implemented)
+- `tests/color/test_creation.py` (rgb, from_hex)
+- `tests/color/test_components.py` (r, g, b, t)
+- `tests/color/test_constants.py` (all named color constants)
+- `tests/color/test_manipulation.py` (new_color)
+- `tests/str/test_conversion.py` (length, tostring, tonumber)
+- `tests/str/test_formatting.py` (split, concat, format)
+- `tests/str/test_manipulation.py` (substring, upper, lower)
+- `tests/str/test_predicates.py` (startswith, endswith, char_at)
+- `tests/str/test_search.py` (contains, pos, replace)
+- `tests/str/test_whitespace.py` (trim, trim_left, trim_right, match)
+- `tests/test_new_functions.py` (color.new alias, replace_all, format_time)
+- `__init__.py` in tests/, tests/math/, tests/color/, tests/str/
+
+### Current State
+- **339 passed, 56 skipped** (all skipped are Series-based math tests)
+- No failures
+
+### Key Decisions
+- Python `math.sqrt(-1)` / `math.log(0)` raise ValueError (tests use `pytest.raises`), unlike JS which returns NaN/-Infinity
+- Python `round(0.005, 2)` = 0.0 due to IEEE 754 float representation (adapted test expectation)
+- Python `color.rgb(r,g,b,100)` produces `"rgba(..., 0.0)"` not `"rgba(..., 0)"` (float formatting)
+- Python `str.split('', '')` raises ValueError (tested as such instead of returning `[]`)
+- Python `substring` uses direct slicing (negative indices and begin>end behave differently from JS)
+- Python counts Unicode code points, not UTF-16 units (emoji = 1 char, not 2)

oakscriptpy-0.1.0/PKG-INFO

@@ -0,0 +1,120 @@
+Metadata-Version: 2.4
+Name: oakscriptpy
+Version: 0.1.0
+Summary: PineScript-like API for technical analysis in Python
+Project-URL: Homepage, https://github.com/deepentropy/oakscriptPy
+Project-URL: Repository, https://github.com/deepentropy/oakscriptPy
+Project-URL: Issues, https://github.com/deepentropy/oakscriptPy/issues
+Author: deepentropy
+License-Expression: MIT
+Keywords: finance,pinescript,ta,technical-analysis,trading
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Financial and Insurance Industry
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Office/Business :: Financial :: Investment
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+
+<p align="center">
+  <img src="./logo.png" width="150" />
+</p>
+
+<h1 align="center">OakScriptPy</h1>
+
+<p align="center">
+  PineScript-like API for technical analysis in Python. Pure Python, zero dependencies.
+</p>
+
+---
+
+## Install
+
+```bash
+pip install oakscriptpy
+```
+
+## Quick Start
+
+```python
+from oakscriptpy import ta, math, color, Series
+from oakscriptpy._types import Bar
+
+# Create bars
+bars = [
+    Bar(time=1, open=10.0, high=12.0, low=9.0, close=11.0, volume=100.0),
+    Bar(time=2, open=11.0, high=13.0, low=10.0, close=12.0, volume=110.0),
+    Bar(time=3, open=12.0, high=14.0, low=11.0, close=13.0, volume=120.0),
+    # ...
+]
+
+# Array-based TA
+closes = [b.close for b in bars]
+sma_values = ta.sma(closes, 3)
+ema_values = ta.ema(closes, 3)
+rsi_values = ta.rsi(closes, 14)
+
+# Series-based TA (with operator overloading)
+from oakscriptpy import ta as ta_series
+close_series = Series.from_bars(bars, lambda b: b.close)
+sma_series = ta_series.sma(close_series, 3)
+ema_series = ta_series.ema(close_series, 3)
+
+# Series arithmetic
+spread = close_series - sma_series
+doubled = close_series * 2
+above_sma = close_series > sma_series
+```
+
+## Namespaces
+
+| Namespace | Functions | Description |
+|-----------|-----------|-------------|
+| `ta` | 64 | Technical analysis (SMA, EMA, RSI, MACD, Bollinger Bands, Ichimoku, SuperTrend, ZigZag, etc.) |
+| `math` | 26 | Mathematical functions (abs, ceil, floor, sqrt, pow, trig, etc.) |
+| `array` | 42+ | Array manipulation (push, pop, sort, slice, binary_search, percentile, etc.) |
+| `color` | 14 + 17 constants | Color creation and manipulation (rgb, from_hex, new_color, named constants) |
+| `str` | 20+ | String functions (split, replace, substring, contains, match, etc.) |
+| `time` | 2 | Time utilities (format_time) |
+| `matrix` | 43 | Matrix operations (add, mult, det, inv, eigenvalues, eigenvectors, etc.) |
+| `line` | 10 | Line drawing objects |
+| `box` | 12 | Box drawing objects |
+| `label` | 10 | Label drawing objects |
+| `linefill` | 4 | Linefill functions |
+| `chart_point` | 3 | Chart point functions |
+| `polyline` | 5 | Polyline functions |
+
+## Series
+
+The `Series` class provides lazy evaluation with Python operator overloading:
+
+```python
+a = Series.from_bars(bars, lambda b: b.close)
+b = Series.from_bars(bars, lambda b: b.open)
+
+# Arithmetic: +, -, *, /, %
+spread = a - b
+
+# Comparison: >, >=, <, <=, eq(), neq()
+bullish = a > b
+
+# Logical: and_(), or_(), not_()
+signal = bullish.and_(a > sma_series)
+
+# History access
+prev_close = a.offset(1)
+```
+
+## Tests
+
+```bash
+uv run pytest
+# 1,224 passed, 56 skipped
+```
+
+## License
+
+MIT

oakscriptpy-0.1.0/README.md

@@ -0,0 +1,99 @@
+<p align="center">
+  <img src="./logo.png" width="150" />
+</p>
+
+<h1 align="center">OakScriptPy</h1>
+
+<p align="center">
+  PineScript-like API for technical analysis in Python. Pure Python, zero dependencies.
+</p>
+
+---
+
+## Install
+
+```bash
+pip install oakscriptpy
+```
+
+## Quick Start
+
+```python
+from oakscriptpy import ta, math, color, Series
+from oakscriptpy._types import Bar
+
+# Create bars
+bars = [
+    Bar(time=1, open=10.0, high=12.0, low=9.0, close=11.0, volume=100.0),
+    Bar(time=2, open=11.0, high=13.0, low=10.0, close=12.0, volume=110.0),
+    Bar(time=3, open=12.0, high=14.0, low=11.0, close=13.0, volume=120.0),
+    # ...
+]
+
+# Array-based TA
+closes = [b.close for b in bars]
+sma_values = ta.sma(closes, 3)
+ema_values = ta.ema(closes, 3)
+rsi_values = ta.rsi(closes, 14)
+
+# Series-based TA (with operator overloading)
+from oakscriptpy import ta as ta_series
+close_series = Series.from_bars(bars, lambda b: b.close)
+sma_series = ta_series.sma(close_series, 3)
+ema_series = ta_series.ema(close_series, 3)
+
+# Series arithmetic
+spread = close_series - sma_series
+doubled = close_series * 2
+above_sma = close_series > sma_series
+```
+
+## Namespaces
+
+| Namespace | Functions | Description |
+|-----------|-----------|-------------|
+| `ta` | 64 | Technical analysis (SMA, EMA, RSI, MACD, Bollinger Bands, Ichimoku, SuperTrend, ZigZag, etc.) |
+| `math` | 26 | Mathematical functions (abs, ceil, floor, sqrt, pow, trig, etc.) |
+| `array` | 42+ | Array manipulation (push, pop, sort, slice, binary_search, percentile, etc.) |
+| `color` | 14 + 17 constants | Color creation and manipulation (rgb, from_hex, new_color, named constants) |
+| `str` | 20+ | String functions (split, replace, substring, contains, match, etc.) |
+| `time` | 2 | Time utilities (format_time) |
+| `matrix` | 43 | Matrix operations (add, mult, det, inv, eigenvalues, eigenvectors, etc.) |
+| `line` | 10 | Line drawing objects |
+| `box` | 12 | Box drawing objects |
+| `label` | 10 | Label drawing objects |
+| `linefill` | 4 | Linefill functions |
+| `chart_point` | 3 | Chart point functions |
+| `polyline` | 5 | Polyline functions |
+
+## Series
+
+The `Series` class provides lazy evaluation with Python operator overloading:
+
+```python
+a = Series.from_bars(bars, lambda b: b.close)
+b = Series.from_bars(bars, lambda b: b.open)
+
+# Arithmetic: +, -, *, /, %
+spread = a - b
+
+# Comparison: >, >=, <, <=, eq(), neq()
+bullish = a > b
+
+# Logical: and_(), or_(), not_()
+signal = bullish.and_(a > sma_series)
+
+# History access
+prev_close = a.offset(1)
+```
+
+## Tests
+
+```bash
+uv run pytest
+# 1,224 passed, 56 skipped
+```
+
+## License
+
+MIT

oakscriptpy-0.1.0/pyproject.toml

@@ -0,0 +1,37 @@
+[project]
+name = "oakscriptpy"
+version = "0.1.0"
+description = "PineScript-like API for technical analysis in Python"
+requires-python = ">=3.11"
+dependencies = []
+readme = "README.md"
+license = "MIT"
+authors = [{ name = "deepentropy" }]
+keywords = ["pinescript", "technical-analysis", "trading", "finance", "ta"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Financial and Insurance Industry",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Office/Business :: Financial :: Investment",
+]
+
+[project.urls]
+Homepage = "https://github.com/deepentropy/oakscriptPy"
+Repository = "https://github.com/deepentropy/oakscriptPy"
+Issues = "https://github.com/deepentropy/oakscriptPy/issues"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/oakscriptpy"]
+
+[dependency-groups]
+dev = [
+    "pytest>=9.0.2",
+]

oakscriptpy-0.1.0/src/oakscriptpy/__init__.py

@@ -0,0 +1,93 @@
+"""OakScriptPy — PineScript-like API for technical analysis in Python."""
+from __future__ import annotations
+
+# Core namespaces (array-based functions)
+from . import ta as ta_core
+from . import math_ as math
+from . import array
+from . import str_ as str
+from . import color
+from . import time_ as time
+from . import matrix
+
+# Drawing object namespaces
+from . import line
+from . import box
+from . import label
+from . import linefill
+from . import chartpoint as chart_point
+from . import polyline
+
+# Series class (self-contained, no context)
+from .series import Series, BarData
+
+# TA-Series namespace (Series-based wrappers)
+from . import ta_series as ta
+
+# Types
+from ._types import (
+    Bar, Line as LineType, Box as BoxType, Label as LabelType,
+    Linefill as LinefillType, ChartPoint, Polyline as PolylineType,
+    Table, TableCell, Plot, HLine, OHLC, PineMatrix,
+    series_float, series_int, series_bool, series_string,
+    Source, color as color_type, na, xloc,
+)
+
+# Metadata types
+from ._metadata import (
+    PlotOptions, HLineOptions, FillOptions, InputMetadata,
+    PlotMetadata, IndicatorMetadata, TimeValue, PlotData,
+    HLineData, FillData, IndicatorResult,
+)
+
+# Utilities
+from ._utils import is_na, nz, ohlc_from_bars, get_close, get_high, get_low, get_open, get_source
+
+# Runtime
+from .runtime import (
+    set_context, clear_context, get_context,
+    register_calculate, recalculate,
+    plot, hline, clear_plots, get_active_plots,
+)
+from .runtime_types import OakScriptContext, OhlcvData, InputConfig
+
+# Inputs
+from .inputs import (
+    input_int, input_float, input_bool, input_string, input_source,
+    enable_auto_recalculate, disable_auto_recalculate, reset_inputs,
+)
+
+# Indicator infrastructure
+from .indicator import indicator, IndicatorMetadataConfig, IndicatorContext, IndicatorInstance
+
+# Input helper
+from . import input_ as input
+
+# Plot helper
+from .plot_ import plot as plot_helper, create_plot, TimeValuePair, PlotResult
+
+# Adapters
+from .adapters import SimpleInputAdapter
+
+# Libraries
+from .lib import ZigZag, calculate_zigzag
+
+
+def alertcondition(_condition: object = None, _title: str | None = None, _message: str | None = None) -> None:
+    """No-op stub for PineScript alertcondition() compatibility."""
+    pass
+
+
+VERSION = "0.1.0"
+
+info = {
+    "name": "OakScriptPy",
+    "version": VERSION,
+    "description": "PineScript-like API for technical analysis in Python",
+    "namespaces": {
+        "core": ["ta", "math", "array", "str", "color", "time", "matrix"],
+        "drawing": ["line", "box", "label", "linefill", "chart_point", "polyline"],
+        "runtime": ["set_context", "plot", "hline", "input_*"],
+        "indicator": ["indicator", "input", "plot_helper", "create_plot"],
+    },
+}

oakscriptpy-0.1.0/src/oakscriptpy/_metadata.py

@@ -0,0 +1,118 @@
+"""Indicator metadata types for oakscriptPy."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any
+
+
+PlotStyle = str  # 'line' | 'stepline' | 'histogram' | 'area' | 'circles' | 'columns' | 'cross' | 'areabr' | 'steplinebr' | 'linebr'
+LineStyle = str  # 'solid' | 'dashed' | 'dotted'
+PlotDisplay = str  # 'all' | 'none' | 'data_window' | 'status_line' | 'pane'
+InputType = str  # 'int' | 'float' | 'bool' | 'string' | 'source' | 'color' | 'timeframe' | 'session'
+
+
+@dataclass
+class PlotOptions:
+    title: str | None = None
+    color: str | None = None
+    linewidth: int | None = None
+    style: str | None = None
+    trackprice: bool | None = None
+    histbase: float | None = None
+    offset: int | None = None
+    join: bool | None = None
+    editable: bool | None = None
+    display: str | None = None
+    transp: int | None = None
+
+
+@dataclass
+class HLineOptions:
+    title: str | None = None
+    color: str | None = None
+    linestyle: str | None = None
+    linewidth: int | None = None
+    editable: bool | None = None
+
+
+@dataclass
+class FillOptions:
+    color: str | None = None
+    transp: int | None = None
+    title: str | None = None
+    editable: bool | None = None
+    fillgaps: bool | None = None
+
+
+@dataclass
+class InputMetadata:
+    type: str
+    name: str
+    title: str
+    defval: Any
+    minval: float | None = None
+    maxval: float | None = None
+    step: float | None = None
+    tooltip: str | None = None
+    inline: str | None = None
+    group: str | None = None
+    options: list[Any] | None = None
+    confirm: bool | None = None
+
+
+@dataclass
+class PlotMetadata:
+    var_name: str
+    title: str
+    color: str
+    linewidth: int
+    style: str
+
+
+@dataclass
+class IndicatorMetadata:
+    title: str
+    shorttitle: str | None = None
+    overlay: bool = False
+    precision: int | None = None
+    format: str | None = None
+    timeframe: str | None = None
+    timeframe_gaps: bool | None = None
+    inputs: list[InputMetadata] | None = None
+    plots: list[PlotMetadata] | None = None
+
+
+@dataclass
+class TimeValue:
+    time: Any
+    value: float
+    color: str | None = None
+
+
+@dataclass
+class PlotData:
+    data: list[TimeValue]
+    options: PlotOptions | None = None
+
+
+@dataclass
+class HLineData:
+    value: float
+    options: HLineOptions | None = None
+
+
+@dataclass
+class FillData:
+    plot1: int | str
+    plot2: int | str
+    options: FillOptions | None = None
+    colors: list[str] | None = None
+
+
+@dataclass
+class IndicatorResult:
+    metadata: IndicatorMetadata
+    plots: dict[str, list[TimeValue]]
+    hlines: list[HLineData] | None = None
+    fills: list[FillData] | None = None