37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
|
import threading
|
|||
|
|
from django.db import connection
|
|||
|
|
g_dbLock = threading.Lock()# 用于操作数据库的全局锁（20240930新增，由于sqlite不支持锁，因此在程序中做锁控制）
|
|||
|
|
|
|||
|
|
class Database(object):
|
|||
|
|
def __init__(self, logger):
|
|||
|
|
self.logger = logger
|
|||
|
|
|
|||
|
|
def select(self, sql, params=None):
|
|||
|
|
data = []
|
|||
|
|
with g_dbLock:
|
|||
|
|
cursor = connection.cursor()
|
|||
|
|
cursor.execute(sql, params or None)
|
|||
|
|
try:
|
|||
|
|
rawData = cursor.fetchall()
|
|||
|
|
col_names = [desc[0] for desc in cursor.description]
|
|||
|
|
for row in rawData:
|
|||
|
|
d = {}
|
|||
|
|
for index, value in enumerate(row):
|
|||
|
|
d[col_names[index]] = value
|
|||
|
|
data.append(d)
|
|||
|
|
except Exception as e:
|
|||
|
|
self.logger.error("Database.select() error:%s,sql:%s" % (str(e),sql))
|
|||
|
|
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def execute(self, sql, params=None):
|
|||
|
|
ret = False
|
|||
|
|
with g_dbLock:
|
|||
|
|
try:
|
|||
|
|
cursor = connection.cursor()
|
|||
|
|
cursor.execute(sql, params or None)
|
|||
|
|
ret = True
|
|||
|
|
except Exception as e:
|
|||
|
|
self.logger.error("Database.execute() error:%s,sql:%s" % (str(e), sql))
|
|||
|
|
return ret
|