backtest-cpp-py/flake.nix at master · MarcPartensky/backtest-cpp-py · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
{
  description = "Event-driven backtesting engine";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
  };

  outputs = {
    self,
    nixpkgs,
    flake-utils,
  }:
    flake-utils.lib.eachDefaultSystem (system: let
      pkgs = nixpkgs.legacyPackages.${system};

      python = pkgs.python312;

      pythonEnv = python.withPackages (ps:
        with ps; [
          numpy
          pandas
          matplotlib
          yfinance
          # dev
          ruff
        ]);
    in {
      devShells.default = pkgs.mkShell {
        packages = [
          pythonEnv
          pkgs.uv
          pkgs.just
          pkgs.ruff
        ];

        shellHook = ''
          echo "backtest dev shell — python $(python --version)"
        '';
      };

      packages.default = pkgs.stdenv.mkDerivation {
        pname = "backtest";
        version = "0.1.0";
        src = ./.;
        buildInputs = [pythonEnv];
        installPhase = ''
          mkdir -p $out/bin $out/lib
          cp backtest.py $out/lib/backtest.py
          cat > $out/bin/backtest << EOF
          #!/bin/sh
          exec ${pythonEnv}/bin/python $out/lib/backtest.py "\$@"
          EOF
          chmod +x $out/bin/backtest
        '';
      };
    });
}