CT_project/管理节点/db_operation.py

import threading
import time
import mysqlhelper
class DBOperation(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
        self._space_dict = {1:{},2:{},3:{}}
        self._command_dict = {1:{},2:{},3:{}}

        self._db = mysqlhelper.MySqLHelper()
        self._isClose = False
    def close(self):
        self._isClose = True

    def query_command_in_unit(self, unit):
        sql = "select * from command_queue WHERE unit=%s"
        return self._db.selectall(sql, unit)

    def query_command_all(self):
        sql = "select * from command_queue"
        return self._db.selectall(sql)
    def query_space_in_unit(self,unit):
        sql = "select * from space WHERE unit=%s"
        return self._db.selectall(sql,unit)
    def query_space_all(self):
        sql = "select * from space"
        return self._db.selectall(sql)
    def update_space_status(self,space_id,statu):
        sql = "update space set statu=%s where id=%s"
        return self._db.update(sql,(statu,space_id))
    def clear_space_data(self,space_id):
        sql = "update space set car_number=NULL where id=%s"
        return self._db.update(sql,space_id)
    def query_vehicle_primary_key(self,car_number):
        sql = "select primary_key from vehicle where car_number=%s"
        return self._db.selectall(sql,car_number)
    def delete_command(self,car_number):
        sql = "delete from command_queue where car_number=%s"
        return self._db.delete(sql,car_number)
    def get_space_dict(self):
        return self._space_dict
    def get_command_dict(self):
        return self._command_dict
    def run(self):
        while not self._isClose:
            for unit in range(1,4):
                res = self.query_space_in_unit(unit)
                if len(res) > 0 and self._space_dict[unit] != res:
                    self._space_dict[unit] = res

            for unit in range(1,4):
                res = self.query_command_in_unit(unit)
                if self._command_dict[unit] != res:
                    self._command_dict[unit] = res

            time.sleep(0.001)



# class ParkManage(threading.Thread):
#     def __init__(self, ip, port, database, user, password, sender):
#         threading.Thread.__init__(self)
#         self.conn = None
#         self.ip = ip
#         self.port = port
#         self.database = database
#         self.user = user
#         self.password = password
#
#
#         self.lock = threading.Lock()
#
#         self.a_unit_park_list = {}
#         self.b_unit_park_list = {}
#         self.c_unit_park_list = {}
#
#         self.command_queue_dict = {}
#         self.a_command_queue_list = {}
#         self.b_command_queue_list = {}
#         self.c_command_queue_list = {}
#
#
#         self.my_sql_str = queue.Queue()
#
#
#
#         self.isClose = False
#         self.g_sender = sender
#
#         self.db_statu_callback = None
#         self.command_update_callback = None
#         self.space_update_callback = None
#
#     def setDbStatuCallback(self, callback):
#         self.db_statu_callback = callback
#     def setCommandUpdateCallback(self,callback):
#         self.command_update_callback = callback
#
#     def setSpaceUpdateCallback(self, callback):
#         self.space_update_callback = callback
#     def connect(self):
#         try:
#             self.conn = pymysql.connect(host=self.ip, port=self.port, database=self.database, charset="utf8",
#                                         user=self.user, passwd=self.password)
#             return True
#         except:
#             return False
#     def putSql(self,SQL):
#         self.my_sql_str.put(SQL)
#
#     def updatePark(self, dict, statu):
#         self.lock.acquire()
#
#         cursor = self.conn.cursor()
#         SQL = "update space set statu=%d where id=%d" % (statu, dict["id"])
#         cursor.execute(SQL)
#         print(SQL)
#         self.conn.commit()
#         cursor.close()
#         self.lock.release()
#
#     def clearPark(self, dict):
#         self.lock.acquire()
#
#         cursor = self.conn.cursor()
#         SQL = "update space set car_number=NULL where id=%d" % (dict["id"])
#         cursor.execute(SQL)
#         print(SQL)
#         self.conn.commit()
#         cursor.close()
#         self.lock.release()
#
#     def processDelete(self, dict):
#         self.lock.acquire()
#         cursor = self.conn.cursor()
#         SQL = "delete from command_queue where car_number='%s';" % (dict["car_number"])
#         cursor.execute(SQL)
#         print(SQL)
#         self.conn.commit()
#         cursor.close()
#         self.lock.release()
#
#     def pickUpPark(self, dict):
#         self.lock.acquire()
#
#         cursor = self.conn.cursor()
#         SQL = "select primary_key from vehicle where car_number='%s'" % (dict["car_number"])
#         cursor.execute(SQL)
#         self.conn.commit()
#         results = cursor.fetchall()
#         cursor.close()
#
#         self.lock.release()
#
#         if len(results) == 1:
#             key = ''.join(results[0])
#             table = msg.pick_table()
#             table.primary_key = key
#             self.g_sender.publish("command_ex", "user_command_port", tf.MessageToString(table, as_utf8=True))
#             QMessageBox.question(None, '提示', '取车消息发送成功！',
#                                  QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容
#
#         else:
#             QMessageBox.warning(None, '警告', '查询结果有误,请检查数据库！',
#                                 QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容
#     def close(self):
#         self.isClose = True
#     def run(self):
#         while self.isClose is not True:
#             try:
#                 self.conn.ping()  # 采用连接对象的ping()函数检测连接状态
#                 self.db_statu_callback(True)
#             except:
#                 self.db_statu_callback(False)
#                 self.connect()
#                 time.sleep(0.5)
#                 continue
#             # 获取一个光标
#             cursor = self.conn.cursor(cursor=pymysql.cursors.DictCursor)
#             SQL = "select * from space"
#             t_a_unit_park_list = []
#             t_b_unit_park_list = []
#             t_c_unit_park_list = []
#             results = cursor.execute(SQL)
#
#
#             # 结果数量大于0
#             if results > 0:
#                 parkspace = cursor.fetchall()
#                 for space in parkspace:
#                     if space["unit"] == 1:
#                         t_a_unit_park_list.append(space)
#                     if space["unit"] == 2:
#                         t_b_unit_park_list.append(space)
#                     if space["unit"] == 3:
#                         t_c_unit_park_list.append(space)
#
#             if self.a_unit_park_list != t_a_unit_park_list:
#                 self.a_unit_park_list = t_a_unit_park_list
#                 self.space_update_callback(self.a_unit_park_list, 'A')
#
#             if self.b_unit_park_list != t_b_unit_park_list:
#                 self.b_unit_park_list = t_b_unit_park_list
#                 self.space_update_callback(self.b_unit_park_list, 'B')
#
#             if self.c_unit_park_list != t_c_unit_park_list:
#                 self.c_unit_park_list = t_c_unit_park_list
#                 self.space_update_callback(self.c_unit_park_list, 'C')
#
#             SQL = "select * from command_queue"
#             t_a_command_list = []
#             t_b_command_list = []
#             t_c_command_list = []
#             results = cursor.execute(SQL)
#             if results > 0:
#                 command_queue = cursor.fetchall()
#                 self.command_queue_dict = command_queue
#                 for command in command_queue:
#                     if command["unit"] == 1:
#                         t_a_command_list.append(command)
#                     if command["unit"] == 2:
#                         t_a_command_list.append(command)
#                     if command["unit"] == 3:
#                         t_a_command_list.append(command)
#
#             if self.a_command_queue_list != t_a_command_list:
#                 self.a_command_queue_list = t_a_command_list
#                 self.command_update_callback(self.a_command_queue_list, 'A')
#
#             elif self.b_command_queue_list != t_b_command_list:
#                 self.b_command_queue_list = t_b_command_list
#                 self.command_update_callback(self.b_command_queue_list, 'B')
#
#             elif self.c_command_queue_list != t_c_command_list:
#                 self.c_command_queue_list = t_c_command_list
#                 self.command_update_callback(self.c_command_queue_list, 'C')
#
#             if self.my_sql_str.qsize() > 0:
#                 SQL = self.my_sql_str.get(False)
#                 cursor.execute(SQL)
#                 self.conn.commit()
#
#             # 关闭光标
#             cursor.close()
#
#             # print("------------------------------")
#             time.sleep(0.001)