tidy3 0.3.0__py3-none-any.whl

tidy3/__init__.py

@@ -0,0 +1,407 @@
+"""tidy3 — dplyr-style lazy data manipulation on Polars."""
+
+from __future__ import annotations
+
+from tidy3.expr import (
+    all,
+    any,
+    between_values,
+    case_when,
+    case_match,
+    coalesce,
+    col,
+    consecutive_id,
+    cumall,
+    cumany,
+    cume_dist,
+    cummean,
+    dense_rank,
+    desc,
+    first,
+    if_else,
+    lag,
+    last,
+    lead,
+    max,
+    mean,
+    median,
+    min,
+    min_rank,
+    na_if,
+    n,
+    n_distinct,
+    near,
+    nth,
+    ntile,
+    percent_rank,
+    row_number,
+    recode,
+    sd,
+    std,
+    var,
+)
+from tidy3.expr import sum as sum  # noqa: A001
+from tidy3.eda import colnames, describe, dim, dtypes, names, ncol, nrow, summary
+from tidy3.frame import TidyFrame, options, tidy
+from tidy3.io import scan_csv, scan_ipc, scan_parquet
+from tidy3.join_spec import (
+    JoinCondition,
+    JoinSpec,
+    between,
+    closest,
+    eq,
+    ge,
+    gt,
+    join_by,
+    le,
+    lt,
+    overlaps,
+    within,
+)
+from tidy3.export import (
+    ExportResult,
+    collect_known_names,
+    nb_export,
+    transform_cell,
+    transform_source,
+)
+from tidy3.partial_run import (
+    looks_like_tidy_pipe,
+    maybe_rewrite_cell,
+    normalize_pipe_source,
+    partial_run,
+)
+from tidy3.tidyselect import (
+    Selector,
+    across,
+    all_of,
+    any_of,
+    c_across,
+    col_range,
+    cols_between,
+    contains,
+    cur_column,
+    cur_group,
+    cur_group_id,
+    cur_group_rows,
+    ends_with,
+    everything,
+    group_cols,
+    group_vars,
+    n_groups,
+    if_all,
+    if_any,
+    is_bool,
+    is_boolean,
+    is_categorical,
+    is_character,
+    is_datetime,
+    is_float,
+    is_integer,
+    is_numeric,
+    is_string,
+    is_temporal,
+    last_col,
+    matches,
+    num_range,
+    pick,
+    starts_with,
+    where,
+)
+from tidy3.tidyr import (
+    complete,
+    drop_na,
+    expand,
+    fill,
+    hoist,
+    nest,
+    nesting,
+    pivot_longer,
+    pivot_wider,
+    replace_na,
+    separate,
+    separate_longer_delim,
+    separate_wider_delim,
+    unite,
+    unnest,
+    unnest_longer,
+    unnest_wider,
+    pack,
+    unpack,
+)
+from tidy3.verbs import (
+    add_count,
+    add_tally,
+    anti_join,
+    arrange,
+    bind_cols,
+    bind_rows,
+    collect,
+    count,
+    cross_join,
+    distinct,
+    drop,
+    filter,
+    filter_out,
+    full_join,
+    glimpse,
+    group_by,
+    group_map,
+    group_modify,
+    group_nest,
+    group_split,
+    head,
+    inner_join,
+    intersect,
+    left_join,
+    mutate,
+    nest_join,
+    peek,
+    pull,
+    relocate,
+    reframe,
+    rename,
+    rename_with,
+    clean_names,
+    make_clean_names,
+    set_names,
+    right_join,
+    rowwise,
+    rows_append,
+    rows_delete,
+    rows_insert,
+    rows_patch,
+    rows_update,
+    rows_upsert,
+    sample_frac,
+    sample_n,
+    select,
+    semi_join,
+    setdiff,
+    setequal,
+    slice,
+    slice_head,
+    slice_max,
+    slice_min,
+    slice_sample,
+    slice_tail,
+    summarise,
+    summarize,
+    symdiff,
+    tally,
+    transmute,
+    to_numpy,
+    ungroup,
+    union,
+    union_all,
+    with_groups,
+)
+
+from tidy3.__version__ import __version__
+
+__all__ = [
+    "TidyFrame",
+    "JoinCondition",
+    "JoinSpec",
+    "Selector",
+    "across",
+    "cur_column",
+    "cur_group",
+    "cur_group_id",
+    "cur_group_rows",
+    "all",
+    "add_count",
+    "add_tally",
+    "anti_join",
+    "any",
+    "all_of",
+    "any_of",
+    "c_across",
+    "arrange",
+    "bind_cols",
+    "bind_rows",
+    "between",
+    "between_values",
+    "case_when",
+    "case_match",
+    "col",
+    "coalesce",
+    "collect",
+    "colnames",
+    "complete",
+    "col_range",
+    "cols_between",
+    "clean_names",
+    "closest",
+    "describe",
+    "dim",
+    "dtypes",
+    "contains",
+    "consecutive_id",
+    "count",
+    "cross_join",
+    "cumall",
+    "cumany",
+    "cume_dist",
+    "cummean",
+    "dense_rank",
+    "desc",
+    "distinct",
+    "drop",
+    "drop_na",
+    "ends_with",
+    "eq",
+    "everything",
+    "expand",
+    "ExportResult",
+    "collect_known_names",
+    "nb_export",
+    "transform_cell",
+    "transform_source",
+    "filter",
+    "filter_out",
+    "fill",
+    "hoist",
+    "first",
+    "full_join",
+    "ge",
+    "glimpse",
+    "group_by",
+    "group_map",
+    "group_modify",
+    "group_nest",
+    "group_split",
+    "group_cols",
+    "group_vars",
+    "n_groups",
+    "gt",
+    "head",
+    "inner_join",
+    "intersect",
+    "if_all",
+    "if_any",
+    "if_else",
+    "is_bool",
+    "is_boolean",
+    "is_categorical",
+    "is_character",
+    "is_datetime",
+    "is_float",
+    "is_integer",
+    "is_numeric",
+    "is_string",
+    "is_temporal",
+    "join_by",
+    "lag",
+    "last",
+    "last_col",
+    "le",
+    "lead",
+    "left_join",
+    "looks_like_tidy_pipe",
+    "lt",
+    "max",
+    "mean",
+    "median",
+    "make_clean_names",
+    "matches",
+    "min",
+    "min_rank",
+    "maybe_rewrite_cell",
+    "mutate",
+    "nest",
+    "nest_join",
+    "nesting",
+    "n",
+    "n_distinct",
+    "na_if",
+    "names",
+    "near",
+    "ncol",
+    "normalize_pipe_source",
+    "nrow",
+    "num_range",
+    "ntile",
+    "nth",
+    "options",
+    "overlaps",
+    "partial_run",
+    "peek",
+    "pick",
+    "percent_rank",
+    "pivot_longer",
+    "pivot_wider",
+    "pull",
+    "relocate",
+    "reframe",
+    "rename",
+    "rename_with",
+    "set_names",
+    "replace_na",
+    "right_join",
+    "rowwise",
+    "rows_append",
+    "rows_delete",
+    "rows_insert",
+    "rows_patch",
+    "rows_update",
+    "rows_upsert",
+    "row_number",
+    "recode",
+    "sample_frac",
+    "sample_n",
+    "scan_csv",
+    "scan_ipc",
+    "scan_parquet",
+    "select",
+    "separate",
+    "separate_longer_delim",
+    "separate_wider_delim",
+    "semi_join",
+    "setdiff",
+    "setequal",
+    "slice",
+    "slice_head",
+    "slice_max",
+    "slice_min",
+    "slice_sample",
+    "slice_tail",
+    "std",
+    "sd",
+    "starts_with",
+    "sum",
+    "summarise",
+    "summarize",
+    "summary",
+    "tally",
+    "tidy",
+    "to_numpy",
+    "transmute",
+    "unite",
+    "ungroup",
+    "unnest",
+    "unnest_longer",
+    "unnest_wider",
+    "pack",
+    "unpack",
+    "union",
+    "union_all",
+    "with_groups",
+    "var",
+    "symdiff",
+    "where",
+    "within",
+    "__version__",
+]
+
+# Inside IPython / SolveIt / Jupyter: enable multi-line ``>>`` rewriting even
+# when the user only did ``from tidy3 import …`` (without %load_ext / addon).
+try:
+    from IPython import get_ipython as _get_ipython
+
+    if _get_ipython is not None and _get_ipython() is not None:
+        from tidy3.jupyter import ensure_ipython_integration as _ensure_ip
+
+        _ensure_ip(quiet=True)
+except Exception:
+    pass

