28 lines
907 B
Python
28 lines
907 B
Python
from configparser import ConfigParser
|
|
import os
|
|
|
|
|
|
ENVIRONMENT_VARIABLE = os.getenv("ENVIRONMENT_VARIABLE", "test")
|
|
def get_path():
|
|
current_directory = os.path.dirname(os.path.abspath(__file__))
|
|
root_path = os.path.abspath(os.path.dirname(current_directory) + os.path.sep + ".")
|
|
return root_path
|
|
|
|
|
|
def get_pg_config():
|
|
conf = ConfigParser()
|
|
conf.read(get_path() + '/config/config.ini')
|
|
return {
|
|
"database": conf[ENVIRONMENT_VARIABLE]['database'],
|
|
"user": conf[ENVIRONMENT_VARIABLE]['user'],
|
|
"password": conf[ENVIRONMENT_VARIABLE]['password'],
|
|
"host": conf[ENVIRONMENT_VARIABLE]['host'],
|
|
"port": conf[ENVIRONMENT_VARIABLE]['port'],
|
|
# 设置默认超时时间statement_timeout=五分钟
|
|
"options":"-c statement_timeout=800000 -c idle_in_transaction_session_timeout=900000"
|
|
}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print(get_path())
|