pub-solar-os/profiles/develop/python/default.nix

{ pkgs, ... }:
let inherit (pkgs) python3Packages;
in
{
  environment.systemPackages =
    let
      packages = pythonPackages:
        with pythonPackages; [
          numpy
          pandas
          ptpython
          requests
          scipy
        ];

      python = pkgs.python3.withPackages packages;

    in
    [ python ];
  environment.sessionVariables = {
    PYTHONSTARTUP =
      let
        startup = pkgs.writers.writePython3 "ptpython.py"
          {
            libraries = with python3Packages; [ ptpython ];
          } ''
          from __future__ import unicode_literals

          from pygments.token import Token

          from ptpython.layout import CompletionVisualisation

          import sys

          ${builtins.readFile ./ptconfig.py}

          try:
              from ptpython.repl import embed
          except ImportError:
              print("ptpython is not available: falling back to standard prompt")
          else:
              sys.exit(embed(globals(), locals(), configure=configure))
        '';
      in
      "${startup}";
  };
}