tidy3/__main__.py

@@ -0,0 +1,141 @@
+"""CLI: ``python -m tidy3 export notebook.ipynb -o script.py``."""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+
+def _cmd_export(args: argparse.Namespace) -> int:
+    from tidy3.export import nb_export
+
+    result = nb_export(
+        args.notebook,
+        args.output,
+        only_export=args.only_export,
+        with_plot3=not args.no_plot3,
+        known_extra=args.known or None,
+    )
+    print(result)
+    if result.warnings:
+        for w in result.warnings:
+            print(f"  warning: {w}", file=sys.stderr)
+    return 0
+
+
+def _cmd_run(args: argparse.Namespace) -> int:
+    """Run a ``.py`` file after applying the same R-style transforms as export."""
+    from tidy3.export import collect_known_names, transform_source
+
+    path = Path(args.script).expanduser().resolve()
+    if not path.is_file():
+        print(f"error: file not found: {path}", file=sys.stderr)
+        return 1
+    src = path.read_text(encoding="utf-8")
+    known = collect_known_names([src])
+    if args.known:
+        known.update(args.known)
+    try:
+        code = transform_source(src, known=known, with_plot3=not args.no_plot3)
+    except SyntaxError as e:
+        print(f"error: transform failed: {e}", file=sys.stderr)
+        return 1
+    ns: dict = {"__name__": "__main__", "__file__": str(path)}
+    # Inject public APIs so short scripts work like notebooks
+    if not args.no_inject:
+        import tidy3 as t3
+        from tidy3.masking import make_named_assign
+
+        for name in t3.__all__:
+            if name.startswith("_"):
+                continue
+            try:
+                ns[name] = getattr(t3, name)
+            except AttributeError:
+                pass
+        ns["tidy3"] = t3
+        ns["__tidy3_assign__"] = make_named_assign
+        try:
+            import plot3 as p3
+
+            for name in getattr(p3, "__all__", ()):
+                if str(name).startswith("_"):
+                    continue
+                try:
+                    ns[name] = getattr(p3, name)
+                except AttributeError:
+                    pass
+            ns["plot3"] = p3
+        except ImportError:
+            pass
+    exec(compile(code, str(path), "exec"), ns, ns)
+    return 0
+
+
+def main(argv: list[str] | None = None) -> int:
+    parser = argparse.ArgumentParser(
+        prog="python -m tidy3",
+        description="tidy3 utilities: export notebooks, run R-style scripts",
+    )
+    sub = parser.add_subparsers(dest="command", required=True)
+
+    p_export = sub.add_parser(
+        "export",
+        help="export a Jupyter notebook to plain Python (nbdev-style)",
+    )
+    p_export.add_argument("notebook", type=str, help="path to .ipynb")
+    p_export.add_argument(
+        "-o",
+        "--output",
+        type=str,
+        default=None,
+        help="output .py path (default: notebook stem + .py)",
+    )
+    p_export.add_argument(
+        "--only-export",
+        action="store_true",
+        help="only include cells marked with #| export",
+    )
+    p_export.add_argument(
+        "--no-plot3",
+        action="store_true",
+        help="do not apply plot3 aes/facet masking",
+    )
+    p_export.add_argument(
+        "--known",
+        action="append",
+        default=[],
+        help="extra non-column name (repeatable), e.g. --known cars",
+    )
+    p_export.set_defaults(func=_cmd_export)
+
+    p_run = sub.add_parser(
+        "run",
+        help="run a .py file after R-style bare-name/backtick transforms",
+    )
+    p_run.add_argument("script", type=str, help="path to .py")
+    p_run.add_argument(
+        "--no-plot3",
+        action="store_true",
+        help="do not apply plot3 aes/facet masking",
+    )
+    p_run.add_argument(
+        "--no-inject",
+        action="store_true",
+        help="do not inject tidy3/plot3 into the script namespace",
+    )
+    p_run.add_argument(
+        "--known",
+        action="append",
+        default=[],
+        help="extra non-column name (repeatable)",
+    )
+    p_run.set_defaults(func=_cmd_run)
+
+    args = parser.parse_args(argv)
+    return int(args.func(args))
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

tidy3/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.3.0"