|
|
from PyQt5 import QtCore
|
|
|
from PyQt5.QtWidgets import *
|
|
|
|
|
|
|
|
|
class CustomEdit(QLineEdit):
|
|
|
"""
|
|
|
带下拉框的文本输入框类（文本下拉输入框），输入文字可以搜索内容
|
|
|
"""
|
|
|
def __init__(self, parent, size=(50, 50, 100, 20), name='edit',
|
|
|
drag=False, text_list=True, search=True, qss_file=''):
|
|
|
"""
|
|
|
初始化控件
|
|
|
:param parent: 控件显示的父对象
|
|
|
:param size: 输入框控件尺寸
|
|
|
:param name: 输入框控件名称
|
|
|
:param drag: 拖放标识位
|
|
|
:param text_list: 是否开启下拉框功能标识位
|
|
|
"""
|
|
|
super(CustomEdit, self).__init__(None, parent)
|
|
|
# 输入框的尺寸
|
|
|
self.w = size[2]
|
|
|
self.h = size[3]
|
|
|
self.setGeometry(*size)
|
|
|
# 输入框名称
|
|
|
self.setObjectName(name)
|
|
|
# 输入框是否支持拖放
|
|
|
self.drag_flag = False
|
|
|
if drag:
|
|
|
self.setAcceptDrops(True)
|
|
|
self.setDragEnabled(True) # 开启可拖放事件
|
|
|
else:
|
|
|
self.setAcceptDrops(False)
|
|
|
self.setDragEnabled(False) # 关闭可拖放事件
|
|
|
|
|
|
self.click_flag = False # 点击状态，为False时显示下拉框
|
|
|
self.text_list = None # 下拉框对象
|
|
|
self.p_text = "" # placehold text，输入框背景上的灰色文字
|
|
|
self.qss_file = qss_file # 下拉框样式文件
|
|
|
self.org_data = [] # 用来进行搜索的数据list
|
|
|
|
|
|
# 如果text_list为True，则初始化下拉框
|
|
|
if text_list:
|
|
|
# 下拉列表模型为StringListModel
|
|
|
self.list_model = QtCore.QStringListModel()
|
|
|
# 初始化下拉列表对象为QListView类型
|
|
|
self.text_list = QListView(parent)
|
|
|
# 下拉列表名称，以list_开头
|
|
|
self.text_list.setObjectName("list_%s" % name)
|
|
|
# 设置下拉列表初始化尺寸
|
|
|
self.text_list.setGeometry(size[0], size[1]+size[3], size[2], size[3])
|
|
|
# 隐藏下拉列表
|
|
|
self.text_list.hide()
|
|
|
# 初始化下拉列表数据
|
|
|
self.text_list.data = []
|
|
|
|
|
|
if search:
|
|
|
self.textChanged.connect(self.on_search_data)
|
|
|
|
|
|
def fill_data(self, text_data_list):
|
|
|
"""
|
|
|
初始化文本下拉输入框控件数据
|
|
|
:param text_data_list: 下拉框数据，必须为纯文本list
|
|
|
"""
|
|
|
# 设置下拉列表数据
|
|
|
self.list_model.setStringList(text_data_list)
|
|
|
self.text_list.setModel(self.list_model)
|
|
|
self.text_list.data = text_data_list
|
|
|
# 填充list数据
|
|
|
if text_data_list:
|
|
|
self.text_list.data = text_data_list
|
|
|
self.org_data = text_data_list[:] # 深拷贝一份数据，用来搜索
|
|
|
self.text_list.resize(self.w, self.h * 0.6 * len(self.text_list.data))
|
|
|
self.setText(text_data_list[0])
|
|
|
else:
|
|
|
self.setText('')
|
|
|
self.text_list.hide()
|
|
|
|
|
|
edit_clicked = QtCore.pyqtSignal() # 定义clicked信号
|
|
|
|
|
|
def mouseReleaseEvent(self, event):
|
|
|
"""
|
|
|
鼠标按键松开时的事件处理
|
|
|
:param event: 鼠标按键松开事件
|
|
|
"""
|
|
|
# 左键松开
|
|
|
if event.button() == QtCore.Qt.LeftButton:
|
|
|
# 下拉列表存在数据，并且未点击过，则显示下拉框
|
|
|
if self.text_list.data and not self.click_flag:
|
|
|
self.text_list.show()
|
|
|
# 下拉列表存在，并且点击过，则隐藏下拉框
|
|
|
elif self.click_flag:
|
|
|
self.text_list.hide()
|
|
|
self.click_flag = not self.click_flag
|
|
|
self.edit_clicked.emit() # 发送clicked信号
|
|
|
|
|
|
def on_select_data(self, func):
|
|
|
"""
|
|
|
选择下拉框数据时的消息响应函数
|
|
|
"""
|
|
|
text = ''
|
|
|
idx = self.text_list.currentIndex()
|
|
|
if self.text_list.data:
|
|
|
text = self.text_list.data[idx.row()]
|
|
|
self.text_list.hide()
|
|
|
self.setText(text)
|
|
|
self.setFocus()
|
|
|
self.setCursorPosition(len(text))
|
|
|
# 选择数据后就隐藏了下拉框，设置点击标识位为False，下次点击才会显示下拉框
|
|
|
self.click_flag = False
|
|
|
if func:
|
|
|
func()
|
|
|
|
|
|
def on_search_data(self):
|
|
|
"""
|
|
|
文本变化时的消息响应，搜索下拉框列表里的数据
|
|
|
"""
|
|
|
# 输入框中文本和List选择文本不一样的时候，才显示下拉框
|
|
|
cur_text = self.text()
|
|
|
list_text = ''
|
|
|
idx = self.text_list.currentIndex()
|
|
|
if self.text_list.data:
|
|
|
list_text = self.text_list.data[idx.row()]
|
|
|
if cur_text != list_text: # and len(cur_text) > 1:
|
|
|
self.text_list.show()
|
|
|
|
|
|
# 根据消息响应传入的data来搜索
|
|
|
self.text_list.data = self.org_data[:]
|
|
|
for _text in self.org_data:
|
|
|
if cur_text not in _text:
|
|
|
self.text_list.data.remove(_text)
|
|
|
# 刷新下拉框
|
|
|
self.list_model.setStringList(self.text_list.data)
|
|
|
self.text_list.setModel(self.list_model)
|
|
|
|
|
|
if not self.click_flag:
|
|
|
self.text_list.hide()
|
|
|
self.click_flag = True
|
|
|
if self.text_list.data: # 有数据的时候就调整下拉框大小
|
|
|
self.resize_text_list()
|
|
|
else:
|
|
|
self.text_list.hide() # 没有数据就隐藏下拉框
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MainWindow(QMainWindow):
|
|
|
"""主窗口，继承了QMainWindow类"""
|
|
|
def __init__(self, name, title):
|
|
|
"""初始化类的成员变量"""
|
|
|
super(MainWindow, self).__init__()
|
|
|
self.w = 0
|
|
|
self.h = 0
|
|
|
self.init_ui(name, title) # 初始化UI界面
|
|
|
|
|
|
def init_ui(self, name, title):
|
|
|
"""初始化UI界面"""
|
|
|
self.w = 140
|
|
|
self.h = 100
|
|
|
|
|
|
self.setObjectName(name) # 设置主窗口对象的名称
|
|
|
self.setWindowTitle(title) # 设置主窗口显示的标题
|
|
|
self.resize(self.w, self.h) # 设置主窗口尺寸
|
|
|
|
|
|
self.custom_edit = CustomEdit(self, size=(10, 10, 120, 24),
|
|
|
name='custom_edit', search=True)
|
|
|
self.custom_edit.setPlaceholderText('我是自定义输入框')
|
|
|
|
|
|
data_list = [i * ('%s' % i) for i in range(15)]
|
|
|
self.custom_edit.fill_data(data_list)
|
|
|
# self.custom_edit = CustomEdit(self, size=(10, 10, 120, 24),
|
|
|
# name='custom_edit', search=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
import sys
|
|
|
app = QApplication(sys.argv)
|
|
|
n = 'CustomUI'
|
|
|
t = '自定义控件'
|
|
|
ex = MainWindow(n, t)
|
|
|
ex.show()
|
|
|
sys.exit(app.exec_())
|