main_gui.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833
  1. #!/usr/bin/env python
  2. # -*- encoding: utf-8 -*-
  3. '''
  4. @Contact : liuyuqi.gov@msn.cn
  5. @Time : 2023/04/27 02:55:59
  6. @License : Copyright © 2017-2022 liuyuqi. All Rights Reserved.
  7. @Desc : repo_sync GUI入口
  8. 打包说明:
  9. 使用PyInstaller可以将此应用打包为单个可执行文件:
  10. 1. 安装必要的包:
  11. pip install pyinstaller
  12. pip install -e . # 以开发模式安装当前包
  13. 2. 打包命令:
  14. pyinstaller --onefile --windowed --icon=icon.ico --add-data "repo_sync/config.yml;repo_sync" main_gui.py
  15. (如果有图标文件,可以通过--icon参数指定)
  16. 3. 打包后的可执行文件将位于dist目录中
  17. 注意:确保在打包前先安装repo_sync包,这样PyInstaller才能正确找到所有依赖。
  18. '''
  19. import sys
  20. import os
  21. import threading
  22. import subprocess
  23. import yaml
  24. try:
  25. from PyQt5.QtWidgets import (
  26. QApplication, QTabWidget, QWidget, QVBoxLayout, QHBoxLayout, QLabel,
  27. QRadioButton, QPushButton, QButtonGroup, QGroupBox, QMessageBox,
  28. QLineEdit, QScrollArea, QFileDialog, QFormLayout, QCheckBox, QTextEdit,
  29. QComboBox, QDialog, QDialogButtonBox, QGridLayout, QToolButton, QListWidget, QListWidgetItem,
  30. QSpacerItem
  31. )
  32. from PyQt5.QtCore import Qt, pyqtSignal, QObject
  33. HAS_QT = True
  34. except ImportError:
  35. HAS_QT = False
  36. print("PyQt5 not installed, GUI mode not available")
  37. # 直接导入repo_sync模块
  38. sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
  39. try:
  40. from repo_sync.repo_sync import RepoSync
  41. from repo_sync.version import __version__
  42. from repo_sync.utils.config_reader import ConfigReader
  43. except ImportError:
  44. print("无法导入repo_sync模块,尝试直接导入...")
  45. try:
  46. from repo_sync import RepoSync
  47. from version import __version__
  48. from utils.config_reader import ConfigReader
  49. except ImportError:
  50. print("导入repo_sync模块失败,请确保repo_sync已正确安装")
  51. __version__ = "未知"
  52. # 创建一个空的RepoSync类作为替代
  53. class RepoSync:
  54. def __init__(self, params):
  55. self.params = params
  56. def run(self):
  57. print("RepoSync模块未找到,无法执行操作")
  58. # 确保config.yml文件存在
  59. def ensure_config_file():
  60. config_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'config.yml')
  61. if not os.path.exists(config_file):
  62. # 创建空的config.yml文件
  63. default_config = {
  64. 'accounts': {
  65. 'github': {
  66. 'enable': 1,
  67. '1': {
  68. 'username': '',
  69. 'token': '',
  70. 'private': True
  71. }
  72. }
  73. },
  74. 'log': {
  75. 'level': 'debug',
  76. 'file': '/tmp/repo_sync.log',
  77. 'max_size': '100MB',
  78. 'max_backups': 3,
  79. 'max_age': 7,
  80. 'console_formatter': {
  81. 'level': 'debug',
  82. 'format': '%(asctime)s - %(levelname)s - %(message)s'
  83. },
  84. 'file_formatter': {
  85. 'level': 'debug',
  86. 'format': '%(asctime)s - %(levelname)s - %(message)s'
  87. }
  88. }
  89. }
  90. with open(config_file, 'w', encoding='utf-8') as f:
  91. yaml.dump(default_config, f, default_flow_style=False)
  92. return config_file
  93. class SettingsTab(QWidget):
  94. def __init__(self, parent=None):
  95. super().__init__(parent)
  96. # 确保config.yml文件存在
  97. ensure_config_file()
  98. # 初始化account_lists字典
  99. self.account_lists = {}
  100. self.config_reader = ConfigReader()
  101. self.init_ui()
  102. def init_ui(self):
  103. layout = QVBoxLayout()
  104. # 创建平台选择标签页
  105. self.platform_tabs = QTabWidget()
  106. # 平台配置
  107. self.platform_configs = {
  108. "github": ["username", "token", "private"],
  109. "gitlab": ["host", "username", "token", "private"],
  110. "gitee": ["username", "token", "private"],
  111. "gitcode": ["username", "token", "private"],
  112. "git.yoq.me": ["username", "token", "private"],
  113. "coding": ["username", "token", "project", "private"],
  114. "aliyun": ["compoanyid", "group_id", "username", "token", "private"],
  115. "cnb": ["username", "token", "private"]
  116. }
  117. # 为每个平台创建标签页
  118. self.platform_pages = {}
  119. for platform in self.platform_configs.keys():
  120. page = QWidget()
  121. page_layout = QVBoxLayout()
  122. page_layout.setSpacing(15) # 增加垂直间距
  123. # 创建水平布局,使账户列表和账户详情并排显示
  124. accounts_details_layout = QHBoxLayout()
  125. accounts_details_layout.setSpacing(20) # 增加水平间距
  126. # 账户列表
  127. account_group = QGroupBox("Accounts")
  128. account_layout = QVBoxLayout()
  129. account_layout.setSpacing(10) # 增加垂直间距
  130. account_list = QListWidget()
  131. account_list.setMinimumWidth(200) # 设置最小宽度
  132. self.account_lists[platform] = account_list
  133. account_list.currentItemChanged.connect(lambda current, previous, p=platform: self.select_account(p, current))
  134. account_buttons = QHBoxLayout()
  135. add_btn = QPushButton("Add Account")
  136. add_btn.clicked.connect(lambda checked, p=platform: self.add_account(p))
  137. delete_btn = QPushButton("Delete Account")
  138. delete_btn.clicked.connect(lambda checked, p=platform: self.delete_account(p))
  139. enable_btn = QPushButton("Enable Account")
  140. enable_btn.clicked.connect(lambda checked, p=platform: self.enable_account(p))
  141. account_buttons.addWidget(add_btn)
  142. account_buttons.addWidget(delete_btn)
  143. account_buttons.addWidget(enable_btn)
  144. account_layout.addWidget(account_list)
  145. account_layout.addLayout(account_buttons)
  146. account_group.setLayout(account_layout)
  147. # 账户详情表单
  148. form_group = QGroupBox("Account Details")
  149. form_layout = QFormLayout()
  150. form_layout.setSpacing(10) # 增加表单项之间的间距
  151. form_group.setLayout(form_layout)
  152. # 将账户列表和账户详情添加到水平布局中
  153. accounts_details_layout.addWidget(account_group, 1) # 1是伸缩因子
  154. accounts_details_layout.addWidget(form_group, 2) # 2是伸缩因子,使表单区域更宽
  155. # 添加水平布局到页面布局
  156. page_layout.addLayout(accounts_details_layout)
  157. # 保存按钮
  158. save_btn = QPushButton("Save Settings")
  159. save_btn.clicked.connect(self.save_settings)
  160. page_layout.addWidget(save_btn)
  161. page.setLayout(page_layout)
  162. self.platform_pages[platform] = {
  163. "form": form_layout
  164. }
  165. self.platform_tabs.addTab(page, platform)
  166. layout.addWidget(self.platform_tabs)
  167. self.setLayout(layout)
  168. # 加载设置
  169. self.load_settings()
  170. def load_settings(self):
  171. # 为每个平台加载账户
  172. for platform in self.platform_configs.keys():
  173. account_list = self.account_lists[platform]
  174. account_list.clear()
  175. # 获取该平台的所有账户
  176. accounts = self.config_reader.get_platform_accounts(platform)
  177. # 添加到列表
  178. for account in accounts:
  179. item = QListWidgetItem(account)
  180. # 如果是启用的账户(账户名为"1"),设置背景色以突出显示
  181. if account == "1":
  182. item.setBackground(Qt.green)
  183. item.setText(f"{account} (启用中)")
  184. account_list.addItem(item)
  185. # 选择第一个账户
  186. if account_list.count() > 0:
  187. account_list.setCurrentRow(0)
  188. self.select_account(platform, account_list.item(0))
  189. def select_account(self, platform, item):
  190. if not item:
  191. return
  192. account = item.text()
  193. # 如果账户名包含 "(启用中)" 后缀,去掉后缀
  194. if " (启用中)" in account:
  195. account = account.split(" (")[0]
  196. form = self.platform_pages[platform]["form"]
  197. # 清空表单
  198. while form.rowCount() > 0:
  199. form.removeRow(0)
  200. # 获取账户配置
  201. account_config = self.config_reader.get_account_config(platform, account)
  202. # 创建表单项
  203. self.field_widgets = {}
  204. for field in self.platform_configs[platform]:
  205. value = account_config.get(field, "")
  206. if field == "private":
  207. widget = QCheckBox()
  208. widget.setChecked(value if isinstance(value, bool) else value.lower() != "false")
  209. else:
  210. widget = QLineEdit()
  211. if field in ["token", "password"]:
  212. widget.setEchoMode(QLineEdit.Password)
  213. widget.setText(str(value))
  214. self.field_widgets[field] = widget
  215. form.addRow(f"{field.capitalize()}:", widget)
  216. def add_account(self, platform):
  217. dialog = AddAccountDialog(platform, self)
  218. if dialog.exec_() == QDialog.Accepted:
  219. account_data = dialog.get_account_data()
  220. account_name = account_data["name"]
  221. if not account_name:
  222. QMessageBox.warning(self, "Warning", "Account name cannot be empty.")
  223. return
  224. # 更新config.yml文件
  225. config = self.config_reader.config
  226. if platform not in config['accounts']:
  227. config['accounts'][platform] = {'enable': 1}
  228. # 添加新账户
  229. config['accounts'][platform][account_name] = {
  230. field: value for field, value in account_data.items() if field != 'name'
  231. }
  232. # 保存配置
  233. with open(self.config_reader.config_path, 'w', encoding='utf-8') as f:
  234. yaml.dump(config, f, default_flow_style=False)
  235. # 重新加载设置
  236. self.config_reader = ConfigReader()
  237. self.load_settings()
  238. # 选择新账户
  239. account_list = self.account_lists[platform]
  240. for i in range(account_list.count()):
  241. if account_list.item(i).text() == account_name:
  242. account_list.setCurrentRow(i)
  243. self.select_account(platform, account_list.item(i))
  244. break
  245. def delete_account(self, platform):
  246. account_list = self.account_lists[platform]
  247. current_item = account_list.currentItem()
  248. if not current_item:
  249. return
  250. account = current_item.text()
  251. # 如果账户名包含 "(启用中)" 后缀,去掉后缀
  252. if " (启用中)" in account:
  253. account = account.split(" (")[0]
  254. reply = QMessageBox.question(
  255. self,
  256. "Confirm Deletion",
  257. f"Are you sure you want to delete the account '{account}' for {platform}?",
  258. QMessageBox.Yes | QMessageBox.No,
  259. QMessageBox.No
  260. )
  261. if reply == QMessageBox.Yes:
  262. # 更新config.yml文件
  263. config = self.config_reader.config
  264. if platform in config['accounts'] and account in config['accounts'][platform]:
  265. del config['accounts'][platform][account]
  266. # 保存配置
  267. with open(self.config_reader.config_path, 'w', encoding='utf-8') as f:
  268. yaml.dump(config, f, default_flow_style=False)
  269. # 重新加载设置
  270. self.config_reader = ConfigReader()
  271. self.load_settings()
  272. def enable_account(self, platform):
  273. account_list = self.account_lists[platform]
  274. current_item = account_list.currentItem()
  275. if not current_item:
  276. return
  277. account = current_item.text()
  278. # 如果账户名包含 "(启用中)" 后缀,去掉后缀
  279. if " (启用中)" in account:
  280. account = account.split(" (")[0]
  281. # 更新config.yml文件
  282. config = self.config_reader.config
  283. if platform in config['accounts']:
  284. # 将当前账户的配置复制到第一个账户位置
  285. account_config = config['accounts'][platform][account]
  286. config['accounts'][platform]['1'] = account_config
  287. # 保存配置
  288. with open(self.config_reader.config_path, 'w', encoding='utf-8') as f:
  289. yaml.dump(config, f, default_flow_style=False)
  290. QMessageBox.information(
  291. self,
  292. "Success",
  293. f"Account '{account}' has been enabled for {platform}."
  294. )
  295. # 重新加载设置
  296. self.config_reader = ConfigReader()
  297. self.load_settings()
  298. def save_settings(self):
  299. # 保存当前显示的账户配置
  300. platform = list(self.platform_configs.keys())[self.platform_tabs.currentIndex()]
  301. account_list = self.account_lists[platform]
  302. current_item = account_list.currentItem()
  303. if current_item and hasattr(self, 'field_widgets'):
  304. account = current_item.text()
  305. # 如果账户名包含 "(启用中)" 后缀,去掉后缀
  306. if " (启用中)" in account:
  307. account = account.split(" (")[0]
  308. # 更新config.yml文件
  309. config = self.config_reader.config
  310. if platform not in config['accounts']:
  311. config['accounts'][platform] = {'enable': 1}
  312. # 更新账户配置
  313. account_config = {}
  314. for field, widget in self.field_widgets.items():
  315. if isinstance(widget, QCheckBox):
  316. value = widget.isChecked()
  317. else:
  318. value = widget.text().strip()
  319. if value:
  320. account_config[field] = value
  321. config['accounts'][platform][account] = account_config
  322. # 保存配置
  323. with open(self.config_reader.config_path, 'w', encoding='utf-8') as f:
  324. yaml.dump(config, f, default_flow_style=False)
  325. QMessageBox.information(self, "Success", "Settings saved successfully!")
  326. class AddAccountDialog(QDialog):
  327. def __init__(self, platform, parent=None):
  328. super().__init__(parent)
  329. self.platform = platform
  330. self.init_ui()
  331. def init_ui(self):
  332. self.setWindowTitle(f"Add {self.platform} Account")
  333. layout = QFormLayout()
  334. layout.setSpacing(10) # 增加表单项之间的间距
  335. # 账户名称
  336. self.name_edit = QLineEdit()
  337. layout.addRow("Account Name:", self.name_edit)
  338. # 其他字段
  339. self.field_widgets = {}
  340. platform_configs = {
  341. "github": ["username", "token", "private"],
  342. "gitlab": ["host", "username", "token", "private"],
  343. "gitee": ["username", "token", "private"],
  344. "gitcode": ["username", "token", "private"],
  345. "git.yoq.me": ["username", "token", "private"],
  346. "coding": ["username", "token", "project", "private"],
  347. "aliyun": ["compoanyid", "group_id", "username", "token", "private"],
  348. "cnb": ["username", "token", "private"]
  349. }
  350. for field in platform_configs[self.platform]:
  351. if field == "private":
  352. widget = QCheckBox()
  353. widget.setChecked(True)
  354. else:
  355. widget = QLineEdit()
  356. if field in ["token", "password"]:
  357. widget.setEchoMode(QLineEdit.Password)
  358. self.field_widgets[field] = widget
  359. layout.addRow(f"{field.capitalize()}:", widget)
  360. # 添加一些空间
  361. layout.addItem(QSpacerItem(20, 20))
  362. # 按钮
  363. buttons = QDialogButtonBox(
  364. QDialogButtonBox.Ok | QDialogButtonBox.Cancel,
  365. Qt.Horizontal, self)
  366. buttons.accepted.connect(self.accept)
  367. buttons.rejected.connect(self.reject)
  368. layout.addRow(buttons)
  369. self.setLayout(layout)
  370. self.setMinimumWidth(400) # 设置对话框最小宽度
  371. def get_account_data(self):
  372. data = {"name": self.name_edit.text().strip()}
  373. for field, widget in self.field_widgets.items():
  374. if isinstance(widget, QCheckBox):
  375. data[field] = widget.isChecked()
  376. else:
  377. data[field] = widget.text().strip()
  378. return data
  379. class MainTab(QWidget):
  380. def __init__(self, parent=None):
  381. super().__init__(parent)
  382. layout = QVBoxLayout()
  383. # 路径选择
  384. path_layout = QHBoxLayout()
  385. self.path_label = QLabel("Local Path: ")
  386. self.path_edit = QLineEdit()
  387. self.path_edit.setText(get_active_explorer_path())
  388. self.path_btn = QPushButton("Browse...")
  389. self.path_btn.clicked.connect(self.choose_path)
  390. path_layout.addWidget(self.path_label)
  391. path_layout.addWidget(self.path_edit)
  392. path_layout.addWidget(self.path_btn)
  393. layout.addLayout(path_layout)
  394. # 操作
  395. op_group = QGroupBox("Operation:")
  396. op_layout = QHBoxLayout()
  397. self.op_buttons = QButtonGroup(self)
  398. for i, op in enumerate(["create", "push", "pull", "clone", "delete"]):
  399. btn = QRadioButton(op.capitalize())
  400. if i == 1:
  401. btn.setChecked(True)
  402. self.op_buttons.addButton(btn, i)
  403. op_layout.addWidget(btn)
  404. op_group.setLayout(op_layout)
  405. layout.addWidget(op_group)
  406. # 平台
  407. pf_group = QGroupBox("Platform:")
  408. pf_layout = QHBoxLayout()
  409. self.pf_buttons = QButtonGroup(self)
  410. self.platforms = ["github", "gitlab", "gitee", "gitcode", "git.yoq.me", "coding", "aliyun", "cnb"]
  411. for i, pf in enumerate(self.platforms):
  412. btn = QRadioButton(pf.capitalize())
  413. if i == 0:
  414. btn.setChecked(True)
  415. self.pf_buttons.addButton(btn, i)
  416. pf_layout.addWidget(btn)
  417. pf_group.setLayout(pf_layout)
  418. layout.addWidget(pf_group)
  419. # 账户选择
  420. account_layout = QHBoxLayout()
  421. account_layout.addWidget(QLabel("Account:"))
  422. self.account_combo = QComboBox()
  423. self.account_combo.setMinimumWidth(200)
  424. account_layout.addWidget(self.account_combo)
  425. account_layout.addStretch()
  426. layout.addLayout(account_layout)
  427. # 按钮区域 (run和cancel)
  428. btn_layout = QHBoxLayout()
  429. self.run_btn = QPushButton("run it")
  430. self.run_btn.clicked.connect(self.run_repo_sync)
  431. self.cancel_btn = QPushButton("Cancel")
  432. self.cancel_btn.clicked.connect(self.cancel_operation)
  433. self.cancel_btn.setEnabled(False)
  434. btn_layout.addWidget(self.run_btn)
  435. btn_layout.addWidget(self.cancel_btn)
  436. layout.addLayout(btn_layout)
  437. # 命令执行结果
  438. result_group = QGroupBox("Execution Result:")
  439. result_layout = QVBoxLayout()
  440. self.result_text = QTextEdit()
  441. self.result_text.setReadOnly(True)
  442. result_layout.addWidget(self.result_text)
  443. result_group.setLayout(result_layout)
  444. layout.addWidget(result_group)
  445. self.setLayout(layout)
  446. # 命令执行相关
  447. self.process = None
  448. self.command_signals = CommandSignals()
  449. self.command_signals.output.connect(self.update_output)
  450. self.command_signals.finished.connect(self.process_finished)
  451. # 初始化配置读取器
  452. self.config_reader = ConfigReader()
  453. # 平台变更时更新账户列表
  454. self.pf_buttons.buttonClicked.connect(self.update_account_list)
  455. self.update_account_list()
  456. def choose_path(self):
  457. path = QFileDialog.getExistingDirectory(self, "Select Directory")
  458. if path:
  459. self.path_edit.setText(path)
  460. def update_account_list(self):
  461. self.account_combo.clear()
  462. pf_id = self.pf_buttons.checkedId()
  463. platform = self.platforms[pf_id]
  464. # 读取所有账户
  465. accounts = self.get_platform_accounts(platform)
  466. # 找出启用的账户
  467. enabled_account = "1"
  468. # 添加账户到下拉框,启用的账户放在最前面
  469. if enabled_account in accounts:
  470. accounts.remove(enabled_account)
  471. self.account_combo.addItem(f"{enabled_account} (启用中)")
  472. # 设置启用账户的背景颜色为绿色
  473. model = self.account_combo.model()
  474. index = model.index(0, 0)
  475. model.setData(index, Qt.green, Qt.BackgroundRole)
  476. for account in accounts:
  477. self.account_combo.addItem(account)
  478. # 默认选择启用的账户
  479. self.account_combo.setCurrentIndex(0)
  480. def get_platform_accounts(self, platform):
  481. # 读取config.yml文件中的所有配置
  482. accounts = set()
  483. # 获取该平台的所有账户
  484. try:
  485. platform_accounts = self.config_reader.get_platform_accounts(platform)
  486. for account in platform_accounts:
  487. accounts.add(account)
  488. except Exception as e:
  489. print(f"Error getting platform accounts: {str(e)}")
  490. # 确保至少有一个默认账户
  491. if not accounts:
  492. accounts.add("1")
  493. return sorted(list(accounts))
  494. def run_repo_sync(self):
  495. repo_path = self.path_edit.text().strip()
  496. if not repo_path:
  497. QMessageBox.warning(self, "Warning", "Please select a local path.")
  498. return
  499. op_id = self.op_buttons.checkedId()
  500. pf_id = self.pf_buttons.checkedId()
  501. op = ["create", "push", "pull", "clone", "delete"][op_id]
  502. pf = self.platforms[pf_id]
  503. # 获取选择的账户名(去掉可能的"(启用中)"后缀)
  504. account_text = self.account_combo.currentText()
  505. account = account_text.split(" (")[0] if " (" in account_text else account_text
  506. # 清空结果区域
  507. self.result_text.clear()
  508. # 检查平台配置
  509. account_config = self.config_reader.get_account_config(pf, account)
  510. if not account_config.get('token'):
  511. QMessageBox.warning(self, "Warning", f"Please configure {pf} token for account '{account}' in Settings tab first.")
  512. return
  513. # 构建参数
  514. args = {
  515. 'command': op,
  516. 'platform': pf,
  517. 'repo_path': repo_path
  518. }
  519. # 如果不是默认账户,需要设置环境变量
  520. if account != "1":
  521. # 保存原始环境变量
  522. self.original_env = {}
  523. for field, value in account_config.items():
  524. env_var = f"{pf}_{field}"
  525. if env_var in os.environ:
  526. self.original_env[env_var] = os.environ[env_var]
  527. os.environ[env_var] = str(value)
  528. # 显示执行命令
  529. cmd_str = f"repo_sync {op} -p {pf} -repo_path {repo_path}"
  530. self.result_text.append(f"Running: {cmd_str}\n")
  531. # 执行命令
  532. self.run_btn.setEnabled(False)
  533. self.cancel_btn.setEnabled(True)
  534. # 重定向标准输出和标准错误
  535. self.original_stdout = sys.stdout
  536. self.original_stderr = sys.stderr
  537. sys.stdout = self.OutputRedirector(self.command_signals)
  538. sys.stderr = self.OutputRedirector(self.command_signals)
  539. # 在新线程中执行命令
  540. self.process_thread = threading.Thread(
  541. target=self.run_module,
  542. args=(args,)
  543. )
  544. self.process_thread.daemon = True
  545. self.process_thread.start()
  546. def run_module(self, args):
  547. try:
  548. # 正确导入repo_sync模块的main函数
  549. from repo_sync.repo_sync import RepoSync
  550. self.running = True
  551. return_code = 0
  552. try:
  553. # 直接创建RepoSync实例并运行
  554. rs = RepoSync(args)
  555. rs.run()
  556. except SystemExit as e:
  557. return_code = e.code if isinstance(e.code, int) else 1
  558. except Exception as e:
  559. self.command_signals.output.emit(f"Error: {str(e)}")
  560. return_code = 1
  561. finally:
  562. self.command_signals.finished.emit(return_code)
  563. # 恢复标准输出和标准错误
  564. sys.stdout = self.original_stdout
  565. sys.stderr = self.original_stderr
  566. # 恢复环境变量
  567. if hasattr(self, 'original_env'):
  568. for var, value in self.original_env.items():
  569. os.environ[var] = value
  570. delattr(self, 'original_env')
  571. except Exception as e:
  572. self.command_signals.output.emit(f"Error: {str(e)}")
  573. self.command_signals.finished.emit(1)
  574. # 恢复标准输出和标准错误
  575. sys.stdout = self.original_stdout
  576. sys.stderr = self.original_stderr
  577. # 输出重定向类
  578. class OutputRedirector:
  579. def __init__(self, signals):
  580. self.signals = signals
  581. self.buffer = ""
  582. def write(self, text):
  583. self.buffer += text
  584. if '\n' in self.buffer:
  585. lines = self.buffer.split('\n')
  586. for line in lines[:-1]:
  587. self.signals.output.emit(line)
  588. self.buffer = lines[-1]
  589. def flush(self):
  590. if self.buffer:
  591. self.signals.output.emit(self.buffer)
  592. self.buffer = ""
  593. def run_process(self, cmd, env=None):
  594. try:
  595. self.process = subprocess.Popen(
  596. cmd,
  597. stdout=subprocess.PIPE,
  598. stderr=subprocess.STDOUT,
  599. text=True,
  600. bufsize=1,
  601. universal_newlines=True,
  602. env=env
  603. )
  604. # 读取输出
  605. for line in self.process.stdout:
  606. self.command_signals.output.emit(line)
  607. self.process.wait()
  608. self.command_signals.finished.emit(self.process.returncode)
  609. except Exception as e:
  610. self.command_signals.output.emit(f"Error: {str(e)}")
  611. self.command_signals.finished.emit(1)
  612. def update_output(self, text):
  613. self.result_text.append(text)
  614. # 自动滚动到底部
  615. self.result_text.verticalScrollBar().setValue(
  616. self.result_text.verticalScrollBar().maximum()
  617. )
  618. def process_finished(self, return_code):
  619. self.process = None
  620. self.run_btn.setEnabled(True)
  621. self.cancel_btn.setEnabled(False)
  622. if return_code == 0:
  623. self.result_text.append("\nOperation completed successfully.")
  624. else:
  625. self.result_text.append(f"\nOperation failed with return code {return_code}.")
  626. def cancel_operation(self):
  627. if self.process:
  628. self.process.terminate()
  629. self.result_text.append("\nOperation cancelled by user.")
  630. else:
  631. # 直接调用模式下,通过退出标志通知线程结束
  632. self.running = False
  633. self.result_text.append("\nOperation cancelled by user.")
  634. # 恢复标准输出和标准错误
  635. if hasattr(self, 'original_stdout') and hasattr(self, 'original_stderr'):
  636. sys.stdout = self.original_stdout
  637. sys.stderr = self.original_stderr
  638. # 恢复环境变量
  639. if hasattr(self, 'original_env'):
  640. for var, value in self.original_env.items():
  641. os.environ[var] = value
  642. delattr(self, 'original_env')
  643. # Explorer路径获取
  644. try:
  645. import win32com.client
  646. def get_active_explorer_path():
  647. shell = win32com.client.Dispatch("Shell.Application")
  648. for window in shell.Windows():
  649. if window.Name in ["文件资源管理器", "Windows Explorer"]:
  650. return window.Document.Folder.Self.Path
  651. return os.getcwd()
  652. except ImportError:
  653. def get_active_explorer_path():
  654. return os.getcwd()
  655. # 命令执行信号类
  656. class CommandSignals(QObject):
  657. output = pyqtSignal(str)
  658. finished = pyqtSignal(int)
  659. class AboutTab(QWidget):
  660. def __init__(self, parent=None):
  661. super().__init__(parent)
  662. layout = QVBoxLayout()
  663. layout.addWidget(QLabel(f"repo_sync tools v{__version__}"))
  664. layout.addWidget(QLabel("作者: liuyuqi.gov@msn.cn"))
  665. layout.addWidget(QLabel("GitHub: https://github.com/jianboy/repo_sync"))
  666. layout.addWidget(QLabel("\n功能说明:"))
  667. layout.addWidget(QLabel("- 支持多个代码托管平台"))
  668. layout.addWidget(QLabel("- 支持创建/推送/拉取/克隆/删除操作"))
  669. layout.addWidget(QLabel("- 自动获取资源管理器当前路径"))
  670. layout.addWidget(QLabel("- 配置信息保存在config.yml文件中"))
  671. layout.addWidget(QLabel("- 支持每个平台配置多个账户"))
  672. layout.addWidget(QLabel("- 命令行执行结果实时显示"))
  673. self.setLayout(layout)
  674. class RepoSyncMainWindow(QTabWidget):
  675. def __init__(self):
  676. super().__init__()
  677. self.setWindowTitle('repo_sync tools v1.12')
  678. self.resize(800, 700)
  679. self.main_tab = MainTab()
  680. self.addTab(self.main_tab, '主界面')
  681. self.settings_tab = SettingsTab()
  682. self.addTab(self.settings_tab, '设置')
  683. self.about_tab = AboutTab()
  684. self.addTab(self.about_tab, '关于')
  685. def main():
  686. """GUI主入口函数"""
  687. if not HAS_QT:
  688. print("PyQt5 not installed. Please install it with: pip install PyQt5")
  689. print("Running in fallback mode...")
  690. # 这里可以添加一个简单的命令行界面作为后备
  691. return
  692. try:
  693. # 确保config.yml文件存在
  694. ensure_config_file()
  695. app = QApplication(sys.argv)
  696. window = RepoSyncMainWindow()
  697. window.show()
  698. sys.exit(app.exec_())
  699. except Exception as e:
  700. print(f"Error starting GUI: {str(e)}")
  701. import traceback
  702. traceback.print_exc()
  703. if __name__ == '__main__':
  704. main()