[Python] load_dotenv()를 통해서 파이썬 환경변수 읽어오기

최근에 openai의 API를 이용할 서비스를 개발하면서, API key를 사용할 일이 있었다. 그런데 API key는 외부에 노출되면 안되는 정보기도 하고 여러 코드에서 중복으로 사용되는 코드라서 환경변수에 등록하고 사용해야 하는데 환경변수에 등록하고 현재 파일에서 읽어오는 과정을 정리하려고 한다.

 

그래서 dotenv 모듈의 load_dotenv 함수를 통해 환경변수를 운영체제의 환경변수로 등록해주는 방법을 정리하려고 하려고 한다.

 

 

우선 dotenv메서드 사용을 위해 필요한 라이브러리를 설치한다.

pip install python-dotenv

 

 

env 파일이 다음과 같이 있다면 

OPENAI_API_KEY="sk-"

 

다음 처럼 환경변수를 읽어올 수 있다.

from dotenv import load_dotenv
import os

load_dotenv()
api_key = os.getenv("OPENAI_API_KEY")

 

 

 

 

 

최종 코드

#.env파일
OPENAI_API_KEY="sk-"

 

 

# 환경변수를 불러와서 사용할 실행 코드
from openai import OpenAI
from dotenv import load_dotenv
import os

load_dotenv()
api_key = os.getenv("OPENAI_API_KEY")

 

 

 

load_dotenv() 동장 방식

 

load_dotenv()메서드는 같은 디렉토리내에 있는 .env파일을 읽어서 환경변수로 등록하는데, 만약 .env 파일이 현재 실행하는 파일과 다른 디렉토리에 있다면, 다음 처럼 명시적으로 환경변수의 경로를 지정해줘서 사용할 수 있다.

load_dotenv(dotenv_path="/your/custom/path/.env")

 

 

default는 현재 디렉토리에서 환경변수를 찾고 없다면, 상위 디렉토리를 찾아가면서 env파일을 찾는다.

 

 

python 문법

from A import B

=> A모듈에서 B 함수를 불러오는 문법. A.B()대신 B()만 쓸 수 있게 해준다.

 

 

 

 

참고 자료 : https://pypi.org/project/python-dotenv/