From 38e13d086ca8cbecbca987a7935eb977ab3e6510 Mon Sep 17 00:00:00 2001 From: midoks Date: Tue, 22 Nov 2022 09:25:15 +0800 Subject: [PATCH] Update pg.py --- plugins/postgresql/class/pg.py | 25 +++++++++++++++++++++---- 1 file changed, 21 insertions(+), 4 deletions(-) diff --git a/plugins/postgresql/class/pg.py b/plugins/postgresql/class/pg.py index 0a0320747..7f34c3c4b 100755 --- a/plugins/postgresql/class/pg.py +++ b/plugins/postgresql/class/pg.py @@ -7,6 +7,10 @@ import sys import psycopg2 +sys.path.append(os.getcwd() + "/class/core") +import mw + + class ORM: __DB_PASS = None __DB_USER = 'postgres' @@ -32,12 +36,25 @@ class ORM: try: try: - self.__DB_CONN = psycopg2.connect(database='postgres', user=self.__DB_USER, password=self.__DB_PASS, - host=self.__DB_HOST, port=int(self.__DB_PORT)) + self.__DB_CONN = psycopg2.connect(database='postgres', + user=self.__DB_USER, + password=self.__DB_PASS, + host=self.__DB_HOST, + port=int(self.__DB_PORT)) except Exception as e: self.__DB_HOST = '127.0.0.1' - self.__DB_CONN = psycopg2.connect(database='postgres', user=self.__DB_USER, password=self.__DB_PASS, - host=self.__DB_HOST, port=int(self.__DB_PORT)) + self.__DB_CONN = psycopg2.connect(database='postgres', + user=self.__DB_USER, + password=self.__DB_PASS, + host=self.__DB_HOST, + port=int(self.__DB_PORT)) + except Exception as e: + self.__DB_HOST = mw.getLocalIp() + self.__DB_CONN = psycopg2.connect(database='postgres', + user=self.__DB_USER, + password=self.__DB_PASS, + host=self.__DB_HOST, + port=int(self.__DB_PORT)) self.__DB_CONN.autocommit = True self.__DB_CUR = self.__DB_CONN.cursor()