def initUI(self):
self.setWindowTitle("QTableWidget 例子")
self.resize(430, 300);
conLayout = QHBoxLayout()
tableWidget = QTableWidget()
tableWidget.setRowCount(4)
tableWidget.setColumnCount(3)
conLayout.addWidget(tableWidget)
tableWidget.setHorizontalHeaderLabels(['姓名', '性别', '体重(kg)'])
newItem = QTableWidgetItem("张三")
tableWidget.setItem(0, 0, newItem)
comBox = QComboBox()
comBox.addItem("男")
comBox.addItem("女")
comBox.setStyleSheet("QComboBox{margin:3px};")
tableWidget.setCellWidget(0, 1, comBox)
searchBtn = QPushButton("修改")
searchBtn.setDown(True)
searchBtn.setStyleSheet("QPushButton{margin:3px};")
tableWidget.setCellWidget(0, 2, searchBtn)
self.setLayout(conLayout)
class App(widgets.QWidget):
def __init__(self):
super().__init__()
self.button = widgets.QPushButton('Request from webservice')
self.text = widgets.QLabel('aiplayground',
alignment=core.Qt.AlignCenter)
self.layout = widgets.QVBoxLayout()
self.layout.addWidget(self.text)
self.layout.addWidget(self.button)
self.setLayout(self.layout)
self.button.clicked.connect(self.magic)
self.api_url = 'http://www.lennartberning.de/aiplayground/api/moods?'
def createTable(self, data):
#Not proud about the following, but it serves the purpose...
data = data.strip('][').split(',')
self.tableWidget = QTableWidget()
self.tableWidget.setColumnCount(3)
self.tableWidget.setRowCount(len(data) / 3)
self.tableWidget.setHorizontalHeaderLabels(
['Mood', 'Is-Positive', 'Timestamp'])
p = 0
for r, list in enumerate(data):
for i in range(3):
if p < len(data):
self.tableWidget.setItem(
r, i,
QTableWidgetItem(
re.sub('[""]', '',
data[p]).replace('[', '').replace(']', '')))
p = p + 1
self.tableWidget.show()
@core.Slot()
def magic(self):
response = self.call_webservice(self.api_url, 'sentiment', 0)
self.text.setText(str(response.status_code))
self.createTable(response.text)
@staticmethod
def call_webservice(api_url, key, direction):
url = '{0}{1}={2}'.format(api_url, key, direction)
r = requests.get(url)
if r.status_code == 200:
return r
else:
return 'Error contacting webservice. Please check connectivity!'
def update_hints_text(
game: RandovaniaGame,
hint_item_names_tree_widget: QtWidgets.QTableWidget,
):
item_database = default_database.item_database_for_game(game)
rows = []
for item in item_database.major_items.values():
rows.append((
item.name,
item.item_category.hint_details[1],
item.item_category.general_details[1],
item.broad_category.hint_details[1],
))
for name, item_category, broad_category in _GAME_SPECIFIC.get(
game, lambda: [])():
rows.append((
name,
item_category.hint_details[1],
item_category.general_details[1],
broad_category.hint_details[1],
))
for ammo in item_database.ammo.values():
rows.append((
ammo.name,
ammo.item_category.hint_details[1],
ammo.item_category.general_details[1],
ammo.broad_category.hint_details[1],
))
hint_item_names_tree_widget.setSortingEnabled(False)
hint_item_names_tree_widget.setRowCount(len(rows))
for i, elements in enumerate(rows):
for j, element in enumerate(elements):
hint_item_names_tree_widget.setItem(
i, j, QtWidgets.QTableWidgetItem(element))
for i in range(4):
hint_item_names_tree_widget.resizeColumnToContents(i)
hint_item_names_tree_widget.setSortingEnabled(True)
class Widget(QWidget):
def __init__(self):
QWidget.__init__(self)
self.items = 0
# Example data
self._data = {
"Water": 24.5,
"Electricity": 55.1,
"Rent": 850.0,
"Supermarket": 230.4,
"Internet": 29.99,
"Spätkauf": 21.85,
"BVG Ticket": 60.0,
"Coffee": 22.45,
"Meetup": 0.0
}
# Left
self.table = QTableWidget()
self.table.setColumnCount(2)
self.table.setHorizontalHeaderLabels(["Description", "Quantity"])
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# QWidget Layout
self.layout = QHBoxLayout()
self.layout.addWidget(self.table)
# Set the layout to the QWidget
self.setLayout(self.layout)
# Fill example data
self.fill_table()
def fill_table(self, data=None):
data = self._data if not data else data
for desc, price in data.items():
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(desc))
self.table.setItem(self.items, 1, QTableWidgetItem(str(price)))
self.items += 1
class myWindow(QMainWindow):
def __init__(self) -> None:
super().__init__()
self.resize(500, 300)
# 非组件对象
self.values = {}
self.ut = UpdateValues()
# 界面组件对象
self.centralwidget = QWidget(self)
self.table = QTableWidget(self.centralwidget)
self.startButton = QPushButton(self.centralwidget, text='开始')
self.stopButton = QPushButton(self.centralwidget, text='停止')
self.stopButton.setDisabled(True)
self.table.setColumnCount(3)
self.table.setHorizontalHeaderItem(0, QTableWidgetItem('GID'))
self.table.setHorizontalHeaderItem(2, QTableWidgetItem('name'))
self.table.setHorizontalHeaderItem(1, QTableWidgetItem('speed'))
self.table.horizontalHeader().setStretchLastSection(True)
self.table.setRowCount(1)
self.layout = QGridLayout(self.centralwidget)
self.layout.addWidget(self.startButton)
self.layout.addWidget(self.stopButton)
self.layout.addWidget(self.table)
self.setCentralWidget(self.centralwidget)
self.ut.resultReady.connect(self.on_changeValue)
self.startButton.clicked.connect(self.changeValue)
self.stopButton.clicked.connect(self.stopUpdate)
def changeValue(self):
self.startButton.setDisabled(True)
self.stopButton.setEnabled(True)
self.ut.isRun = 1
self.ut.start()
def stopUpdate(self):
self.ut.isRun = 0
self.startButton.setEnabled(True)
self.stopButton.setDisabled(True)
@Slot()
def on_changeValue(self, values: dict):
print(values)
self.table.setItem(0, 0, QTableWidgetItem(str(values['gid'])))
self.table.setItem(0, 1, QTableWidgetItem(str(values['file'])))
self.table.setItem(0, 2, QTableWidgetItem(str(values['speed'])))
def initUI(self):
self.setWindowTitle("QTableWidget 例子")
self.resize(430, 230)
conLayout = QHBoxLayout()
tableWidget = QTableWidget()
tableWidget.setRowCount(4)
tableWidget.setColumnCount(3)
conLayout.addWidget(tableWidget)
tableWidget.setHorizontalHeaderLabels(['姓名', '性别', '体重(kg)'])
newItem = QTableWidgetItem("张三")
tableWidget.setItem(0, 0, newItem)
newItem = QTableWidgetItem("男")
tableWidget.setItem(0, 1, newItem)
newItem = QTableWidgetItem("80")
tableWidget.setItem(0, 2, newItem)
# 将表格变为禁止编辑
# tableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)
# 设置表格为整行选择
# tableWidget.setSelectionBehavior( QAbstractItemView.SelectRows)
# 将行和列的大小设为与内容相匹配
# tableWidget.resizeColumnsToContents()
# tableWidget.resizeRowsToContents()
# 表格表头的显示与隐藏
# tableWidget.verticalHeader().setVisible(False)
# tableWidget.horizontalHeader().setVisible(False)
# 不显示表格单元格的分割线
# tableWidget.setShowGrid(False)
# 不显示垂直表头
tableWidget.verticalHeader().setVisible(False)
self.setLayout(conLayout)
def initUI(self):
self.setWindowTitle("QTableWidget 例子")
self.resize(430, 230)
conLayout = QHBoxLayout()
tableWidget = QTableWidget()
tableWidget.setRowCount(4)
tableWidget.setColumnCount(3)
conLayout.addWidget(tableWidget)
tableWidget.setHorizontalHeaderLabels(['姓名', '性别', '体重(kg)'])
tableWidget.setSpan(0, 0, 3, 1)
newItem = QTableWidgetItem("张三")
tableWidget.setItem(0, 0, newItem)
newItem = QTableWidgetItem("男")
tableWidget.setItem(0, 1, newItem)
newItem = QTableWidgetItem("160")
tableWidget.setItem(0, 2, newItem)
self.setLayout(conLayout)
def initUI(self):
self.setWindowTitle("QTableWidget 例子")
self.resize(430, 230);
conLayout = QHBoxLayout()
tableWidget = QTableWidget()
tableWidget.setRowCount(4)
tableWidget.setColumnCount(3)
conLayout.addWidget(tableWidget)
tableWidget.setHorizontalHeaderLabels(['姓名', '性别', '体重(kg)'])
newItem = QTableWidgetItem("张三")
newItem.setForeground(QBrush(QColor(255, 0, 0)))
tableWidget.setItem(0, 0, newItem)
newItem = QTableWidgetItem("男")
newItem.setForeground(QBrush(QColor(255, 0, 0)))
tableWidget.setItem(0, 1, newItem)
newItem = QTableWidgetItem("160")
newItem.setForeground(QBrush(QColor(255, 0, 0)))
tableWidget.setItem(0, 2, newItem)
self.setLayout(conLayout)
class Buscador(QWidget):
status_signal = Signal(str)
def __init__(self):
self.db = ClientesDB()
QWidget.__init__(self)
Font = QFont()
Font.setBold(True) # Labels em Negrito
# Entry:
self.entry_nome = QLineEdit()
self.entry_nome.setText("Nome para Busca")
# Botões
self.button_busca = QPushButton("&Busca")
self.button_busca.clicked.connect(self.buscar)
self.button_busca.setShortcut("Ctrl+B")
self.button_limpar = QPushButton("Limpar")
self.button_limpar.clicked.connect(self.limpar)
self.button_limpar.setShortcut("ESC")
# Tabela
self.clientes = 0
self.tabela_clientes = QTableWidget()
self.tabela_clientes.setColumnCount(4)
self.tabela_clientes.setHorizontalHeaderLabels([
"Nome",
"Número",
"CPF",
"Endereço",
])
self.tabela_clientes.horizontalHeader().setSectionResizeMode(
QHeaderView.ResizeToContents)
self.tabela_clientes.horizontalHeader().setStretchLastSection(True)
self.tabela_clientes.resizeColumnsToContents()
self.tabela_clientes.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.tabela_clientes.itemDoubleClicked.connect(self.info_cliente)
#Leiaute:
self.layout = QVBoxLayout()
self.layout_busca = QHBoxLayout()
self.layout_busca.addWidget(self.entry_nome)
self.layout_busca.addWidget(self.button_busca)
self.layout.addLayout(self.layout_busca)
self.layout.addWidget(self.tabela_clientes)
self.setLayout(self.layout)
@Slot()
def buscar(self):
nome_buscado = self.entry_nome.text()
data = self.db.busca(nome_buscado)
self.limpar()
self.status_signal.emit("Feito")
for cliente in data:
nome = QTableWidgetItem(cliente['nome'])
numero = QTableWidgetItem(cliente['numero'])
cpf = QTableWidgetItem(cliente['cpf'])
endereco = QTableWidgetItem(cliente['endereco'])
nome.setTextAlignment(Qt.AlignCenter)
numero.setTextAlignment(Qt.AlignCenter)
cpf.setTextAlignment(Qt.AlignCenter)
endereco.setTextAlignment(Qt.AlignCenter)
self.tabela_clientes.insertRow(self.clientes)
self.tabela_clientes.setItem(self.clientes, 0, nome)
self.tabela_clientes.setItem(self.clientes, 1, numero)
self.tabela_clientes.setItem(self.clientes, 2, cpf)
self.tabela_clientes.setItem(self.clientes, 3, endereco)
self.clientes += 1
@Slot()
def limpar(self):
self.tabela_clientes.clearContents()
self.tabela_clientes.setRowCount(0)
self.clientes = 0
@Slot()
def info_cliente(self):
pass
class MainWindow(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
# Variaveis
self.separador = ";" # Separador padrao de colunas em um arquivo txt ou csv
self.selected = np.array([1, 24, 48, 96]).astype('timedelta64[h]') # selecionados ao iniciar o programa, modificavel.
self.fileformat = '' # Reservado para o formato do arquivo a ser aberto. Pode ser .xlsx ou .odf. ou .csv e assim vai.
# facilita o acesso a variavel.
self.timedeltastr = ("1 Hora","2 Horas", "3 Horas", "4 Horas","12 Horas",
"24 Horas", "48 Horas", "72 horas", "96 horas", "30 Dias")
self.timedeltas = np.array([1, 2, 3, 4, 12, 24, 48, 72, 96, 24*30]).astype('timedelta64[h]')
self.linktimedelta = dict([(self.timedeltas[x], self.timedeltastr[x]) for x in range(len(self.timedeltastr))])
self.datastring = ["DD/MM/AAAA",'AAAA/MM/DD', "AAAA-MM-DD", "DD-MM-AAAA"]
self.dataformat = ["%d/%m/%Y", "%Y/%m/%d", "%Y-%m-%d", "%d-%m-%Y"]
self.linkdata = dict([(self.datastring[x], self.dataformat[x]) for x in range(len(self.dataformat))])
self.timestring = ["hh:mm", "hh:mm:ss", "hh:mm:ss.ms"]
self.timeformat = ["%H:%M", "%H:%M:%S", "%H:%M:%S.%f"]
self.linktime = dict([(self.timestring[x], self.timeformat[x]) for x in range(len(self.timeformat))])
#Janela Principal
widget = QWidget()
self.setCentralWidget(widget)
# Inicializa os Widgets
self.folder = QLineEdit("Salvar Como...")
self.path = QLineEdit("Abrir arquivo...")
#
buttonOpen = QPushButton('Abrir')
buttonSave = QPushButton("Destino")
Processar = QPushButton('Executar')
Ajuda = QPushButton('Informações')
#
groupBox2 = QGroupBox("Delimitador")
self.delimitador1 = QRadioButton("Ponto-Vírgula")
self.delimitador2 = QRadioButton("Vírgula")
self.delimitador3 = QRadioButton("Ponto")
self.delimitador4 = QRadioButton("Tabulação")
#
checkGroup = QGroupBox("Mais opções")
text3 = QPushButton("Configurações")
text2 = QLabel("Formato da Data")
text1 = QLabel("Formato da Hora")
self.FormatoTime = QComboBox()
self.FormatoTime.addItems(self.timestring)
self.FormatoData = QComboBox()
self.FormatoData.addItems(self.datastring)
#
text = QLabel("Por favor, selecione na tabela abaixo as colunas a utilizar:")
self.ignore = QRadioButton("Possui Cabeçalho") # True se estiver selecionado, False caso nao
#
self.Tabela = QTableWidget(15,15)
self.startTable()
# Layouts
MainLayout = QVBoxLayout()
Gridlayout = QGridLayout()
Gridlayout.addWidget(self.path, 0, 0)
Gridlayout.addWidget(self.folder, 1, 0)
Gridlayout.addWidget(buttonOpen, 0, 1)
Gridlayout.addWidget(buttonSave, 1, 1)
Gridlayout.addWidget(Processar, 0, 3)
Gridlayout.addWidget(Ajuda, 1, 3)
Gridlayout.setColumnStretch(0, 2)
Gridlayout.setColumnStretch(3, 1)
Gridlayout.setColumnMinimumWidth(2, 20)
SecondLayout = QHBoxLayout()
SecondLayout.addWidget(groupBox2)
SecondLayout.addSpacing(40)
SecondLayout.addWidget(checkGroup)
#
SepLayout = QVBoxLayout()
SepLayout.addWidget(self.delimitador1)
SepLayout.addWidget(self.delimitador2)
SepLayout.addWidget(self.delimitador3)
SepLayout.addWidget(self.delimitador4)
#
OptionsLayout = QVBoxLayout()
OptionsLayout.addWidget(text3)
OptionsLayout.addWidget(text2)
OptionsLayout.addWidget(self.FormatoData)
OptionsLayout.addWidget(text1)
OptionsLayout.addWidget(self.FormatoTime)
ThirdLayout = QVBoxLayout()
ThirdLayout.addWidget(self.ignore)
ThirdLayout.addWidget(text)
MainLayout.addLayout(Gridlayout)
MainLayout.addLayout(SecondLayout)
MainLayout.addLayout(ThirdLayout)
MainLayout.addWidget(self.Tabela)
# Coloca o Layout principal na Janela
widget.setLayout(MainLayout)
# Comandos dos Widgets e edicoes.
groupBox2.setLayout(SepLayout)
self.delimitador1.setChecked(True)
self.folder.setReadOnly(True)
self.path.setReadOnly(True)
checkGroup.setLayout(OptionsLayout)
buttonOpen.clicked.connect(self.searchFile)
buttonSave.clicked.connect(self.getNewFile)
self.delimitador1.clicked.connect(self.updateDelimiter)
self.delimitador2.clicked.connect(self.updateDelimiter)
self.delimitador3.clicked.connect(self.updateDelimiter)
self.delimitador4.clicked.connect(self.updateDelimiter)
Ajuda.clicked.connect(self.help)
Processar.clicked.connect(self.taskStart)
text3.clicked.connect(self.openSubWindow)
# Propriedades da janela principal
height = 480
width = 640
myappid = 'GePlu.release1_0.0' # arbitrary string
ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(myappid)
self.setWindowIcon(QIcon(r'images\icon6.ico'))
self.setFixedSize(width, height)
self.setWindowTitle("GePlu")
def openSubWindow(self):
dialog = MyDialog(self)
dialog.show()
dialog.exec_()
def taskStart(self):
''' Inicia a execucao do programa, se houver algum erro durante o processo
notifica o usuario e deixa a funcao imediatamente'''
if self.folder.isModified() and self.path.isModified(): # o usuario entrou com os enderecos?
start = time.time()
# Conhece do que se trata cada coluna na tabela e armazena essa informacao.
header = {}
for col in range(15):
header[self.Tabela.cellWidget(0, col).currentText()] = col
boolean = 0 if self.ignore.isChecked() else None
# Le o arquivo e retorna um dataframe de strings
df_master = utils.read_file(self, header = boolean)
df_master = dt.data_filter(self, df_master, header)
df_master = dt.convert_dtype(self, df_master)
# Computa os acumulados para cada intervalo de tempo (key).
for key in self.selected:
array = dt.compute(df_master.index.to_numpy(), df_master['Observado'].to_numpy(), key)
df_master[self.linktimedelta[key]] = pd.Series(array, df_master.index)
# salva o arquivo final.
utils.save_file(self, df_master)
# Fim do processo
end = time.time()
# Notifica o usuario
QMessageBox.information(self, "Notificação", "Tarefa realizada com sucesso!\nTempo de execução: {} s.".format(round(end-start, 2)))
# Reseta a tabela e os endereço do arquivo aberto e onde salvar, na janela principal.
self.resetProgram()
else:
QMessageBox.warning(self, "Notificação", "Houve um erro no arquivo ou diretório especificado.\nPor favor, selecione um caminho válido.")
def help(self):
x = 'Caso tenha alguma dúvida, abra o documento em PDF presente\nna pasta do programa ou entre em contato.\n\nGeplu\n1.0.0'
msgBox = QMessageBox.information(self, "Informação", x)
def resetProgram(self):
self.Tabela.clearContents()
self.startTable()
self.folder.setText("Salvar Como...")
self.path.setText("Selecionar o arquivo...")
self.folder.setModified(False)
self.folder.setModified(False)
def updateDelimiter(self):
separadores = {"Ponto-Vírgula": ';', "Vírgula":",", "Ponto":".", "Tabulação":"\t"}
for x in [self.delimitador1, self.delimitador3, self.delimitador2, self.delimitador4]:
if x.isChecked():
self.separador = separadores[x.text()]
break
if self.path.isModified():
self.updateTable()
def startTable(self):
for col in range(self.Tabela.columnCount()):
combo = QComboBox()
combo.addItems(["Selecionar","Data & Hora","Prec. Observada","Data","Hora","Nível do Rio"])
self.Tabela.setCellWidget(0, col, combo)
def updateTable(self):
# Guarda a primeira linha
n_col = self.Tabela.columnCount()
textos = [0]*n_col
for col in range(n_col): textos[col] = self.Tabela.cellWidget(0, col).currentText()
self.Tabela.clearContents()
self.startTable()
for col in range(n_col): self.Tabela.cellWidget(0, col).setCurrentText(textos[col])
# Mostra na tabela as primeiras 14 linhas do arquivo que o usuario deseja abrir/utilizar.
data_df = utils.read_file(self, 14).to_numpy()
for row in range(data_df.shape[0]):
for col in range(data_df.shape[1]):
self.Tabela.setItem(row+1, col, QTableWidgetItem(data_df[row][col]))
def searchFile(self):
address, x = QFileDialog.getOpenFileName(self, "Selecione um arquivo",
filter = "Text files (*.txt *.csv *.xlsx)"
)
if len(address) > 0:
self.path.setText(address)
self.path.setModified(True)
self.fileformat = address[address.index('.'):]
self.updateTable()
def getNewFile(self):
address, x = QFileDialog.getSaveFileName(self, "Salvar como",
filter = "Text files (*.csv);; Text files (*.txt);; Planilha do Microsoft Excel (*.xlsx)"
)
if len(address) > 0:
self.folder.setText(address)
self.folder.setModified(True)
class Table(QWidget):
def __init__(self):
super().__init__()
self.initUI()
def initUI(self):
self.setWindowTitle("QTableWidget demo")
self.resize(500, 300);
conLayout = QHBoxLayout()
self.tableWidget = QTableWidget()
self.tableWidget.setRowCount(5)
self.tableWidget.setColumnCount(3)
conLayout.addWidget(self.tableWidget)
self.tableWidget.setHorizontalHeaderLabels(['姓名', '性别', '体重'])
self.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
newItem = QTableWidgetItem("张三")
self.tableWidget.setItem(0, 0, newItem)
newItem = QTableWidgetItem("男")
self.tableWidget.setItem(0, 1, newItem)
newItem = QTableWidgetItem("160")
self.tableWidget.setItem(0, 2, newItem)
# 表格中第二行记录
newItem = QTableWidgetItem("李四")
self.tableWidget.setItem(1, 0, newItem)
newItem = QTableWidgetItem("女")
self.tableWidget.setItem(1, 1, newItem)
newItem = QTableWidgetItem("170")
self.tableWidget.setItem(1, 2, newItem)
self.tableWidget.setContextMenuPolicy(Qt.CustomContextMenu) ######允许右键产生子菜单
self.tableWidget.customContextMenuRequested.connect(self.generateMenu) ####右键菜单
self.setLayout(conLayout)
def generateMenu(self, pos):
print(pos)
row_num = -1
for i in self.tableWidget.selectionModel().selection().indexes():
row_num = i.row()
menu = QMenu()
item1 = menu.addAction(u"选项一")
item2 = menu.addAction(u"选项二")
item3 = menu.addAction(u"选项三")
action = menu.exec_(self.tableWidget.mapToGlobal(pos))
if action == item1:
print('您选了选项一，当前行文字内容是：', self.tableWidget.item(row_num, 0).text(),
self.tableWidget.item(row_num, 1).text(), self.tableWidget.item(row_num, 2).text())
elif action == item2:
print('您选了选项二，当前行文字内容是：', self.tableWidget.item(row_num, 0).text(),
self.tableWidget.item(row_num, 1).text(), self.tableWidget.item(row_num, 2).text())
elif action == item3:
print('您选了选项三，当前行文字内容是：', self.tableWidget.item(row_num, 0).text(),
self.tableWidget.item(row_num, 1).text(), self.tableWidget.item(row_num, 2).text())
else:
return
class Widget(QWidget):
def __init__(self):
QWidget.__init__(self)
self.items = 0
# Example data
self._data = {
"Water": 24.5,
"Electricity": 55.1,
"Rent": 850.0,
"Supermarket": 230.4,
"Internet": 29.99,
"Spätkauf": 21.85,
"BVG Ticket": 60.0,
"Coffee": 22.45,
"Meetup": 0.0
}
# Left
self.table = QTableWidget()
self.table.setColumnCount(2)
self.table.setHorizontalHeaderLabels(["Description", "Quantity"])
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# Chart
self.chart_view = QChartView()
self.chart_view.setRenderHint(QPainter.Antialiasing)
# Right
self.description = QLineEdit()
self.quantity = QLineEdit()
self.add = QPushButton("Add")
self.clear = QPushButton("Clear")
self.quit = QPushButton("Quit")
self.plot = QPushButton("Plot")
# Disabling 'Add' button
self.add.setEnabled(False)
self.right = QVBoxLayout()
self.right.setContentsMargins(10, 10, 10, 10)
self.right.addWidget(QLabel("Description"))
self.right.addWidget(self.description)
self.right.addWidget(QLabel("Quantity"))
self.right.addWidget(self.quantity)
self.right.addWidget(self.add)
self.right.addWidget(self.plot)
self.right.addWidget(self.chart_view)
self.right.addWidget(self.clear)
self.right.addWidget(self.quit)
# QWidget Layout
self.layout = QHBoxLayout()
self.layout.addWidget(self.table)
self.layout.addLayout(self.right)
# Set the layout to the QWidget
self.setLayout(self.layout)
# Signals and Slots
self.add.clicked.connect(self.add_element)
self.quit.clicked.connect(self.quit_application)
self.plot.clicked.connect(self.plot_data)
self.clear.clicked.connect(self.clear_table)
self.description.textChanged[str].connect(self.check_disable)
self.quantity.textChanged[str].connect(self.check_disable)
# Fill example data
self.fill_table()
@Slot()
def add_element(self):
des = self.description.text()
qty = self.quantity.text()
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(des))
self.table.setItem(self.items, 1, QTableWidgetItem(qty))
self.description.setText("")
self.quantity.setText("")
self.items += 1
@Slot()
def check_disable(self, s):
if not self.description.text() or not self.quantity.text():
self.add.setEnabled(False)
else:
self.add.setEnabled(True)
@Slot()
def plot_data(self):
# Get table information
series = QPieSeries()
for i in range(self.table.rowCount()):
text = self.table.item(i, 0).text()
number = float(self.table.item(i, 1).text())
series.append(text, number)
chart = QChart()
chart.addSeries(series)
chart.legend().setAlignment(Qt.AlignLeft)
self.chart_view.setChart(chart)
@Slot()
def quit_application(self):
QApplication.quit()
def fill_table(self, data=None):
data = self._data if not data else data
for desc, price in data.items():
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(desc))
self.table.setItem(self.items, 1, QTableWidgetItem(str(price)))
self.items += 1
@Slot()
def clear_table(self):
self.table.setRowCount(0)
self.items = 0
class Widget(QWidget):
def __init__(self):
QWidget.__init__(self)
self.items = 0
# Example data
self._data = {
"Water": 24.5,
"Electricity": 55.1,
"Rent": 850.0,
"Supermarket": 230.4,
"Internet": 29.99,
"Spätkauf": 21.85,
"BVG Ticket": 60.0,
"Coffee": 22.45,
"Meetup": 0.0
}
# Left
self.table = QTableWidget()
self.table.setColumnCount(2)
self.table.setHorizontalHeaderLabels(["Description", "Quantity"])
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# Right
self.description = QLineEdit()
self.quantity = QLineEdit()
self.add = QPushButton("Add")
self.clear = QPushButton("Clear")
self.quit = QPushButton("Quit")
self.right = QVBoxLayout()
self.right.setContentsMargins(10, 10, 10, 10)
self.right.addWidget(QLabel("Description"))
self.right.addWidget(self.description)
self.right.addWidget(QLabel("Quantity"))
self.right.addWidget(self.quantity)
self.right.addWidget(self.add)
self.right.addStretch()
self.right.addWidget(self.quit)
# QWidget Layout
self.layout = QHBoxLayout()
self.layout.addWidget(self.table)
self.layout.addLayout(self.right)
# Set the layout to the QWidget
self.setLayout(self.layout)
# Signals and Slots
self.add.clicked.connect(self.add_element)
self.quit.clicked.connect(self.quit_application)
self.clear.clicked.connect(self.clear_table)
# Fill example data
self.fill_table()
@Slot()
def add_element(self):
des = self.description.text()
qty = self.quantity.text()
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(des))
self.table.setItem(self.items, 1, QTableWidgetItem(qty))
self.description.setText("")
self.quantity.setText("")
self.items += 1
@Slot()
def quit_application(self):
QApplication.quit()
def fill_table(self, data=None):
data = self._data if not data else data
for desc, price in data.items():
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(desc))
self.table.setItem(self.items, 1, QTableWidgetItem(str(price)))
self.items += 1
@Slot()
def clear_table(self):
self.table.setRowCount(0)
self.items = 0
class Setting_Ui(QWidget):
def __init__(self, persepolis_setting):
super().__init__()
icon = QIcon()
self.persepolis_setting = persepolis_setting
# add support for other languages
locale = str(self.persepolis_setting.value('settings/locale'))
QLocale.setDefault(QLocale(locale))
self.translator = QTranslator()
if self.translator.load(':/translations/locales/ui_' + locale, 'ts'):
QCoreApplication.installTranslator(self.translator)
self.setWindowIcon(
QIcon.fromTheme('persepolis', QIcon(':/persepolis.svg')))
self.setWindowTitle(
QCoreApplication.translate("setting_ui_tr", 'Preferences'))
# set ui direction
ui_direction = self.persepolis_setting.value('ui_direction')
if ui_direction == 'rtl':
self.setLayoutDirection(Qt.RightToLeft)
elif ui_direction in 'ltr':
self.setLayoutDirection(Qt.LeftToRight)
global icons
icons = ':/' + str(
self.persepolis_setting.value('settings/icons')) + '/'
# main layout
window_verticalLayout = QVBoxLayout(self)
# setting_tabWidget
self.setting_tabWidget = QTabWidget(self)
# download_options_tab
self.download_options_tab = QWidget()
download_options_tab_verticalLayout = QVBoxLayout(
self.download_options_tab)
download_options_tab_verticalLayout.setContentsMargins(21, 21, 0, 0)
# tries
tries_horizontalLayout = QHBoxLayout()
self.tries_label = QLabel(self.download_options_tab)
tries_horizontalLayout.addWidget(self.tries_label)
self.tries_spinBox = QSpinBox(self.download_options_tab)
self.tries_spinBox.setMinimum(1)
tries_horizontalLayout.addWidget(self.tries_spinBox)
download_options_tab_verticalLayout.addLayout(tries_horizontalLayout)
# wait
wait_horizontalLayout = QHBoxLayout()
self.wait_label = QLabel(self.download_options_tab)
wait_horizontalLayout.addWidget(self.wait_label)
self.wait_spinBox = QSpinBox(self.download_options_tab)
wait_horizontalLayout.addWidget(self.wait_spinBox)
download_options_tab_verticalLayout.addLayout(wait_horizontalLayout)
# time_out
time_out_horizontalLayout = QHBoxLayout()
self.time_out_label = QLabel(self.download_options_tab)
time_out_horizontalLayout.addWidget(self.time_out_label)
self.time_out_spinBox = QSpinBox(self.download_options_tab)
time_out_horizontalLayout.addWidget(self.time_out_spinBox)
download_options_tab_verticalLayout.addLayout(
time_out_horizontalLayout)
# connections
connections_horizontalLayout = QHBoxLayout()
self.connections_label = QLabel(self.download_options_tab)
connections_horizontalLayout.addWidget(self.connections_label)
self.connections_spinBox = QSpinBox(self.download_options_tab)
self.connections_spinBox.setMinimum(1)
self.connections_spinBox.setMaximum(16)
connections_horizontalLayout.addWidget(self.connections_spinBox)
download_options_tab_verticalLayout.addLayout(
connections_horizontalLayout)
# rpc_port
self.rpc_port_label = QLabel(self.download_options_tab)
self.rpc_horizontalLayout = QHBoxLayout()
self.rpc_horizontalLayout.addWidget(self.rpc_port_label)
self.rpc_port_spinbox = QSpinBox(self.download_options_tab)
self.rpc_port_spinbox.setMinimum(1024)
self.rpc_port_spinbox.setMaximum(65535)
self.rpc_horizontalLayout.addWidget(self.rpc_port_spinbox)
download_options_tab_verticalLayout.addLayout(
self.rpc_horizontalLayout)
# wait_queue
wait_queue_horizontalLayout = QHBoxLayout()
self.wait_queue_label = QLabel(self.download_options_tab)
wait_queue_horizontalLayout.addWidget(self.wait_queue_label)
self.wait_queue_time = MyQDateTimeEdit(self.download_options_tab)
self.wait_queue_time.setDisplayFormat('H:mm')
wait_queue_horizontalLayout.addWidget(self.wait_queue_time)
download_options_tab_verticalLayout.addLayout(
wait_queue_horizontalLayout)
# don't check certificate checkBox
self.dont_check_certificate_checkBox = QCheckBox(
self.download_options_tab)
download_options_tab_verticalLayout.addWidget(
self.dont_check_certificate_checkBox)
# change aria2 path
aria2_path_verticalLayout = QVBoxLayout()
self.aria2_path_checkBox = QCheckBox(self.download_options_tab)
aria2_path_verticalLayout.addWidget(self.aria2_path_checkBox)
aria2_path_horizontalLayout = QHBoxLayout()
self.aria2_path_lineEdit = QLineEdit(self.download_options_tab)
aria2_path_horizontalLayout.addWidget(self.aria2_path_lineEdit)
self.aria2_path_pushButton = QPushButton(self.download_options_tab)
aria2_path_horizontalLayout.addWidget(self.aria2_path_pushButton)
aria2_path_verticalLayout.addLayout(aria2_path_horizontalLayout)
download_options_tab_verticalLayout.addLayout(
aria2_path_verticalLayout)
download_options_tab_verticalLayout.addStretch(1)
self.setting_tabWidget.addTab(self.download_options_tab, "")
# save_as_tab
self.save_as_tab = QWidget()
save_as_tab_verticalLayout = QVBoxLayout(self.save_as_tab)
save_as_tab_verticalLayout.setContentsMargins(20, 30, 0, 0)
# download_folder
self.download_folder_horizontalLayout = QHBoxLayout()
self.download_folder_label = QLabel(self.save_as_tab)
self.download_folder_horizontalLayout.addWidget(
self.download_folder_label)
self.download_folder_lineEdit = QLineEdit(self.save_as_tab)
self.download_folder_horizontalLayout.addWidget(
self.download_folder_lineEdit)
self.download_folder_pushButton = QPushButton(self.save_as_tab)
self.download_folder_horizontalLayout.addWidget(
self.download_folder_pushButton)
save_as_tab_verticalLayout.addLayout(
self.download_folder_horizontalLayout)
# temp_download_folder
self.temp_horizontalLayout = QHBoxLayout()
self.temp_download_label = QLabel(self.save_as_tab)
self.temp_horizontalLayout.addWidget(self.temp_download_label)
self.temp_download_lineEdit = QLineEdit(self.save_as_tab)
self.temp_horizontalLayout.addWidget(self.temp_download_lineEdit)
self.temp_download_pushButton = QPushButton(self.save_as_tab)
self.temp_horizontalLayout.addWidget(self.temp_download_pushButton)
save_as_tab_verticalLayout.addLayout(self.temp_horizontalLayout)
# create subfolder
self.subfolder_checkBox = QCheckBox(self.save_as_tab)
save_as_tab_verticalLayout.addWidget(self.subfolder_checkBox)
save_as_tab_verticalLayout.addStretch(1)
self.setting_tabWidget.addTab(self.save_as_tab, "")
# notifications_tab
self.notifications_tab = QWidget()
notification_tab_verticalLayout = QVBoxLayout(self.notifications_tab)
notification_tab_verticalLayout.setContentsMargins(21, 21, 0, 0)
self.enable_notifications_checkBox = QCheckBox(self.notifications_tab)
notification_tab_verticalLayout.addWidget(
self.enable_notifications_checkBox)
self.sound_frame = QFrame(self.notifications_tab)
self.sound_frame.setFrameShape(QFrame.StyledPanel)
self.sound_frame.setFrameShadow(QFrame.Raised)
verticalLayout = QVBoxLayout(self.sound_frame)
self.volume_label = QLabel(self.sound_frame)
verticalLayout.addWidget(self.volume_label)
self.volume_dial = QDial(self.sound_frame)
self.volume_dial.setProperty("value", 100)
verticalLayout.addWidget(self.volume_dial)
notification_tab_verticalLayout.addWidget(self.sound_frame)
# message_notification
message_notification_horizontalLayout = QHBoxLayout()
self.notification_label = QLabel(self.notifications_tab)
message_notification_horizontalLayout.addWidget(
self.notification_label)
self.notification_comboBox = QComboBox(self.notifications_tab)
message_notification_horizontalLayout.addWidget(
self.notification_comboBox)
notification_tab_verticalLayout.addLayout(
message_notification_horizontalLayout)
notification_tab_verticalLayout.addStretch(1)
self.setting_tabWidget.addTab(self.notifications_tab, "")
# style_tab
self.style_tab = QWidget()
style_tab_verticalLayout = QVBoxLayout(self.style_tab)
style_tab_verticalLayout.setContentsMargins(21, 21, 0, 0)
# style
style_horizontalLayout = QHBoxLayout()
self.style_label = QLabel(self.style_tab)
style_horizontalLayout.addWidget(self.style_label)
self.style_comboBox = QComboBox(self.style_tab)
style_horizontalLayout.addWidget(self.style_comboBox)
style_tab_verticalLayout.addLayout(style_horizontalLayout)
# language
language_horizontalLayout = QHBoxLayout()
self.lang_label = QLabel(self.style_tab)
language_horizontalLayout.addWidget(self.lang_label)
self.lang_comboBox = QComboBox(self.style_tab)
language_horizontalLayout.addWidget(self.lang_comboBox)
style_tab_verticalLayout.addLayout(language_horizontalLayout)
language_horizontalLayout = QHBoxLayout()
self.lang_label.setText(
QCoreApplication.translate("setting_ui_tr", "Language: "))
# color scheme
self.color_label = QLabel(self.style_tab)
language_horizontalLayout.addWidget(self.color_label)
self.color_comboBox = QComboBox(self.style_tab)
language_horizontalLayout.addWidget(self.color_comboBox)
style_tab_verticalLayout.addLayout(language_horizontalLayout)
# icons
icons_horizontalLayout = QHBoxLayout()
self.icon_label = QLabel(self.style_tab)
icons_horizontalLayout.addWidget(self.icon_label)
self.icon_comboBox = QComboBox(self.style_tab)
icons_horizontalLayout.addWidget(self.icon_comboBox)
style_tab_verticalLayout.addLayout(icons_horizontalLayout)
self.icons_size_horizontalLayout = QHBoxLayout()
self.icons_size_label = QLabel(self.style_tab)
self.icons_size_horizontalLayout.addWidget(self.icons_size_label)
self.icons_size_comboBox = QComboBox(self.style_tab)
self.icons_size_horizontalLayout.addWidget(self.icons_size_comboBox)
style_tab_verticalLayout.addLayout(self.icons_size_horizontalLayout)
# font
font_horizontalLayout = QHBoxLayout()
self.font_checkBox = QCheckBox(self.style_tab)
font_horizontalLayout.addWidget(self.font_checkBox)
self.fontComboBox = QFontComboBox(self.style_tab)
font_horizontalLayout.addWidget(self.fontComboBox)
self.font_size_label = QLabel(self.style_tab)
font_horizontalLayout.addWidget(self.font_size_label)
self.font_size_spinBox = QSpinBox(self.style_tab)
self.font_size_spinBox.setMinimum(1)
font_horizontalLayout.addWidget(self.font_size_spinBox)
style_tab_verticalLayout.addLayout(font_horizontalLayout)
self.setting_tabWidget.addTab(self.style_tab, "")
window_verticalLayout.addWidget(self.setting_tabWidget)
# start persepolis in system tray if browser executed
self.start_persepolis_if_browser_executed_checkBox = QCheckBox(
self.style_tab)
style_tab_verticalLayout.addWidget(
self.start_persepolis_if_browser_executed_checkBox)
# hide window if close button clicked
self.hide_window_checkBox = QCheckBox(self.style_tab)
style_tab_verticalLayout.addWidget(self.hide_window_checkBox)
# Enable system tray icon
self.enable_system_tray_checkBox = QCheckBox(self.style_tab)
style_tab_verticalLayout.addWidget(self.enable_system_tray_checkBox)
# after_download dialog
self.after_download_checkBox = QCheckBox()
style_tab_verticalLayout.addWidget(self.after_download_checkBox)
# show_menubar_checkbox
self.show_menubar_checkbox = QCheckBox()
style_tab_verticalLayout.addWidget(self.show_menubar_checkbox)
# show_sidepanel_checkbox
self.show_sidepanel_checkbox = QCheckBox()
style_tab_verticalLayout.addWidget(self.show_sidepanel_checkbox)
# hide progress window
self.show_progress_window_checkbox = QCheckBox()
style_tab_verticalLayout.addWidget(self.show_progress_window_checkbox)
# add persepolis to startup
self.startup_checkbox = QCheckBox()
style_tab_verticalLayout.addWidget(self.startup_checkbox)
# keep system awake
self.keep_awake_checkBox = QCheckBox()
style_tab_verticalLayout.addWidget(self.keep_awake_checkBox)
style_tab_verticalLayout.addStretch(1)
# columns_tab
self.columns_tab = QWidget()
columns_tab_verticalLayout = QVBoxLayout(self.columns_tab)
columns_tab_verticalLayout.setContentsMargins(21, 21, 0, 0)
# creating checkBox for columns
self.show_column_label = QLabel()
self.column0_checkBox = QCheckBox()
self.column1_checkBox = QCheckBox()
self.column2_checkBox = QCheckBox()
self.column3_checkBox = QCheckBox()
self.column4_checkBox = QCheckBox()
self.column5_checkBox = QCheckBox()
self.column6_checkBox = QCheckBox()
self.column7_checkBox = QCheckBox()
self.column10_checkBox = QCheckBox()
self.column11_checkBox = QCheckBox()
self.column12_checkBox = QCheckBox()
columns_tab_verticalLayout.addWidget(self.show_column_label)
columns_tab_verticalLayout.addWidget(self.column0_checkBox)
columns_tab_verticalLayout.addWidget(self.column1_checkBox)
columns_tab_verticalLayout.addWidget(self.column2_checkBox)
columns_tab_verticalLayout.addWidget(self.column3_checkBox)
columns_tab_verticalLayout.addWidget(self.column4_checkBox)
columns_tab_verticalLayout.addWidget(self.column5_checkBox)
columns_tab_verticalLayout.addWidget(self.column6_checkBox)
columns_tab_verticalLayout.addWidget(self.column7_checkBox)
columns_tab_verticalLayout.addWidget(self.column10_checkBox)
columns_tab_verticalLayout.addWidget(self.column11_checkBox)
columns_tab_verticalLayout.addWidget(self.column12_checkBox)
columns_tab_verticalLayout.addStretch(1)
self.setting_tabWidget.addTab(self.columns_tab, '')
# video_finder_tab
self.video_finder_tab = QWidget()
video_finder_layout = QVBoxLayout(self.video_finder_tab)
video_finder_layout.setContentsMargins(21, 21, 0, 0)
video_finder_tab_verticalLayout = QVBoxLayout()
max_links_horizontalLayout = QHBoxLayout()
# max_links_label
self.max_links_label = QLabel(self.video_finder_tab)
max_links_horizontalLayout.addWidget(self.max_links_label)
# max_links_spinBox
self.max_links_spinBox = QSpinBox(self.video_finder_tab)
self.max_links_spinBox.setMinimum(1)
self.max_links_spinBox.setMaximum(16)
max_links_horizontalLayout.addWidget(self.max_links_spinBox)
video_finder_tab_verticalLayout.addLayout(max_links_horizontalLayout)
self.video_finder_dl_path_horizontalLayout = QHBoxLayout()
self.video_finder_frame = QFrame(self.video_finder_tab)
self.video_finder_frame.setLayout(video_finder_tab_verticalLayout)
video_finder_tab_verticalLayout.addStretch(1)
video_finder_layout.addWidget(self.video_finder_frame)
self.setting_tabWidget.addTab(self.video_finder_tab, "")
# shortcut tab
self.shortcut_tab = QWidget()
shortcut_tab_verticalLayout = QVBoxLayout(self.shortcut_tab)
shortcut_tab_verticalLayout.setContentsMargins(21, 21, 0, 0)
# shortcut_table
self.shortcut_table = QTableWidget(self)
self.shortcut_table.setColumnCount(2)
self.shortcut_table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.shortcut_table.setSelectionMode(QAbstractItemView.SingleSelection)
self.shortcut_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.shortcut_table.verticalHeader().hide()
shortcut_table_header = [
QCoreApplication.translate("setting_ui_tr", 'Action'),
QCoreApplication.translate("setting_ui_tr", 'Shortcut')
]
self.shortcut_table.setHorizontalHeaderLabels(shortcut_table_header)
shortcut_tab_verticalLayout.addWidget(self.shortcut_table)
self.setting_tabWidget.addTab(
self.shortcut_tab,
QCoreApplication.translate("setting_ui_tr", "Shortcuts"))
# Actions
actions_list = [
QCoreApplication.translate('setting_ui_tr', 'Quit'),
QCoreApplication.translate('setting_ui_tr',
'Minimize to System Tray'),
QCoreApplication.translate('setting_ui_tr',
'Remove Download Items'),
QCoreApplication.translate('setting_ui_tr',
'Delete Download Items'),
QCoreApplication.translate('setting_ui_tr',
'Move Selected Items Up'),
QCoreApplication.translate('setting_ui_tr',
'Move Selected Items Down'),
QCoreApplication.translate('setting_ui_tr',
'Add New Download Link'),
QCoreApplication.translate('setting_ui_tr', 'Add New Video Link'),
QCoreApplication.translate('setting_ui_tr',
'Import Links from Text File')
]
# add actions to the shortcut_table
j = 0
for action in actions_list:
item = QTableWidgetItem(str(action))
# align center
item.setTextAlignment(0x0004 | 0x0080)
# insert item in shortcut_table
self.shortcut_table.insertRow(j)
self.shortcut_table.setItem(j, 0, item)
j = j + 1
self.shortcut_table.resizeColumnsToContents()
# window buttons
buttons_horizontalLayout = QHBoxLayout()
buttons_horizontalLayout.addStretch(1)
self.defaults_pushButton = QPushButton(self)
buttons_horizontalLayout.addWidget(self.defaults_pushButton)
self.cancel_pushButton = QPushButton(self)
self.cancel_pushButton.setIcon(QIcon(icons + 'remove'))
buttons_horizontalLayout.addWidget(self.cancel_pushButton)
self.ok_pushButton = QPushButton(self)
self.ok_pushButton.setIcon(QIcon(icons + 'ok'))
buttons_horizontalLayout.addWidget(self.ok_pushButton)
window_verticalLayout.addLayout(buttons_horizontalLayout)
# set style_tab for default
self.setting_tabWidget.setCurrentIndex(3)
# labels and translations
self.setWindowTitle(
QCoreApplication.translate("setting_ui_tr", "Preferences"))
self.tries_label.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Set number of tries if download failed.</p></body></html>"
))
self.tries_label.setText(
QCoreApplication.translate("setting_ui_tr", "Number of tries: "))
self.tries_spinBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Set number of tries if download failed.</p></body></html>"
))
self.wait_label.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Set the seconds to wait between retries. Download manager will retry downloads when the HTTP server returns a 503 response.</p></body></html>"
))
self.wait_label.setText(
QCoreApplication.translate(
"setting_ui_tr", "Wait period between retries (seconds): "))
self.wait_spinBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Set the seconds to wait between retries. Download manager will retry downloads when the HTTP server returns a 503 response.</p></body></html>"
))
self.time_out_label.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Set timeout in seconds. </p></body></html>"
))
self.time_out_label.setText(
QCoreApplication.translate("setting_ui_tr", "Timeout (seconds): "))
self.time_out_spinBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Set timeout in seconds. </p></body></html>"
))
self.connections_label.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Using multiple connections can help speed up your download.</p></body></html>"
))
self.connections_label.setText(
QCoreApplication.translate("setting_ui_tr",
"Number of connections: "))
self.connections_spinBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Using multiple connections can help speed up your download.</p></body></html>"
))
self.rpc_port_label.setText(
QCoreApplication.translate("setting_ui_tr", "RPC port number: "))
self.rpc_port_spinbox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p> Specify a port number for JSON-RPC/XML-RPC server to listen to. Possible Values: 1024 - 65535 Default: 6801 </p></body></html>"
))
self.wait_queue_label.setText(
QCoreApplication.translate(
"setting_ui_tr",
'Wait period between each download in queue:'))
self.dont_check_certificate_checkBox.setText(
QCoreApplication.translate(
"setting_ui_tr", "Don't use certificate to verify the peers"))
self.dont_check_certificate_checkBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>This option avoids SSL/TLS handshake failure. But use it at your own risk!</p></body></html>"
))
self.aria2_path_checkBox.setText(
QCoreApplication.translate("setting_ui_tr",
'Change Aria2 default path'))
self.aria2_path_pushButton.setText(
QCoreApplication.translate("setting_ui_tr", 'Change'))
aria2_path_tooltip = QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Attention: Wrong path may cause problems! Do it carefully or don't change default setting!</p></body></html>"
)
self.aria2_path_checkBox.setToolTip(aria2_path_tooltip)
self.aria2_path_lineEdit.setToolTip(aria2_path_tooltip)
self.aria2_path_pushButton.setToolTip(aria2_path_tooltip)
self.setting_tabWidget.setTabText(
self.setting_tabWidget.indexOf(self.download_options_tab),
QCoreApplication.translate("setting_ui_tr", "Download Options"))
self.download_folder_label.setText(
QCoreApplication.translate("setting_ui_tr", "Download folder: "))
self.download_folder_pushButton.setText(
QCoreApplication.translate("setting_ui_tr", "Change"))
self.temp_download_label.setText(
QCoreApplication.translate("setting_ui_tr",
"Temporary download folder: "))
self.temp_download_pushButton.setText(
QCoreApplication.translate("setting_ui_tr", "Change"))
self.subfolder_checkBox.setText(
QCoreApplication.translate(
"setting_ui_tr",
"Create subfolders for Music,Videos, ... in default download folder"
))
self.setting_tabWidget.setTabText(
self.setting_tabWidget.indexOf(self.save_as_tab),
QCoreApplication.translate("setting_ui_tr", "Save As"))
self.enable_notifications_checkBox.setText(
QCoreApplication.translate("setting_ui_tr",
"Enable Notification Sounds"))
self.volume_label.setText(
QCoreApplication.translate("setting_ui_tr", "Volume: "))
self.setting_tabWidget.setTabText(
self.setting_tabWidget.indexOf(self.notifications_tab),
QCoreApplication.translate("setting_ui_tr", "Notifications"))
self.style_label.setText(
QCoreApplication.translate("setting_ui_tr", "Style: "))
self.color_label.setText(
QCoreApplication.translate("setting_ui_tr", "Color scheme: "))
self.icon_label.setText(
QCoreApplication.translate("setting_ui_tr", "Icons: "))
self.icons_size_label.setText(
QCoreApplication.translate("setting_ui_tr",
"Toolbar icons size: "))
self.notification_label.setText(
QCoreApplication.translate("setting_ui_tr", "Notification type: "))
self.font_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", "Font: "))
self.font_size_label.setText(
QCoreApplication.translate("setting_ui_tr", "Size: "))
self.hide_window_checkBox.setText(
QCoreApplication.translate(
"setting_ui_tr", "Hide main window if close button clicked."))
self.hide_window_checkBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>This feature may not work in your operating system.</p></body></html>"
))
self.start_persepolis_if_browser_executed_checkBox.setText(
QCoreApplication.translate(
'setting_ui_tr',
'If browser is opened, start Persepolis in system tray'))
self.enable_system_tray_checkBox.setText(
QCoreApplication.translate("setting_ui_tr",
"Enable system tray icon"))
self.after_download_checkBox.setText(
QCoreApplication.translate(
"setting_ui_tr",
"Show download complete dialog when download is finished"))
self.show_menubar_checkbox.setText(
QCoreApplication.translate("setting_ui_tr", "Show menubar"))
self.show_sidepanel_checkbox.setText(
QCoreApplication.translate("setting_ui_tr", "Show side panel"))
self.show_progress_window_checkbox.setText(
QCoreApplication.translate("setting_ui_tr",
"Show download progress window"))
self.startup_checkbox.setText(
QCoreApplication.translate("setting_ui_tr",
"Run Persepolis at startup"))
self.keep_awake_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", "Keep system awake!"))
self.keep_awake_checkBox.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>This option will prevent the system from going to sleep.\
It is necessary if your power manager is suspending the system automatically. </p></body></html>"
))
self.wait_queue_time.setToolTip(
QCoreApplication.translate(
"setting_ui_tr",
"<html><head/><body><p>Format HH:MM</p></body></html>"))
self.setting_tabWidget.setTabText(
self.setting_tabWidget.indexOf(self.style_tab),
QCoreApplication.translate("setting_ui_tr", "Preferences"))
# columns_tab
self.show_column_label.setText(
QCoreApplication.translate("setting_ui_tr", 'Show these columns:'))
self.column0_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'File Name'))
self.column1_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Status'))
self.column2_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Size'))
self.column3_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Downloaded'))
self.column4_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Percentage'))
self.column5_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Connections'))
self.column6_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Transfer Rate'))
self.column7_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Estimated Time Left'))
self.column10_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'First Try Date'))
self.column11_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Last Try Date'))
self.column12_checkBox.setText(
QCoreApplication.translate("setting_ui_tr", 'Category'))
self.setting_tabWidget.setTabText(
self.setting_tabWidget.indexOf(self.columns_tab),
QCoreApplication.translate("setting_ui_tr",
"Columns Customization"))
# Video Finder options tab
self.setting_tabWidget.setTabText(
self.setting_tabWidget.indexOf(self.video_finder_tab),
QCoreApplication.translate("setting_ui_tr",
"Video Finder Options"))
self.max_links_label.setText(
QCoreApplication.translate(
"setting_ui_tr", 'Maximum number of links to capture:<br/>'
'<small>(If browser sends multiple video links at a time)</small>'
))
# window buttons
self.defaults_pushButton.setText(
QCoreApplication.translate("setting_ui_tr", "Defaults"))
self.cancel_pushButton.setText(
QCoreApplication.translate("setting_ui_tr", "Cancel"))
self.ok_pushButton.setText(
QCoreApplication.translate("setting_ui_tr", "OK"))
class Widget(QWidget):
def __init__(self):
QWidget.__init__(self)
self.items = 0
# Left
self.table = QTableWidget()
self.table.setColumnCount(2)
self.table.setHorizontalHeaderLabels(["Description", "Quantity"])
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# Chart
self.chart_view = QChartView()
self.chart_view.setRenderHint(QPainter.Antialiasing)
# Right
self.description = QLineEdit()
self.quantity = QLineEdit()
self.add = QPushButton("Add")
self.clear = QPushButton("Clear")
self.quit = QPushButton("Quit")
self.plot = QPushButton("Plot")
# Disabling 'Add' button
self.add.setEnabled(False)
self.right = QVBoxLayout()
self.right.setContentsMargins(10, 10, 10, 10)
self.right_top = QGridLayout()
self.right_top.addWidget(QLabel("Description"), 0, 0, 1, 1)
self.right_top.addWidget(self.description, 0, 1, 1, 3)
self.right_top.addWidget(QLabel("Quantity"), 1, 0, 1, 1)
self.right_top.addWidget(self.quantity, 1, 1, 1, 1)
self.right_top.addWidget(self.add, 1, 2, 1, 2)
self.right.addLayout(self.right_top)
self.right.addWidget(self.chart_view)
self.right_bottom = QGridLayout()
self.right_bottom.addWidget(self.plot, 0, 0, 1, 2)
self.right_bottom.addWidget(self.clear, 1, 0)
self.right_bottom.addWidget(self.quit, 1, 1)
self.right.addLayout(self.right_bottom)
# QWidget Layout
self.layout = QHBoxLayout()
self.layout.addWidget(self.table)
self.layout.addLayout(self.right)
# Set the layout to the QWidget
self.setLayout(self.layout)
# Signals and Slots
self.add.clicked.connect(self.add_element)
self.quit.clicked.connect(self.quit_application)
self.plot.clicked.connect(self.plot_data)
self.clear.clicked.connect(self.clear_table)
self.description.textChanged[str].connect(self.check_disable)
self.quantity.textChanged[str].connect(self.check_disable)
@Slot()
def add_element(self):
des = self.description.text()
qty = self.quantity.text()
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(des))
self.table.setItem(self.items, 1, QTableWidgetItem(qty))
self.description.setText("")
self.quantity.setText("")
self.items += 1
@Slot()
def check_disable(self, s):
if not self.description.text() or not self.quantity.text():
self.add.setEnabled(False)
else:
self.add.setEnabled(True)
@Slot()
def plot_data(self):
# Get table information
series = QBarSeries()
for i in range(self.table.rowCount()):
text = self.table.item(i, 0).text()
number = float(self.table.item(i, 1).text())
bar_set = QBarSet(text)
bar_set.append(number)
series.append(bar_set)
chart = QChart()
chart.addSeries(series)
chart.legend().setAlignment(Qt.AlignLeft)
self.chart_view.setChart(chart)
@Slot()
def quit_application(self):
QApplication.quit()
@Slot()
def clear_table(self):
self.table.setRowCount(0)
self.items = 0
class DownloadWindow(QDialog):
def __init__(self,
parent: Optional[QWidget] = None,
url: str = '') -> None:
super().__init__(parent, )
if parent:
self.setWindowTitle('Download Mod')
else:
self.setWindowTitle(getTitleString('Download Mod'))
self.setAttribute(Qt.WA_DeleteOnClose)
mainLayout = QVBoxLayout(self)
mainLayout.setContentsMargins(5, 5, 5, 5)
self.signals = DownloadWindowEvents(self)
# URL input
gbUrl = QGroupBox('Mod URL')
gbUrlLayout = QVBoxLayout()
gbUrl.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
self.url = QLineEdit()
self.url.setPlaceholderText(
'https://www.nexusmods.com/witcher3/mods/...')
self.url.setText(url)
self.url.textChanged.connect(lambda: self.validateUrl(self.url.text()))
gbUrlLayout.addWidget(self.url)
self.urlInfo = QLabel('🌐')
self.urlInfo.setContentsMargins(4, 4, 4, 4)
self.urlInfo.setMinimumHeight(36)
self.urlInfo.setWordWrap(True)
gbUrlLayout.addWidget(self.urlInfo)
gbUrl.setLayout(gbUrlLayout)
mainLayout.addWidget(gbUrl)
# File selection
gbFiles = QGroupBox('Mod Files')
gbFilesLayout = QVBoxLayout()
gbFiles.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.files = QTableWidget(0, 4)
self.files.setVerticalScrollMode(QAbstractItemView.ScrollPerPixel)
self.files.setHorizontalScrollMode(QAbstractItemView.ScrollPerPixel)
self.files.setContextMenuPolicy(Qt.CustomContextMenu)
self.files.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.files.setSelectionBehavior(QAbstractItemView.SelectRows)
self.files.setWordWrap(False)
self.files.setSortingEnabled(True)
self.files.setFocusPolicy(Qt.StrongFocus)
self.files.verticalHeader().hide()
self.files.setSortingEnabled(True)
self.files.sortByColumn(2, Qt.DescendingOrder)
self.files.verticalHeader().setVisible(False)
self.files.verticalHeader().setDefaultSectionSize(25)
self.files.horizontalHeader().setHighlightSections(False)
self.files.horizontalHeader().setStretchLastSection(True)
self.files.setHorizontalHeaderLabels(
['File Name', 'Version', 'Upload Date', 'Description'])
self.files.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.files.verticalScrollBar().valueChanged.connect(
lambda: self.files.clearFocus())
self.files.itemSelectionChanged.connect(lambda: self.validateFiles())
self.files.setDisabled(True)
self.files.setStyleSheet('''
QTableView {
gridline-color: rgba(255,255,255,1);
}
QTableView::item {
padding: 5px;
margin: 1px 0;
}
QTableView::item:!selected:hover {
background-color: rgb(217, 235, 249);
padding: 0;
}
''')
gbFilesLayout.addWidget(self.files)
_mouseMoveEvent = self.files.mouseMoveEvent
self.files.hoverIndexRow = -1
def mouseMoveEvent(event: QMouseEvent) -> None:
self.files.hoverIndexRow = self.files.indexAt(event.pos()).row()
_mouseMoveEvent(event)
self.files.mouseMoveEvent = mouseMoveEvent # type: ignore
self.files.setItemDelegate(ModListItemDelegate(self.files))
self.files.setMouseTracking(True)
gbFiles.setLayout(gbFilesLayout)
mainLayout.addWidget(gbFiles)
# Actions
actionsLayout = QHBoxLayout()
actionsLayout.setAlignment(Qt.AlignRight)
self.download = QPushButton('Download', self)
self.download.clicked.connect(lambda: self.downloadEvent())
self.download.setAutoDefault(True)
self.download.setDefault(True)
self.download.setDisabled(True)
actionsLayout.addWidget(self.download)
cancel = QPushButton('Cancel', self)
cancel.clicked.connect(self.cancelEvent)
actionsLayout.addWidget(cancel)
mainLayout.addLayout(actionsLayout)
# Setup
self.setMinimumSize(QSize(420, 420))
self.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.resize(QSize(720, 420))
self.finished.connect(
lambda: self.validateUrl.cancel()) # type: ignore
self.finished.connect(
lambda: self.downloadEvent.cancel()) # type: ignore
self.modId = 0
self.validateUrl(self.url.text())
def cancelEvent(self) -> None:
self.close()
@debounce(200, cancel_running=True)
async def validateUrl(self, url: str) -> bool:
self.download.setDisabled(True)
self.files.setDisabled(True)
self.files.clearSelection()
self.files.clearFocus()
self.files.clearContents()
self.files.setRowCount(0)
self.files.setSortingEnabled(False)
self.url.setStyleSheet('')
self.modId = 0
if not url:
self.urlInfo.setText('''
<font color="#888">Please enter a valid mod url.</font>
''')
return False
modId = getModId(url)
if not modId:
self.files.setDisabled(True)
self.url.setStyleSheet('''
*{
border: 1px solid #B22222;
padding: 1px 0px;
}
''')
self.urlInfo.setText('''
<font color="#888">Please enter a valid mod url.</font>
''')
return False
self.urlInfo.setText('🌐')
try:
filesResponse = await getModFiles(modId)
except (RequestError, ResponseError, Exception) as e:
self.url.setStyleSheet('''
*{
border: 1px solid #B22222;
padding: 1px 0px;
}
''')
self.urlInfo.setText(f'''
<font color="#888">Could not get mod files: {e}.</font>
''')
return False
try:
files = filesResponse['files']
if not len(files):
self.urlInfo.setText(f'''
<font color="#888">Mod "{modId}" has no files!</font>
''')
return False
self.files.setRowCount(len(files))
for i in range(len(files)):
file = files[i]
fileid = int(file['file_id'])
name = str(file['name'])
version = str(file['version'])
_uploadtime = dateparser.parse(file['uploaded_time'])
uploadtime = _uploadtime.astimezone(tz=None).strftime(
'%Y-%m-%d %H:%M:%S') if _uploadtime else '?'
description = html.unescape(str(file['description']))
nameItem = QTableWidgetItem(name)
nameItem.setToolTip(name)
nameItem.setData(Qt.UserRole, fileid)
self.files.setItem(i, 0, nameItem)
versionItem = QTableWidgetItem(version)
versionItem.setToolTip(version)
self.files.setItem(i, 1, versionItem)
uploadtimeItem = QTableWidgetItem(uploadtime)
uploadtimeItem.setToolTip(uploadtime)
self.files.setItem(i, 2, uploadtimeItem)
descriptionItem = QTableWidgetItem(description)
descriptionItem.setToolTip(description)
self.files.setItem(i, 3, descriptionItem)
except KeyError as e:
logger.exception(
f'Could not find key "{str(e)}" in mod files response')
self.urlInfo.setText(f'''
<font color="#888">Could not find key "{str(e)}" in mod files response.</font>
''')
return False
self.urlInfo.setText(f'''
<font color="#888">Found {len(files)} available files.</font>
''')
self.files.resizeColumnsToContents()
self.files.setDisabled(False)
self.files.setSortingEnabled(True)
self.modId = modId
return True
def validateFiles(self) -> bool:
selection = self.files.selectionModel().selectedRows()
if len(selection) > 0:
self.download.setText(f'Download {len(selection)} mods')
self.download.setDisabled(False)
return True
return False
@debounce(25, cancel_running=True)
async def downloadEvent(self) -> None:
self.download.setDisabled(True)
self.url.setDisabled(True)
selection = self.files.selectionModel().selectedRows()
files = [
self.files.item(index.row(), 0).data(Qt.UserRole)
for index in selection
]
self.files.setDisabled(True)
try:
urls = await asyncio.gather(
*[getModFileUrls(self.modId, file) for file in files],
loop=asyncio.get_running_loop())
except (RequestError, ResponseError, Exception) as e:
self.url.setStyleSheet('''
*{
border: 1px solid #B22222;
padding: 1px 0px;
}
''')
self.urlInfo.setText(f'''
<font color="#888">Could not download mod files: {e}.</font>
''')
return
try:
self.signals.download.emit([url[0]['URI'] for url in urls])
except KeyError as e:
logger.exception(
f'Could not find key "{str(e)}" in file download response')
self.urlInfo.setText(f'''
<font color="#888">Could not find key "{str(e)}" in file download response.</font>
''')
return
self.close()
class MainWidget(QWidget):
def __init__(self, base):
QWidget.__init__(self)
self.base = base
self.setWindowIcon(QIcon('icon.ico'))
self.setWindowTitle(TITLE)
self.set_background_color(QColor(255, 255, 255))
self.process_header_widget = QWidget()
self.process_header_layout = QHBoxLayout(self.process_header_widget)
self.process_header_layout.setContentsMargins(0, 0, 0, 0)
self.process_label = QLabel('Available processes:')
self.github_label = QLabel(
'<a href="https://github.com/darktohka/p3dephaser">GitHub</a>')
self.github_label.setOpenExternalLinks(True)
self.refresh_button = QPushButton('Refresh')
self.refresh_button.clicked.connect(self.refresh_processes)
self.refresh_button.setFixedSize(100, 23)
self.multifile_widget = QWidget()
self.multifile_layout = QHBoxLayout(self.multifile_widget)
self.multifile_layout.setContentsMargins(0, 0, 0, 0)
self.multifileLabel = QLabel('Requested multifile names:')
self.multifileBox = QLineEdit(self)
self.multifileBox.returnPressed.connect(self.begin_scan)
self.multifile_layout.addWidget(self.multifileLabel)
self.multifile_layout.addWidget(self.multifileBox)
self.scan_button = QPushButton('Scan')
self.scan_button.clicked.connect(self.begin_scan)
self.process_list_box = QListWidget()
self.process_header_layout.addWidget(self.process_label)
self.process_header_layout.addStretch(1)
self.process_header_layout.addWidget(self.github_label)
self.process_header_layout.addWidget(self.refresh_button)
self.result_table = QTableWidget()
self.result_table.setColumnCount(3)
self.result_table.horizontalHeader().setStretchLastSection(True)
self.result_table.horizontalHeader().setSectionResizeMode(
QHeaderView.ResizeMode.Stretch)
for i, header in enumerate(('Process', 'Multifile', 'Password')):
self.result_table.setHorizontalHeaderItem(i,
QTableWidgetItem(header))
self.base_layout = QVBoxLayout(self)
self.base_layout.setContentsMargins(15, 15, 15, 15)
self.base_layout.addWidget(self.process_header_widget)
self.base_layout.addWidget(self.process_list_box)
self.base_layout.addWidget(self.multifile_widget)
self.base_layout.addWidget(self.scan_button)
self.base_layout.addWidget(self.result_table)
self.refresh_processes()
self.thread_pool = QThreadPool()
self.worker = None
self.process_name = None
self.stop_event = threading.Event()
def set_background_color(self, color):
self.setAutoFillBackground(True)
palette = self.palette()
palette.setColor(self.backgroundRole(), color)
self.setPalette(palette)
def get_processes(self):
processes = []
for proc in psutil.process_iter():
processes.append(proc.as_dict(attrs=['pid', 'name']))
processes.sort(
key=lambda process: (process['name'].lower(), process['pid']))
return processes
def refresh_processes(self):
self.process_list_box.clear()
processes = self.get_processes()
for process in processes:
name = process['name']
pid = process['pid']
self.process_list_box.addItem(f'{name} (PID {pid})')
def begin_scan(self):
if self.worker:
self.stop_event.set()
self.scan_button.setEnabled(False)
return
items = self.process_list_box.selectedItems()
if not items:
QMessageBox.warning(self, TITLE,
'Please choose a process from the list!')
return
process = items[0].text()[:-1].split(' ')
self.process_name = ' '.join(process[:-2])
pid = int(process[-1])
multifiles = self.multifileBox.text().split()
if not multifiles:
QMessageBox.warning(self, TITLE,
'Please choose some multifiles to target!')
return
multifile_names = '\n'.join(
[f'- {multifile}' for multifile in multifiles])
question = f'Do you really want to scan {self.process_name} for the following multifiles?\n\n{multifile_names}'
if QMessageBox.question(
self, TITLE, question,
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
) != QMessageBox.StandardButton.Yes:
return
self.count = 0
self.setWindowTitle(f'{TITLE} - Scanning...')
self.scan_button.setText('Stop')
self.worker = ScanWorker(self, pid, multifiles)
self.worker.signals.finished.connect(self.scan_over)
self.worker.signals.error.connect(self.error_occurred)
self.worker.signals.progress.connect(self.report_progress)
self.thread_pool.start(self.worker)
def scan_over(self):
self.worker = None
self.stop_event.clear()
self.scan_button.setText('Scan')
self.scan_button.setEnabled(True)
self.setWindowTitle(TITLE)
QMessageBox.information(
self, TITLE,
f'Scan complete!\n\n{self.count} password{"s have" if self.count != 1 else " has"} been found.'
)
def error_occurred(self, error):
exc, value, message = error
QMessageBox.critical(
self, TITLE,
f'An error has occurred while trying to scan this process!\n\n{exc} {value}\n\n{message}'
)
def report_progress(self, multifile, password):
self.count += 1
index = self.result_table.rowCount()
self.result_table.insertRow(index)
for i, value in enumerate((self.process_name, multifile, password)):
self.result_table.setItem(index, i, QTableWidgetItem(value))
app = QApplication(sys.argv)
# Create and populate the tableWidget
tableWidget = QTableWidget(4, 4)
tableWidget.setItemDelegate(StarDelegate())
tableWidget.setEditTriggers(QAbstractItemView.DoubleClicked
| QAbstractItemView.SelectedClicked)
tableWidget.setSelectionBehavior(QAbstractItemView.SelectRows)
tableWidget.setHorizontalHeaderLabels(
["Title", "Genre", "Artist", "Rating"])
data = [["Mass in B-Minor", "Baroque", "J.S. Bach", 5],
["Three More Foxes", "Jazz", "Maynard Ferguson", 4],
["Sex Bomb", "Pop", "Tom Jones", 3],
["Barbie Girl", "Pop", "Aqua", 5]]
for r in range(len(data)):
tableWidget.setItem(r, 0, QTableWidgetItem(data[r][0]))
tableWidget.setItem(r, 1, QTableWidgetItem(data[r][1]))
tableWidget.setItem(r, 2, QTableWidgetItem(data[r][2]))
item = QTableWidgetItem()
item.setData(0, StarRating(data[r][3]).starCount)
tableWidget.setItem(r, 3, item)
tableWidget.resizeColumnsToContents()
tableWidget.resize(500, 300)
tableWidget.show()
sys.exit(app.exec_())
class AnnotationsDialog(QDialog):
def __init__(self, parent, onset, duration, description):
super().__init__(parent)
self.setWindowTitle("Edit Annotations")
self.table = QTableWidget(len(onset), 3)
for row, annotation in enumerate(zip(onset, duration, description)):
self.table.setItem(row, 0, IntTableWidgetItem(annotation[0]))
self.table.setItem(row, 1, IntTableWidgetItem(annotation[1]))
self.table.setItem(row, 2, QTableWidgetItem(annotation[2]))
self.table.setHorizontalHeaderLabels(["Onset", "Duration", "Type"])
self.table.horizontalHeader().setStretchLastSection(True)
self.table.verticalHeader().setVisible(False)
self.table.setShowGrid(False)
self.table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.table.setSortingEnabled(True)
self.table.sortByColumn(0, Qt.AscendingOrder)
vbox = QVBoxLayout(self)
vbox.addWidget(self.table)
hbox = QHBoxLayout()
self.add_button = QPushButton("+")
self.remove_button = QPushButton("-")
buttonbox = QDialogButtonBox(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
hbox.addWidget(self.add_button)
hbox.addWidget(self.remove_button)
hbox.addStretch()
hbox.addWidget(buttonbox)
vbox.addLayout(hbox)
buttonbox.accepted.connect(self.accept)
buttonbox.rejected.connect(self.reject)
self.table.itemSelectionChanged.connect(self.toggle_buttons)
self.remove_button.clicked.connect(self.remove_event)
self.add_button.clicked.connect(self.add_event)
self.toggle_buttons()
self.resize(500, 500)
@Slot()
def toggle_buttons(self):
"""Toggle + and - buttons."""
n_items = len(self.table.selectedItems())
if n_items == 3: # one row (3 items) selected
self.add_button.setEnabled(True)
self.remove_button.setEnabled(True)
elif n_items > 3: # more than one row selected
self.add_button.setEnabled(False)
self.remove_button.setEnabled(True)
else: # no rows selected
self.add_button.setEnabled(False)
self.remove_button.setEnabled(False)
def add_event(self):
current_row = self.table.selectedIndexes()[0].row()
pos = int(self.table.item(current_row, 0).data(Qt.DisplayRole))
self.table.setSortingEnabled(False)
self.table.insertRow(current_row)
self.table.setItem(current_row, 0, IntTableWidgetItem(pos))
self.table.setItem(current_row, 1, IntTableWidgetItem(0))
self.table.setItem(current_row, 2, QTableWidgetItem("New Annotation"))
self.table.setSortingEnabled(True)
def remove_event(self):
rows = {index.row() for index in self.table.selectedIndexes()}
self.table.clearSelection()
for row in sorted(rows, reverse=True):
self.table.removeRow(row)
class LocationDialog(QDialog):
def __init__(self, parent=None):
super(LocationDialog, self).__init__(parent)
self.format_combo = QComboBox()
self.format_combo.addItem("Native")
self.format_combo.addItem("INI")
self.scope_cCombo = QComboBox()
self.scope_cCombo.addItem("User")
self.scope_cCombo.addItem("System")
self.organization_combo = QComboBox()
self.organization_combo.addItem("Trolltech")
self.organization_combo.setEditable(True)
self.application_combo = QComboBox()
self.application_combo.addItem("Any")
self.application_combo.addItem("Application Example")
self.application_combo.addItem("Assistant")
self.application_combo.addItem("Designer")
self.application_combo.addItem("Linguist")
self.application_combo.setEditable(True)
self.application_combo.setCurrentIndex(3)
format_label = QLabel("&Format:")
format_label.setBuddy(self.format_combo)
scope_label = QLabel("&Scope:")
scope_label.setBuddy(self.scope_cCombo)
organization_label = QLabel("&Organization:")
organization_label.setBuddy(self.organization_combo)
application_label = QLabel("&Application:")
application_label.setBuddy(self.application_combo)
self.locations_groupbox = QGroupBox("Setting Locations")
self.locations_table = QTableWidget()
self.locations_table.setSelectionMode(QAbstractItemView.SingleSelection)
self.locations_table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.locations_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.locations_table.setColumnCount(2)
self.locations_table.setHorizontalHeaderLabels(("Location", "Access"))
self.locations_table.horizontalHeader().setSectionResizeMode(0, QHeaderView.Stretch)
self.locations_table.horizontalHeader().resizeSection(1, 180)
self.button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
self.format_combo.activated.connect(self.update_locations)
self.scope_cCombo.activated.connect(self.update_locations)
self.organization_combo.lineEdit().editingFinished.connect(self.update_locations)
self.application_combo.lineEdit().editingFinished.connect(self.update_locations)
self.button_box.accepted.connect(self.accept)
self.button_box.rejected.connect(self.reject)
locations_layout = QVBoxLayout(self.locations_groupbox)
locations_layout.addWidget(self.locations_table)
mainLayout = QGridLayout(self)
mainLayout.addWidget(format_label, 0, 0)
mainLayout.addWidget(self.format_combo, 0, 1)
mainLayout.addWidget(scope_label, 1, 0)
mainLayout.addWidget(self.scope_cCombo, 1, 1)
mainLayout.addWidget(organization_label, 2, 0)
mainLayout.addWidget(self.organization_combo, 2, 1)
mainLayout.addWidget(application_label, 3, 0)
mainLayout.addWidget(self.application_combo, 3, 1)
mainLayout.addWidget(self.locations_groupbox, 4, 0, 1, 2)
mainLayout.addWidget(self.button_box, 5, 0, 1, 2)
self.update_locations()
self.setWindowTitle("Open Application Settings")
self.resize(650, 400)
def format(self):
if self.format_combo.currentIndex() == 0:
return QSettings.NativeFormat
else:
return QSettings.IniFormat
def scope(self):
if self.scope_cCombo.currentIndex() == 0:
return QSettings.UserScope
else:
return QSettings.SystemScope
def organization(self):
return self.organization_combo.currentText()
def application(self):
if self.application_combo.currentText() == "Any":
return ''
return self.application_combo.currentText()
def update_locations(self):
self.locations_table.setUpdatesEnabled(False)
self.locations_table.setRowCount(0)
for i in range(2):
if i == 0:
if self.scope() == QSettings.SystemScope:
continue
actualScope = QSettings.UserScope
else:
actualScope = QSettings.SystemScope
for j in range(2):
if j == 0:
if not self.application():
continue
actualApplication = self.application()
else:
actualApplication = ''
settings = QSettings(self.format(), actualScope,
self.organization(), actualApplication)
row = self.locations_table.rowCount()
self.locations_table.setRowCount(row + 1)
item0 = QTableWidgetItem()
item0.setText(settings.fileName())
item1 = QTableWidgetItem()
disable = not (settings.childKeys() or settings.childGroups())
if row == 0:
if settings.isWritable():
item1.setText("Read-write")
disable = False
else:
item1.setText("Read-only")
self.button_box.button(QDialogButtonBox.Ok).setDisabled(disable)
else:
item1.setText("Read-only fallback")
if disable:
item0.setFlags(item0.flags() & ~Qt.ItemIsEnabled)
item1.setFlags(item1.flags() & ~Qt.ItemIsEnabled)
self.locations_table.setItem(row, 0, item0)
self.locations_table.setItem(row, 1, item1)
self.locations_table.setUpdatesEnabled(True)
:param str code:
:return:
"""
code_hex = code.replace('#', '')
rgb = tuple(int(code_hex[i:i + 2], 16) for i in (0, 2, 3))
return QColor.fromRgb(rgb[0], rgb[1], rgb[2])
if __name__ == '__main__':
colors = [("Red", "#FF0000"), ("Green", "#00FF00"), ("Blue", "#0000FF"),
("Black", "#000000"), ("White", "#FFFFFF"),
("Electric Green", "#41CD52"), ("Dark Blue", "#222840"),
("Yellow", "#F9E56d")]
app = QApplication()
table = QTableWidget()
table.setRowCount(len(colors))
table.setColumnCount(len(colors[0]) + 1)
table.setHorizontalHeaderLabels(['Name', 'Hex Code', 'Color'])
for i, (name, code) in enumerate(colors):
item_name = QTableWidgetItem(name)
item_code = QTableWidgetItem(code)
item_color = QTableWidgetItem()
item_color.setBackground(get_rgb_from_hex(code))
table.setItem(i, 0, item_name)
table.setItem(i, 1, item_code)
table.setItem(i, 2, item_color)
table.show()
sys.exit(app.exec_())
class ApplicationWindow(QMainWindow):
def __init__(self, parent=None):
QMainWindow.__init__(self, parent)
self.column_names = ["Column A", "Column B", "Column C"]
# Central widget
self._main = QWidget()
self.setCentralWidget(self._main)
# Main menu bar
self.menu = self.menuBar()
self.menu_file = self.menu.addMenu("File")
exit = QAction("Exit", self, triggered=qApp.quit)
self.menu_file.addAction(exit)
self.menu_about = self.menu.addMenu("&About")
about = QAction("About Qt",
self,
shortcut=QKeySequence(QKeySequence.HelpContents),
triggered=qApp.aboutQt)
self.menu_about.addAction(about)
# Figure (Left)
self.fig = Figure(figsize=(5, 3))
self.canvas = FigureCanvas(self.fig)
# Sliders (Left)
self.slider_azim = QSlider(minimum=0,
maximum=360,
orientation=Qt.Horizontal)
self.slider_elev = QSlider(minimum=0,
maximum=360,
orientation=Qt.Horizontal)
self.slider_azim_layout = QHBoxLayout()
self.slider_azim_layout.addWidget(
QLabel("{}".format(self.slider_azim.minimum())))
self.slider_azim_layout.addWidget(self.slider_azim)
self.slider_azim_layout.addWidget(
QLabel("{}".format(self.slider_azim.maximum())))
self.slider_elev_layout = QHBoxLayout()
self.slider_elev_layout.addWidget(
QLabel("{}".format(self.slider_elev.minimum())))
self.slider_elev_layout.addWidget(self.slider_elev)
self.slider_elev_layout.addWidget(
QLabel("{}".format(self.slider_elev.maximum())))
# Table (Right)
self.table = QTableWidget()
header = self.table.horizontalHeader()
header.setSectionResizeMode(QHeaderView.Stretch)
# ComboBox (Right)
self.combo = QComboBox()
self.combo.addItems(
["Wired", "Surface", "Triangular Surface", "Sphere"])
# Right layout
rlayout = QVBoxLayout()
rlayout.setContentsMargins(1, 1, 1, 1)
rlayout.addWidget(QLabel("Plot type:"))
rlayout.addWidget(self.combo)
rlayout.addWidget(self.table)
# Left layout
llayout = QVBoxLayout()
rlayout.setContentsMargins(1, 1, 1, 1)
llayout.addWidget(self.canvas, 88)
llayout.addWidget(QLabel("Azimuth:"), 1)
llayout.addLayout(self.slider_azim_layout, 5)
llayout.addWidget(QLabel("Elevation:"), 1)
llayout.addLayout(self.slider_elev_layout, 5)
# Main layout
layout = QHBoxLayout(self._main)
layout.addLayout(llayout, 70)
layout.addLayout(rlayout, 30)
# Signal and Slots connections
self.combo.currentTextChanged.connect(self.combo_option)
self.slider_azim.valueChanged.connect(self.rotate_azim)
self.slider_elev.valueChanged.connect(self.rotate_elev)
# Initial setup
self.plot_wire()
self._ax.view_init(30, 30)
self.slider_azim.setValue(30)
self.slider_elev.setValue(30)
self.fig.canvas.mpl_connect("button_release_event", self.on_click)
# Matplotlib slot method
def on_click(self, event):
azim, elev = self._ax.azim, self._ax.elev
self.slider_azim.setValue(azim + 180)
self.slider_elev.setValue(elev + 180)
# Utils methods
def set_table_data(self, X, Y, Z):
for i in range(len(X)):
self.table.setItem(i, 0, QTableWidgetItem("{:.2f}".format(X[i])))
self.table.setItem(i, 1, QTableWidgetItem("{:.2f}".format(Y[i])))
self.table.setItem(i, 2, QTableWidgetItem("{:.2f}".format(Z[i])))
def set_canvas_table_configuration(self, row_count, data):
self.fig.set_canvas(self.canvas)
self._ax = self.canvas.figure.add_subplot(projection="3d")
self._ax.set_xlabel(self.column_names[0])
self._ax.set_ylabel(self.column_names[1])
self._ax.set_zlabel(self.column_names[2])
self.table.setRowCount(row_count)
self.table.setColumnCount(3)
self.table.setHorizontalHeaderLabels(self.column_names)
self.set_table_data(data[0], data[1], data[2])
# Plot methods
def plot_wire(self):
# Data
self.X, self.Y, self.Z = axes3d.get_test_data(0.03)
self.set_canvas_table_configuration(len(self.X[0]),
(self.X[0], self.Y[0], self.Z[0]))
self._ax.plot_wireframe(self.X,
self.Y,
self.Z,
rstride=10,
cstride=10,
cmap="viridis")
self.canvas.draw()
def plot_surface(self):
# Data
self.X, self.Y = np.meshgrid(np.linspace(-6, 6, 30),
np.linspace(-6, 6, 30))
self.Z = np.sin(np.sqrt(self.X**2 + self.Y**2))
self.set_canvas_table_configuration(len(self.X[0]),
(self.X[0], self.Y[0], self.Z[0]))
self._ax.plot_surface(self.X,
self.Y,
self.Z,
rstride=1,
cstride=1,
cmap="viridis",
edgecolor="none")
self.canvas.draw()
def plot_triangular_surface(self):
# Data
radii = np.linspace(0.125, 1.0, 8)
angles = np.linspace(0, 2 * np.pi, 36, endpoint=False)[..., np.newaxis]
self.X = np.append(0, (radii * np.cos(angles)).flatten())
self.Y = np.append(0, (radii * np.sin(angles)).flatten())
self.Z = np.sin(-self.X * self.Y)
self.set_canvas_table_configuration(len(self.X),
(self.X, self.Y, self.Z))
self._ax.plot_trisurf(self.X,
self.Y,
self.Z,
linewidth=0.2,
antialiased=True)
self.canvas.draw()
def plot_sphere(self):
# Data
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)
self.X = 10 * np.outer(np.cos(u), np.sin(v))
self.Y = 10 * np.outer(np.sin(u), np.sin(v))
self.Z = 9 * np.outer(np.ones(np.size(u)), np.cos(v))
self.set_canvas_table_configuration(len(self.X),
(self.X[0], self.Y[0], self.Z[0]))
self._ax.plot_surface(self.X, self.Y, self.Z)
self.canvas.draw()
# Slots
@Slot()
def combo_option(self, text):
if text == "Wired":
self.plot_wire()
elif text == "Surface":
self.plot_surface()
elif text == "Triangular Surface":
self.plot_triangular_surface()
elif text == "Sphere":
self.plot_sphere()
@Slot()
def rotate_azim(self, value):
self._ax.view_init(self._ax.elev, value)
self.fig.set_canvas(self.canvas)
self.canvas.draw()
@Slot()
def rotate_elev(self, value):
self._ax.view_init(value, self._ax.azim)
self.fig.set_canvas(self.canvas)
self.canvas.draw()