script.py

#!/run/current-system/sw/bin/env nix-shell
#! nix-shell -f script_env.nix -i python3 -p python3

import click

@click.command()
def test():
    print("hi")

script_env.nix

with import <nixpkgs> {};
python3.withPackages (ps: [ps.click])