def delete_all_record(self):
listModel = QStringListModel()
listModel.setStringList([])
self.MainWindow.requestRecord.setModel(listModel)
all_record = read_all_record()
all_record['api'] = {}
write_all_record(all_record)
def delete_record(self, index):
res = QMessageBox.question(self.MainWindow, '确认', '确认删除?', QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
if res == QMessageBox.Yes:
self.MainWindow.requestRecord.model().removeRow(index)
uuid = self.record_index.pop(index)
self.record_data.pop(index)
all_record = read_all_record()
if all_record:
del all_record['api'][uuid]
write_all_record(all_record)
else:
QMessageBox.warning(self.MainWindow, '警告', '小伙子，不要乱搞哦！')
def save_new_ip(self, **kwargs):
# 去重
for i in self.read_history_ip():
if i['ip'] == kwargs['ip']:
return
if not self.Window.autoGetIp.isChecked(
) and not self.Window.autoGetDns.isChecked():
history_ip_list = self.read_history_ip()
history_ip_list.insert(0, kwargs)
all_record = read_all_record()
all_record['ip'] = history_ip_list
write_all_record(all_record)
self.load_ip()
def load_request_record(self):
"""将请求记录重新加载到ListView"""
all_recode = read_all_record()
if all_recode.get('api'):
for i in all_recode['api']:
url = all_recode['api'][i]['url']
method = all_recode['api'][i]['method']
params = all_recode['api'][i]['params']
# 保存加载到ListView中的索引
self.record_index.append(method + '_' + url + "_" + params)
# 索引对应的数据
self.record_data.append(dict(url=url, method=method, params=params))
# 创建列表字符串对象，跟QListView配合使用
listModel = QStringListModel()
listModel.setStringList(self.record_index)
self.MainWindow.requestRecord.setContextMenuPolicy(3)
self.MainWindow.requestRecord.setModel(listModel)
# 未每个item加入右键功能
self.MainWindow.requestRecord.customContextMenuRequested[QPoint].connect(self.list_widget_menu)
def save_request(self):
"""保存请求记录"""
# 读取请求记录
all_record = read_all_record()
# 生成此次请求的唯一ID
uuid = self.method + "_" + self.url + "_" + self.MainWindow.parameter.toPlainText()
# 相同的请求不保存
if all_record['api'].get(uuid):
return
# 保存新请求
all_record['api'][uuid] = dict(
method=self.method,
url=self.url,
params=self.MainWindow.parameter.toPlainText()
)
# 添加至索引列表
self.record_index.insert(0, uuid)
# 更新保存的请求记录
write_all_record(all_record)
self.MainWindow.requestRecord.model().setData(1, self.url, 0)
def read_history_ip():
if not os.path.exists('static'):
os.mkdir("static")
if not os.path.exists('static/config.json'):
return []
return read_all_record().get('ip', [])