From e8a3f8b014bf88af6dfb8e31e8b59f3780f49705 Mon Sep 17 00:00:00 2001 From: Mr Chen Date: Wed, 15 May 2024 04:16:49 +0800 Subject: [PATCH] Update orm.py --- class/plugin/orm.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/class/plugin/orm.py b/class/plugin/orm.py index 3e96ff9f1..65d3a2135 100755 --- a/class/plugin/orm.py +++ b/class/plugin/orm.py @@ -25,7 +25,19 @@ class ORM: '''连接数据库''' try: - if os.path.exists(self.__DB_SOCKET): + if self.__DB_HOST != 'localhost': + try: + self.__DB_CONN = pymysql.connect(host=self.__DB_HOST, user=self.__DB_USER, passwd=self.__DB_PASS, + database=self.__DB_NAME, + port=int(self.__DB_PORT), charset=self.__DB_CHARSET, connect_timeout=1, + cursorclass=pymysql.cursors.DictCursor) + except Exception as e: + self.__DB_HOST = '127.0.0.1' + self.__DB_CONN = pymysql.connect(host=self.__DB_HOST, user=self.__DB_USER, passwd=self.__DB_PASS, + database=self.__DB_NAME, + port=int(self.__DB_PORT), charset=self.__DB_CHARSET, connect_timeout=1, + cursorclass=pymysql.cursors.DictCursor) + else if os.path.exists(self.__DB_SOCKET): try: self.__DB_CONN = pymysql.connect(host=self.__DB_HOST, user=self.__DB_USER, passwd=self.__DB_PASS, database=self.__DB_NAME,