blob: 03936c175a6f941d6405ba0ad29ad84b594e39e4 (
plain)
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
|
#!/bin/sh
[ "$1" = python3-dotenv ] || exit 0
python3 - << 'EOF'
import os
import tempfile
from dotenv import dotenv_values, load_dotenv, set_key, get_key
# Write a temp .env file and parse it
with tempfile.NamedTemporaryFile(mode='w', suffix='.env', delete=False) as f:
f.write('FOO=bar\n')
f.write('BAZ=123\n')
f.write('QUOTED="hello world"\n')
env_path = f.name
try:
values = dotenv_values(env_path)
assert values['FOO'] == 'bar', f"got FOO={values['FOO']}"
assert values['BAZ'] == '123', f"got BAZ={values['BAZ']}"
assert values['QUOTED'] == 'hello world', f"got QUOTED={values['QUOTED']}"
# Test load_dotenv sets environment variables
load_dotenv(env_path, override=True)
assert os.environ.get('FOO') == 'bar'
assert os.environ.get('BAZ') == '123'
finally:
os.unlink(env_path)
print("python-dotenv OK")
EOF
|
