admin/DeepSeekSelfTool
1
0
Fork 0
mirror of https://github.com/ChinaRan0/DeepSeekSelfTool.git synced 2025-11-15 15:26:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ollamaMain.py

Browse Source
This commit is contained in:
宗宸·谢尔比 2025-02-11 11:39:05 +08:00 committed by GitHub
parent 1cbe81a6dd
commit f7fa670111
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ollamaMain.py
View File

@ -171,6 +171,7 @@ class CyberScanner(QtWidgets.QMainWindow):
left_layout.addWidget(self.lbl_path) left_layout.addWidget(self.lbl_path)
# 模式选择 # 模式选择
# 模式选择
mode_group = QtWidgets.QGroupBox("🔧 检测模式") mode_group = QtWidgets.QGroupBox("🔧 检测模式")
mode_group.setStyleSheet(""" mode_group.setStyleSheet("""
QGroupBox { QGroupBox {
@ -193,6 +194,14 @@ class CyberScanner(QtWidgets.QMainWindow):
mode_group.setLayout(mode_layout) mode_group.setLayout(mode_layout)
left_layout.addWidget(mode_group) left_layout.addWidget(mode_group)
# 新增：是否审计 JavaScript 文件的复选框
self.checkbox_audit_js = QtWidgets.QCheckBox("审计 静态 文件")
self.checkbox_audit_js.setChecked(True) # 默认选中
self.checkbox_audit_js.setStyleSheet("""
QCheckBox { color: #00ff00; padding: 8px; }
QCheckBox::indicator { width: 20px; height: 20px; }
""")
left_layout.addWidget(self.checkbox_audit_js)
# 文件树 # 文件树
self.file_tree = QtWidgets.QTreeView() self.file_tree = QtWidgets.QTreeView()
self.file_model = QtWidgets.QFileSystemModel() self.file_model = QtWidgets.QFileSystemModel()
@ -292,6 +301,12 @@ class CyberScanner(QtWidgets.QMainWindow):
def scan_code_files(self, directory): def scan_code_files(self, directory):
allowed_ext = ['.php', '.jsp', '.asp', '.js', '.html', '.py', '.java'] allowed_ext = ['.php', '.jsp', '.asp', '.js', '.html', '.py', '.java']
# 如果用户选择不审计 静态 文件，则从允许的扩展名中移除 .js
if not self.checkbox_audit_js.isChecked():
allowed_ext.remove('.js')
allowed_ext.remove('.html')
code_files = {} code_files = {}
for root, _, files in os.walk(directory): for root, _, files in os.walk(directory):