Initial YakPanel commit

2026-04-07 02:04:22 +05:30
commit 2826d3e7f3
5359 changed files with 1390724 additions and 0 deletions
--- a/class_v2/db_mysql_v2.py
+++ b/class_v2/db_mysql_v2.py
@@ -0,0 +1,374 @@
+#coding: utf-8
+# +-------------------------------------------------------------------
+# | YakPanel
+# +-------------------------------------------------------------------
+# | Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved.
+# +-------------------------------------------------------------------
+# | Author: hwliang <hwl@yakpanel.com>
+# +-------------------------------------------------------------------
+
+import re,os,sys,public,json
+import pymysql
+
+class panelMysql:
+    __DB_PASS = None
+    __DB_USER = 'root'
+    __DB_NAME = None
+    __DB_PORT = 3306
+    __DB_HOST = 'localhost'
+    __DB_PREFIX = ''
+    __DB_CONN = None
+    __DB_CUR  = None
+    __DB_ERR  = None
+    __DB_TABLE   = ""              # 被操作的表名称
+    __OPT_WHERE  = ""              # where条件
+    __OPT_LIMIT  = ""              # limit条件
+    __OPT_ORDER  = ""              # order条件
+    __OPT_FIELD  = "*"             # field条件
+    __OPT_PARAM  = ()              # where值
+    _USER = None
+    _ex = None
+
+    def __init__(self):
+        pass
+
+    def set_name(self,name):
+        self.__DB_NAME = str(name)
+        return self
+
+    def set_prefix(self,prefix):
+        self.__DB_PREFIX = prefix
+        return self
+
+    def set_host(self,host,port,name,username,password,prefix = ''):
+        self.__DB_HOST = host
+        self.__DB_PORT = int(port)
+        self.__DB_NAME = name
+        if self.__DB_NAME: self.__DB_NAME = str(self.__DB_NAME)
+        self.__DB_USER = str(username)
+        self._USER = str(username)
+        self.__DB_PASS = str(password)
+        self.__DB_PREFIX = prefix
+        if not self.__GetConn(): return False
+        return self
+
+    #连接MYSQL数据库
+    def __GetConn(self):
+        try:
+           # print(self.__DB_HOST,self.__DB_PORT,self.__DB_NAME,self.__DB_USER,self.__DB_PASS)
+            self.__DB_CONN = pymysql.connect(host=self.__DB_HOST,user=self.__DB_USER,passwd=str(self.__DB_PASS),db=self.__DB_NAME,port=self.__DB_PORT,connect_timeout=15,read_timeout=60,write_timeout=60)
+        except Exception as ex:
+            self.__DB_ERR = "error: " + str(ex)
+            self._ex = ex
+            print(ex)
+            if self.__DB_ERR.find("timed out") != -1 or self.__DB_ERR.find("is not allowed to connect") != -1: return False
+            try:
+                self.__DB_CONN = pymysql.connect(host=self.__DB_HOST,user=self.__DB_USER,passwd=str(self.__DB_PASS),db=self.__DB_NAME,port=self.__DB_PORT)
+            except Exception as ex:
+                self.__DB_ERR = "error: " + str(ex)
+                self._ex = ex
+                print(ex)
+                return False
+        self.__DB_CUR  = self.__DB_CONN.cursor()
+        return True
+
+    def table(self,table):
+        #设置表名
+        self.__DB_TABLE = self.__DB_PREFIX + table
+        return self
+
+
+    def where(self,where,param):
+        #WHERE条件
+        if where:
+            self.__OPT_WHERE = " WHERE " + where
+            self.__OPT_PARAM = self.__to_tuple(param)
+        return self
+
+    def __to_tuple(self,param):
+        #将参数转换为tuple
+        if type(param) != tuple:
+            if type(param) == list:
+                param = tuple(param)
+            else:
+                param = (param,)
+        return param
+
+
+    def order(self,order):
+        #ORDER条件
+        if len(order):
+            self.__OPT_ORDER = " ORDER BY "+order
+        return self
+
+
+    def limit(self,limit):
+        #LIMIT条件
+        limit = str(limit)
+        if len(limit):
+            self.__OPT_LIMIT = " LIMIT "+ limit
+        return self
+
+
+    def field(self,field):
+        #FIELD条件
+        if len(field):
+            self.__OPT_FIELD = field
+        return self
+
+
+    def select(self):
+        #查询数据集
+        self.__GetConn()
+        if not self.__DB_CUR: return self.__DB_ERR
+        try:
+            self.__get_columns()
+            sql = "SELECT " + self.__OPT_FIELD + " FROM " + self.__DB_TABLE + self.__OPT_WHERE + self.__OPT_ORDER + self.__OPT_LIMIT
+            self.__DB_CUR.execute(sql,self.__OPT_PARAM)
+            data = self.__DB_CUR.fetchall()
+            #构造字典系列
+            if self.__OPT_FIELD != "*":
+                fields = self.__format_field(self.__OPT_FIELD.split(','))
+                tmp = []
+                for row in data:
+                    i=0
+                    tmp1 = {}
+                    for key in fields:
+                        tmp1[key.strip('`')] = row[i]
+                        i += 1
+                    tmp.append(tmp1)
+                    del(tmp1)
+                data = tmp
+                del(tmp)
+            else:
+                #将元组转换成列表
+                tmp = list(map(list, data))
+                data = tmp
+                del(tmp)
+            self.__close()
+            return data
+        except Exception as ex:
+            self._ex = ex
+            return "error: " + str(ex)
+
+    def get(self):
+        self.__get_columns()
+        return self.select()
+
+    def __format_field(self, field):
+        import re
+        fields = []
+        for key in field:
+            s_as = re.search(r'\s+as\s+', key, flags=re.IGNORECASE)
+            if s_as:
+                as_tip = s_as.group()
+                key = key.split(as_tip)[1]
+            fields.append(key)
+        return fields
+
+    def __get_columns(self):
+        if self.__OPT_FIELD == '*':
+            tmp_cols = self.query(
+                "select COLUMN_NAME from information_schema.COLUMNS where table_name = '{}' and table_schema = '{}';"
+                .format(self.__DB_TABLE, self.__DB_NAME), False)
+            cols = []
+            for col in tmp_cols:
+                cols.append('`' + col[0] + '`')
+            if len(cols) > 0: self.__OPT_FIELD = ','.join(cols)
+
+    def getField(self, keyName):
+        #取回指定字段
+        try:
+            result = self.field(keyName).select()
+            if len(result) != 0:
+                return result[0][keyName]
+            return result
+        except:
+            return None
+
+    def setField(self, keyName, keyValue):
+        #更新指定字段
+        return self.save(keyName, (keyValue, ))
+
+    def find(self):
+        #取一行数据
+        try:
+            result = self.limit("1").select()
+            if len(result) == 1:
+                return result[0]
+            return result
+        except:
+            return None
+
+    def count(self):
+        #取行数
+        key = "COUNT(*)"
+        data = self.field(key).select()
+        try:
+            return int(data[0][key])
+        except:
+            return 0
+
+    def add(self, keys, param):
+        #插入数据
+        self.__GetConn()
+        self.__DB_CONN.text_factory = str
+        try:
+            values = ""
+            for key in keys.split(','):
+                values += "%s,"
+            values = values[0:len(values) - 1]
+            sql = "INSERT INTO " + self.__DB_TABLE + "(" + keys + ") " + "VALUES(" + values + ")"
+            self.__DB_CUR.execute(sql, self.__to_tuple(param))
+            id = self.__DB_CUR.lastrowid
+            self.__close()
+            self.__DB_CONN.commit()
+            return id
+        except Exception as ex:
+            self._ex = ex
+            return "error: " + str(ex)
+
+    #插入数据
+    def insert(self, pdata):
+        if not pdata: return False
+        keys, param = self.__format_pdata(pdata)
+        return self.add(keys, param)
+
+    #更新数据
+    def update(self, pdata):
+        if not pdata: return False
+        keys, param = self.__format_pdata(pdata)
+        return self.save(keys, param)
+
+    #构造数据
+    def __format_pdata(self, pdata):
+        keys = pdata.keys()
+        keys_tmp = []
+        for k in keys:
+            keys_tmp.append("`{}`".format(k))
+        keys_str = ','.join(keys_tmp)
+
+        param = []
+        for k in keys:
+            #if pdata[k] == None: pdata[k] = ''
+            param.append(pdata[k])
+        return keys_str, tuple(param)
+
+    def addAll(self, keys, param):
+        #插入数据
+        self.__GetConn()
+        self.__DB_CONN.text_factory = str
+        try:
+            values = ""
+            for key in keys.split(','):
+                values += "%s,"
+            values = values[0:len(values) - 1]
+            sql = "INSERT INTO " + self.__DB_TABLE + "(" + keys + ") " + "VALUES(" + values + ")"
+            result = self.__DB_CUR.execute(sql, self.__to_tuple(param))
+            return True
+        except Exception as ex:
+            self._ex = ex
+            return "error: " + str(ex)
+
+    def commit(self):
+        self.__close()
+        self.__DB_CONN.commit()
+
+    def save(self, keys, param):
+        #更新数据
+        self.__GetConn()
+        self.__DB_CONN.text_factory = str
+        try:
+            opt = ""
+            for key in keys.split(','):
+                opt += key + "=%s,"
+            opt = opt[0:len(opt) - 1]
+            sql = "UPDATE " + self.__DB_TABLE + " SET " + opt + self.__OPT_WHERE
+
+            #处理拼接WHERE与UPDATE参数
+            if param:
+                tmp = list(self.__to_tuple(param))
+                for arg in self.__OPT_PARAM:
+                    tmp.append(arg)
+                self.__OPT_PARAM = tuple(tmp)
+                self.__DB_CUR.execute(sql,self.__OPT_PARAM)
+            else:
+                self.__DB_CUR.execute(sql)
+            self.__close()
+            self.__DB_CONN.commit()
+            return self.__DB_CUR.rowcount
+        except Exception as ex:
+            self._ex = ex
+            return "error: " + str(ex)
+
+    def delete(self, id=None):
+        #删除数据
+        self.__GetConn()
+        try:
+            if id:
+                self.__OPT_WHERE = " WHERE id=%s"
+                self.__OPT_PARAM = (id, )
+            sql = "DELETE FROM " + self.__DB_TABLE + self.__OPT_WHERE
+            self.__DB_CUR.execute(sql, self.__OPT_PARAM)
+            self.__close()
+            self.__DB_CONN.commit()
+            return self.__DB_CUR.rowcount
+        except Exception as ex:
+            return "error: " + str(ex)
+
+    def execute(self,sql,param = ()):
+        #执行SQL语句返回受影响行
+        if not self.__GetConn(): return self.__DB_ERR
+        try:
+            if param:
+                self.__OPT_PARAM = list(self.__to_tuple(param))
+                result = self.__DB_CUR.execute(sql,self.__OPT_PARAM)
+            else:
+                result = self.__DB_CUR.execute(sql)
+            self.__DB_CONN.commit()
+            self.__close()
+            return result
+        except Exception as ex:
+            self._ex = ex
+            return ex
+
+
+    def query(self,sql,is_close=True,param=()):
+        #执行SQL语句返回数据集
+        if not self.__GetConn(): return self.__DB_ERR
+        try:
+            if param:
+                self.__OPT_PARAM = list(self.__to_tuple(param))
+                self.__DB_CUR.execute(sql,self.__OPT_PARAM)
+            else:
+                self.__DB_CUR.execute(sql)
+            result = self.__DB_CUR.fetchall()
+            #将元组转换成列表
+            data = list(map(list,result))
+            if is_close: self.__Close()
+            return data
+        except Exception as ex:
+            self._ex = ex
+            return ex
+
+
+    #关闭连接
+    def __Close(self):
+        self.__DB_CUR.close()
+        self.__DB_CONN.close()
+
+    def __close(self):
+        #清理条件属性
+        self.__OPT_WHERE = ""
+        self.__OPT_FIELD = "*"
+        self.__OPT_ORDER = ""
+        self.__OPT_LIMIT = ""
+        self.__OPT_PARAM = ()
+
+    def close(self):
+        #释放资源
+        try:
+            self.__DB_CUR.close()
+            self.__DB_CUR.close()
+        except:
+            pass