def removeSelected(self,menu):
global remove_flag
remove_flag = 1
while remove_flag != 2 :
sleep(0.1)
gid_list = []
for row in range(self.download_table.rowCount()):
status = self.download_table.item(row , 1).text()
item = self.download_table.item(row , 0)
if (item.checkState() == 2) and (status == 'complete' or status == 'error' or status == 'stopped' ):
gid = self.download_table.item(row , 8 ).text()
gid_list.append(gid)
for gid in gid_list:
for i in range(self.download_table.rowCount()):
row_gid = self.download_table.item(i , 8).text()
if gid == row_gid :
row = i
break
file_name = self.download_table.item(row , 0).text()
sleep(0.5)
self.download_table.removeRow(row)
#remove gid of download from download list file
f = Open(download_list_file)
download_list_file_lines = f.readlines()
f.close()
f = Open(download_list_file , "w")
for i in download_list_file_lines:
if i.strip() != gid:
f.writelines(i.strip() + "\n")
f.close()
#remove gid of download from active download list file
f = Open(download_list_file_active)
download_list_file_active_lines = f.readlines()
f.close()
f = Open(download_list_file_active , "w")
for i in download_list_file_active_lines:
if i.strip() != gid:
f.writelines(i.strip() + "\n")
f.close()
#remove download_info_file
download_info_file = download_info_folder + "/" + gid
f = Open(download_info_file)
f.close()
f.remove()
#remove file of download form download temp folder
if file_name != '***' and status != 'complete' :
file_name_path = temp_download_folder + "/" + str(file_name)
os.system('rm "' + str(file_name_path) +'"')
file_name_aria = file_name_path + str('.aria2')
os.system('rm "' + str(file_name_aria) +'"')
remove_flag = 0
def okButtonPressed(self):
if self.dont_show_checkBox.isChecked() == True:
f = Open(self.setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
setting_dict['after-dialog'] = 'no'
f = Open(self.setting_file, 'w')
f.writelines(str(setting_dict))
f.close()
self.close()
def removeButtonPressed(self,button):
self.removeAction.setEnabled(False)
global remove_flag
if remove_flag !=3 :
remove_flag = 1
while remove_flag != 2 :
sleep(0.1)
selected_row_return = self.selectedRow()
if selected_row_return != None:
gid = self.download_table.item(selected_row_return , 8 ).text()
file_name = self.download_table.item(selected_row_return , 0).text()
status = self.download_table.item(selected_row_return , 1).text()
sleep(0.5)
self.download_table.removeRow(selected_row_return)
#remove gid of download from download list file
f = Open(download_list_file)
download_list_file_lines = f.readlines()
f.close()
f = Open(download_list_file , "w")
for i in download_list_file_lines:
if i.strip() != gid:
f.writelines(i.strip() + "\n")
f.close()
#remove gid of download from active download list file
f = Open(download_list_file_active)
download_list_file_active_lines = f.readlines()
f.close()
f = Open(download_list_file_active , "w")
for i in download_list_file_active_lines:
if i.strip() != gid:
f.writelines(i.strip() + "\n")
f.close()
#remove download_info_file
download_info_file = download_info_folder + "/" + gid
f = Open(download_info_file)
f.close()
f.remove()
#remove file of download form download temp folder
if file_name != '***' and status != 'complete' :
file_name_path = temp_download_folder + "/" + str(file_name)
os.system('rm "' + str(file_name_path) +'"')
file_name_aria = file_name_path + str('.aria2')
os.system('rm "' + str(file_name_aria) +'"')
else:
self.statusbar.showMessage("Please select an item first!")
remove_flag = 0
self.selectedRow()
def run(self):
global remove_flag
global shutdown_notification
while True:
while shutdown_notification == 0 and aria_startup_answer != 'ready':
sleep (1)
while shutdown_notification != 1:
#if remove_flag is equal to 1, it means that user pressed remove or delete button . so checking download information must stop until removing done!
if remove_flag == 1 :
remove_flag = 2
while remove_flag != 0 :
sleep(0.2)
sleep(0.3)
f = Open(download_list_file_active)
download_list_file_active_lines = f.readlines()
f.close()
if len(download_list_file_active_lines) != 0 :
for line in download_list_file_active_lines:
gid = line.strip()
try:
answer = download.downloadStatus(gid)
except:
answer = 'None'
if answer == 'ready' :
sleep(0.2)
download_info_file = download_info_folder + "/" + gid
if os.path.isfile(download_info_file) == True:
self.DOWNLOAD_INFO_SIGNAL.emit(gid)
shutdown_notification = 2
break
def notifySend(message1, message2, time, sound, systemtray=None):
if sound == 'ok':
playNotification('notifications/ok.ogg')
elif sound == 'fail':
playNotification('notifications/fail.ogg')
elif sound == 'warning':
playNotification('notifications/you.ogg')
elif sound == 'critical':
playNotification('notifications/connection.ogg')
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
enable_notification = str(setting_dict['notification'])
time = str(time)
message1 = str(message1)
message2 = str(message2)
#using Qt notification or Native system notification
if enable_notification == 'QT notification':
systemtray.showMessage(message1, message2, 1, 10000)
else:
os.system(
"notify-send --icon='persepolis' --app-name='Persepolis Download Manager' --expire-time='"
+ time + "' '" + message1 + "' \ '" + message2 + "' ")
def afterCheckBoxToggled(self, checkBoxes):
if self.after_checkBox.isChecked() == True:
self.after_frame.setEnabled(True)
else:
self.after_frame.setEnabled(False)
f = Open('/tmp/persepolis/shutdown/' + self.gid, 'w')
f.writelines('canceled')
f.close()
def okPushButtonPressed(self, button):
self.setting_dict['max-tries'] = self.tries_spinBox.value()
self.setting_dict['retry-wait'] = self.wait_spinBox.value()
self.setting_dict['timeout'] = self.time_out_spinBox.value()
self.setting_dict['connections'] = self.connections_spinBox.value()
self.setting_dict[
'download_path'] = self.download_folder_lineEdit.text()
self.setting_dict[
'download_path_temp'] = self.temp_download_lineEdit.text()
self.setting_dict['sound-volume'] = self.volume_dial.value()
self.setting_dict['style'] = self.style_comboBox.currentText()
self.setting_dict['color-scheme'] = self.color_comboBox.currentText()
self.setting_dict['icons'] = self.icon_comboBox.currentText()
self.setting_dict[
'notification'] = self.notification_comboBox.currentText()
self.setting_dict['font-size'] = self.font_size_spinBox.value()
#if user select qt notification >> enable_system_tray icon
if self.setting_dict['notification'] == 'QT notification':
self.enable_system_tray_checkBox.setChecked(True)
#enable_system_tray_checkBox
if self.enable_system_tray_checkBox.isChecked() == True:
self.setting_dict['tray-icon'] = 'yes'
self.parent.system_tray_icon.show()
self.parent.minimizeAction.setEnabled(True)
self.parent.trayAction.setChecked(True)
else:
self.setting_dict['tray-icon'] = 'no'
self.parent.system_tray_icon.hide()
self.parent.minimizeAction.setEnabled(False)
self.parent.trayAction.setChecked(False)
#after_download_checkBox
if self.after_download_checkBox.isChecked() == True:
self.setting_dict['after-dialog'] = 'yes'
else:
self.setting_dict['after-dialog'] = 'no'
#this section creates temporary download folder and download folder and download sub folders if they did not existed.
download_path_temp = self.setting_dict['download_path_temp']
download_path = self.setting_dict['download_path']
folder_list = [download_path_temp, download_path]
for folder in [
'Audios', 'Videos', 'Others', 'Documents', 'Compressed'
]:
folder_list.append(download_path + '/' + folder)
for folder in folder_list:
if os.path.isdir(folder) == False:
os.mkdir(folder)
if self.enable_notifications_checkBox.isChecked() == True:
self.setting_dict['sound'] = 'yes'
else:
self.setting_dict['sound'] = 'no'
f = Open(setting_file, 'w')
f.writelines(str(self.setting_dict))
f.close()
self.close()
def checkFlashgot(self):
sleep(0.5)
flashgot_file = Open("/tmp/persepolis-flashgot")
flashgot_line = flashgot_file.readlines()
flashgot_file.close()
flashgot_file.remove()
flashgot_add_link_dictionary_str = flashgot_line[0]
flashgot_add_link_dictionary = ast.literal_eval(flashgot_add_link_dictionary_str)
self.flashgotAddLink(flashgot_add_link_dictionary)
def afterCheckBoxToggled(self, checkBoxes):
if self.after_checkBox.isChecked() == True:
self.after_frame.setEnabled(True)
else:
self.after_frame.setEnabled(False)
shutdown_file = os.path.join(persepolis_tmp, 'shutdown', self.gid)
f = Open(shutdown_file, 'w')
f.writelines('canceled')
f.close()
def openDefaultDownloadFolder(self,menu):
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
download_path = setting_dict ['download_path']
if os.path.isdir(download_path):
os.system("xdg-open '" + download_path + "'" )
else:
notifySend(str(download_path) ,'Not Found' , 5000 , 'warning' , systemtray = self.system_tray_icon )
def shutDown(gid, password=None):
shutdown_notification_file = os.path.join(persepolis_tmp, 'shutdown', gid)
f = Open(shutdown_notification_file, 'w')
f.writelines('wait')
f.close()
shutdown_notification = "wait"
while shutdown_notification == "wait":
sleep(1)
f = Open(shutdown_notification_file, 'r')
shutdown_notification_file_lines = f.readlines()
f.close()
shutdown_notification = str(
shutdown_notification_file_lines[0].strip())
osCommands.remove(shutdown_notification_file)
if shutdown_notification == "shutdown":
print("shutdown in 20 seconds")
logger.sendToLog("Shutting down in 20 seconds", "INFO")
sleep(20)
if os_type == 'Linux':
os.system('echo "' + password + '" |sudo -S poweroff')
elif os_type == 'Darwin':
os.system('echo "' + password + '" |sudo -S shutdown -h now ')
elif os_type == 'Windows':
CREATE_NO_WINDOW = 0x08000000
subprocess.Popen(['shutdown', '-S'],
shell=False,
creationflags=CREATE_NO_WINDOW)
elif os_type == 'FreeBSD' or os_type == 'OpenBSD':
os.system('echo "' + password + '" |sudo -S shutdown -p now ')
def callBack(self , add_link_dictionary):
gid = self.gidGenerator()
download_info_file_list = ['***','waiting','***','***','***','***','***','***',gid , add_link_dictionary]
download_info_file = config_folder + "/download_info/" + gid
os.system("touch " + download_info_file )
writeList(download_info_file , download_info_file_list)
self.download_table.insertRow(0)
j = 0
download_info_file_list[9] = str(download_info_file_list[9])
for i in download_info_file_list :
item = QTableWidgetItem(i)
self.download_table.setItem(0,j,item)
j = j + 1
if self.selectAction.isChecked() == True:
item = self.download_table.item(0 , 0)
item.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled)
item.setCheckState(QtCore.Qt.Unchecked)
f = Open (download_list_file , "a")
f.writelines(gid + "\n")
f.close()
f = Open (download_list_file_active , "a")
f.writelines(gid + "\n")
f.close()
new_download = DownloadLink(gid)
self.threadPool.append(new_download)
self.threadPool[len(self.threadPool) - 1].start()
self.progressBarOpen(gid)
if add_link_dictionary['start_hour'] == None :
message = "Download Starts"
else:
message = "Download Scheduled"
notifySend(message ,'' , 10000 , 'no', systemtray = self.system_tray_icon )
def gidGenerator(self):
my_gid = hex(random.randint(1152921504606846976,18446744073709551615))
my_gid = my_gid[2:18]
my_gid = str(my_gid)
f = Open(download_list_file_active)
active_gid_list = f.readlines()
f.close()
while my_gid in active_gid_list :
my_gid = self.gidGenerator()
active_gids = download.activeDownloads()
while my_gid in active_gids:
my_gid = self.gidGenerator()
return my_gid
def pauseAllDownloads(self,menu):
#get active gid of downloads from aria
active_gids = download.activeDownloads()
#check that if gid is in download_list_file_active
f = Open(download_list_file_active)
download_list_file_active_lines = f.readlines()
f.close()
for i in range(len(download_list_file_active_lines)):
download_list_file_active_lines[i] = download_list_file_active_lines[i].strip()
for gid in active_gids :
if gid in download_list_file_active_lines :
answer = download.downloadPause(gid)
if answer == 'None':
notifySend("Aria2 did not respond!" , "Try agian!" , 10000 , 'critical' , systemtray = self.system_tray_icon )
sleep(0.3)
def playNotification(file):
#getting user setting from setting file
#setting_dict['sound'] >>> enable or disable sound
#setting_dict['sound-volume'] >>>Specify volume of sound . between 0 to 100
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
enable_notification = str(setting_dict['sound'])
volume_percent = int(setting_dict['sound-volume'])
#Paplay volume value must be between 0 (silent) and 65536 (100% volume)
volume = int((65536 * volume_percent) / 100)
if enable_notification == 'yes':
os.system("paplay --volume='" + str(volume) + "' '" + file + "' &")
def queueChanged(self, combo):
if str(self.add_queue_comboBox.currentText()) == 'Create new queue':
new_queue = self.parent.createQueue(combo)
if new_queue != None:
# clearing comboBox
self.add_queue_comboBox.clear()
# loading queue list again!
f_queues_list = Open(queues_list_file)
queues_list_file_lines = f_queues_list.readlines()
f_queues_list.close()
for queue in queues_list_file_lines:
queue_strip = queue.strip()
self.add_queue_comboBox.addItem(str(queue_strip))
self.add_queue_comboBox.addItem('Single Downloads')
self.add_queue_comboBox.addItem(QIcon(icons + 'add_queue'),
'Create new queue')
# finding index of new_queue and setting comboBox for it
index = self.add_queue_comboBox.findText(str(new_queue))
self.add_queue_comboBox.setCurrentIndex(index)
else:
self.add_queue_comboBox.setCurrentIndex(0)
def __init__(self, parent):
super().__init__(parent)
self.parent = parent
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
self.setting_dict = ast.literal_eval(setting_dict_str)
#initialization
self.tries_spinBox.setValue(int(self.setting_dict['max-tries']))
self.wait_spinBox.setValue(int(self.setting_dict['retry-wait']))
self.time_out_spinBox.setValue(int(self.setting_dict['timeout']))
self.connections_spinBox.setValue(int(
self.setting_dict['connections']))
self.download_folder_lineEdit.setText(
str(self.setting_dict['download_path']))
self.temp_download_lineEdit.setText(
str(self.setting_dict['download_path_temp']))
self.volume_label.setText('Volume : ' +
str(self.setting_dict['sound-volume']))
self.volume_dial.setValue(int(self.setting_dict['sound-volume']))
#set style
available_styles = QStyleFactory.keys()
for style in available_styles:
self.style_comboBox.addItem(style)
self.style_comboBox.addItem('System')
current_style_index = self.style_comboBox.findText(
str(self.setting_dict['style']))
if current_style_index != -1:
self.style_comboBox.setCurrentIndex(current_style_index)
#set color_scheme
color_scheme = [
'System', 'Persepolis Dark Red', 'Persepolis Dark Blue',
'Persepolis ArcDark Red', 'Persepolis ArcDark Blue'
]
self.color_comboBox.addItems(color_scheme)
current_color_index = self.color_comboBox.findText(
str(self.setting_dict['color-scheme']))
if current_color_index != -1:
self.color_comboBox.setCurrentIndex(current_color_index)
#set icons
icons = ['Archdroid-Red', 'Archdroid-Blue']
self.icon_comboBox.addItems(icons)
current_icons_index = self.icon_comboBox.findText(
str(self.setting_dict['icons']))
if current_icons_index != -1:
self.icon_comboBox.setCurrentIndex(current_icons_index)
#set notification
notifications = ['Native notification', 'QT notification']
self.notification_comboBox.addItems(notifications)
current_notification_index = self.notification_comboBox.findText(
str(self.setting_dict['notification']))
if current_notification_index != -1:
self.notification_comboBox.setCurrentIndex(
current_notification_index)
#set font
font_setting = QFont()
font_setting.setFamily(str(self.setting_dict['font']))
self.fontComboBox.setCurrentFont(font_setting)
self.font_size_spinBox.setValue(int(self.setting_dict['font-size']))
#sound frame
self.sound_frame.setEnabled(False)
self.enable_notifications_checkBox.toggled.connect(self.soundFrame)
if str(self.setting_dict['sound']) == 'yes':
self.enable_notifications_checkBox.setChecked(True)
else:
self.enable_notifications_checkBox.setChecked(False)
#connect folder buttons
self.download_folder_lineEdit.setEnabled(False)
self.download_folder_pushButton.clicked.connect(
self.downloadFolderPushButtonClicked)
self.temp_download_lineEdit.setEnabled(False)
self.temp_download_pushButton.clicked.connect(
self.tempDownloadPushButtonClicked)
#font change
self.fontComboBox.currentFontChanged.connect(self.fontChanged)
#dial
self.volume_dial.setNotchesVisible(True)
self.volume_dial.valueChanged.connect(self.dialChanged)
#tray icon
if str(self.setting_dict['tray-icon']) == 'yes':
self.enable_system_tray_checkBox.setChecked(True)
else:
self.enable_notifications_checkBox.setChecked(False)
#after download dialog
if str(self.setting_dict['after-dialog']) == 'yes':
self.after_download_checkBox.setChecked(True)
else:
self.after_download_checkBox.setChecked(False)
#ok cancel default button
self.cancel_pushButton.clicked.connect(self.cancelButtonPressed)
self.defaults_pushButton.clicked.connect(
self.defaultsPushButtonPressed)
self.ok_pushButton.clicked.connect(self.okPushButtonPressed)
def __init__(self,callback,gid):
super().__init__()
self.callback = callback
self.gid = gid
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
global connections
connections = int(setting_dict['connections'])
#connect folder_pushButton
self.folder_pushButton.clicked.connect(self.changeFolder)
self.download_folder_lineEdit.setEnabled(False)
self.ok_pushButton.setEnabled(False)
self.link_lineEdit.textChanged.connect(self.linkLineChanged)
# connect OK and canel button
self.cancel_pushButton.clicked.connect(self.cancelButtonPressed)
self.ok_pushButton.clicked.connect(self.okButtonPressed)
#frames and checkBoxes
self.proxy_frame.setEnabled(False)
self.proxy_checkBox.toggled.connect(self.proxyFrame)
self.download_frame.setEnabled(False)
self.download_checkBox.toggled.connect(self.downloadFrame)
self.limit_frame.setEnabled(False)
self.limit_checkBox.toggled.connect(self.limitFrame)
self.start_frame.setEnabled(False)
self.start_checkBox.toggled.connect(self.startFrame)
self.end_frame.setEnabled(False)
self.end_checkBox.toggled.connect(self.endFrame)
#initialization
self.connections_spinBox.setValue(connections)
download_info_file = download_info_folder + "/" + self.gid
download_info_file_list = readList(download_info_file)
self.add_link_dictionary = download_info_file_list[9]
#disable folder_frame when download is complete
status = download_info_file_list[1]
if status == 'complete':
self.folder_frame.setEnabled(False)
#link
self.link_lineEdit.setText(self.add_link_dictionary['link'])
#ip_lineEdit initialization
if self.add_link_dictionary['ip'] != None :
self.proxy_checkBox.setChecked(True)
self.ip_lineEdit.setText(self.add_link_dictionary['ip'])
#port_spinBox initialization
try:
self.port_spinBox.setValue(int(self.add_link_dictionary['port']))
except :
pass
#proxy user lineEdit initialization
try:
self.proxy_user_lineEdit.setText(self.add_link_dictionary['proxy_user'])
except :
pass
#proxy pass lineEdit initialization
try:
self.proxy_pass_lineEdit.setText(self.add_link_dictionary['proxy_passwd'])
except :
pass
#download UserName initialization
if self.add_link_dictionary['download_user'] != None :
self.download_checkBox.setChecked(True)
self.download_user_lineEdit.setText(self.add_link_dictionary['download_user'])
#download PassWord initialization
try:
self.download_pass_lineEdit.setText(self.add_link_dictionary['download_passwd'])
except :
pass
#folder_path
if self.add_link_dictionary['final_download_path'] != None :
self.download_folder_lineEdit.setText(str(self.add_link_dictionary['final_download_path']))
else:
try:
self.download_folder_lineEdit.setText(self.add_link_dictionary['download_path'])
except:
pass
#connections
try:
self.connections_spinBox.setValue(int(self.add_link_dictionary['connections']))
except :
pass
#limit speed
limit = str(self.add_link_dictionary['limit'])
if limit != '0' :
self.limit_checkBox.setChecked(True)
limit_number = limit[0:-1]
limit_unit = limit[-1]
self.limit_spinBox.setValue(int(limit_number))
if limit_unit == "K":
self.limit_comboBox.setCurrentIndex(0)
else :
self.limit_comboBox.setCurrentIndex(1)
#start_time
if self.add_link_dictionary['start_hour'] != None :
self.start_checkBox.setChecked(True)
self.start_hour_spinBox.setValue(int(self.add_link_dictionary['start_hour']))
self.start_minute_spinBox.setValue(int(self.add_link_dictionary['start_minute']))
#end_time
if self.add_link_dictionary['end_hour'] != None :
self.end_checkBox.setChecked(True)
self.end_hour_spinBox.setValue(int(self.add_link_dictionary['end_hour']))
self.end_minute_spinBox.setValue(int(self.add_link_dictionary['end_minute']))
def __init__(self, callback, gid, persepolis_setting):
super().__init__(persepolis_setting)
self.persepolis_setting = persepolis_setting
self.download_later_pushButton.hide() # hiding download_later_pushButton
self.change_name_checkBox.hide() # hiding change_name_checkBox
self.change_name_lineEdit.hide() # hiding change_name_lineEdit
self.callback = callback
self.gid = gid
global connections
connections = int(
self.persepolis_setting.value('settings/connections'))
# connect folder_pushButton
self.folder_pushButton.clicked.connect(self.changeFolder)
self.download_folder_lineEdit.setEnabled(False)
self.ok_pushButton.setEnabled(False)
self.link_lineEdit.textChanged.connect(self.linkLineChanged)
# connect OK and canel button
self.cancel_pushButton.clicked.connect(self.close)
self.ok_pushButton.clicked.connect(self.okButtonPressed)
#frames and checkBoxes
self.proxy_frame.setEnabled(False)
self.proxy_checkBox.toggled.connect(self.proxyFrame)
self.download_frame.setEnabled(False)
self.download_checkBox.toggled.connect(self.downloadFrame)
self.limit_frame.setEnabled(False)
self.limit_checkBox.toggled.connect(self.limitFrame)
self.start_frame.setEnabled(False)
self.start_checkBox.toggled.connect(self.startFrame)
self.end_frame.setEnabled(False)
self.end_checkBox.toggled.connect(self.endFrame)
# initialization
self.connections_spinBox.setValue(connections)
download_info_file = download_info_folder + "/" + self.gid
download_info_file_list = readList(download_info_file)
self.add_link_dictionary = download_info_file_list[9]
# disable folder_frame when download is complete
status = download_info_file_list[1]
if status == 'complete':
self.folder_frame.setEnabled(False)
# link
self.link_lineEdit.setText(self.add_link_dictionary['link'])
# ip_lineEdit initialization
if self.add_link_dictionary['ip'] != None:
self.proxy_checkBox.setChecked(True)
self.ip_lineEdit.setText(self.add_link_dictionary['ip'])
# port_spinBox initialization
try:
self.port_spinBox.setValue(
int(self.add_link_dictionary['port']))
except:
pass
# proxy user lineEdit initialization
try:
self.proxy_user_lineEdit.setText(
self.add_link_dictionary['proxy_user'])
except:
pass
# proxy pass lineEdit initialization
try:
self.proxy_pass_lineEdit.setText(
self.add_link_dictionary['proxy_passwd'])
except:
pass
# download UserName initialization
if self.add_link_dictionary['download_user'] != None:
self.download_checkBox.setChecked(True)
self.download_user_lineEdit.setText(
self.add_link_dictionary['download_user'])
# download PassWord initialization
try:
self.download_pass_lineEdit.setText(
self.add_link_dictionary['download_passwd'])
except:
pass
# folder_path
if self.add_link_dictionary['final_download_path'] != None:
self.download_folder_lineEdit.setText(
str(self.add_link_dictionary['final_download_path']))
else:
try:
self.download_folder_lineEdit.setText(
self.add_link_dictionary['download_path'])
except:
pass
# connections
try:
self.connections_spinBox.setValue(
int(self.add_link_dictionary['connections']))
except:
pass
# finding categories name and adding them to add_queue_comboBox
self.add_queue_comboBox.addItem('Single Downloads')
f_queues_list = Open(queues_list_file)
queues_list_file_lines = f_queues_list.readlines()
f_queues_list.close()
for queue in queues_list_file_lines:
queue_strip = queue.strip()
self.add_queue_comboBox.addItem(str(queue_strip))
# finding current queue and setting it!
self.current_category = str(download_info_file_list[12])
current_category_index = self.add_queue_comboBox.findText(
self.current_category)
self.add_queue_comboBox.setCurrentIndex(current_category_index)
# add_queue_comboBox event
self.add_queue_comboBox.currentIndexChanged.connect(self.queueChanged)
# limit speed
limit = str(self.add_link_dictionary['limit'])
if limit != '0':
self.limit_checkBox.setChecked(True)
limit_number = limit[0:-1]
limit_unit = limit[-1]
self.limit_spinBox.setValue(int(limit_number))
if limit_unit == "K":
self.limit_comboBox.setCurrentIndex(0)
else:
self.limit_comboBox.setCurrentIndex(1)
# start_time
if self.add_link_dictionary['start_hour'] != None:
self.start_checkBox.setChecked(True)
self.start_hour_spinBox.setValue(
int(self.add_link_dictionary['start_hour']))
self.start_minute_spinBox.setValue(
int(self.add_link_dictionary['start_minute']))
# end_time
if self.add_link_dictionary['end_hour'] != None:
self.end_checkBox.setChecked(True)
self.end_hour_spinBox.setValue(
int(self.add_link_dictionary['end_hour']))
self.end_minute_spinBox.setValue(
int(self.add_link_dictionary['end_minute']))
# setting window size and position
size = self.persepolis_setting.value(
'PropertiesWindow/size', QSize(700, 500))
position = self.persepolis_setting.value(
'PropertiesWindow/position', QPoint(300, 300))
self.resize(size)
self.move(position)
def okButtonPressed(self, button):
if self.proxy_checkBox.isChecked() == False:
ip = None
port = None
proxy_user = None
proxy_passwd = None
else:
ip = self.ip_lineEdit.text()
if not(ip):
ip = None
port = str(self.port_spinBox.value())
if not(port):
port = None
proxy_user = self.proxy_user_lineEdit.text()
if not(proxy_user):
proxy_user = None
proxy_passwd = self.proxy_pass_lineEdit.text()
if not(proxy_passwd):
proxy_passwd = None
if self.download_checkBox.isChecked() == False:
download_user = None
download_passwd = None
else:
download_user = self.download_user_lineEdit.text()
if not(download_user):
download_user = None
download_passwd = self.download_pass_lineEdit.text()
if not(download_passwd):
download_passwd = None
if self.limit_checkBox.isChecked() == False:
limit = 0
else:
if self.limit_comboBox.currentText() == "KB/S":
limit = str(self.limit_spinBox.value()) + str("K")
else:
limit = str(self.limit_spinBox.value()) + str("M")
if self.start_checkBox.isChecked() == False:
start_hour = None
start_minute = None
else:
start_hour = str(self.start_hour_spinBox.value())
start_minute = str(self.start_minute_spinBox.value())
if self.end_checkBox.isChecked() == False:
end_hour = None
end_minute = None
else:
end_hour = str(self.end_hour_spinBox.value())
end_minute = str(self.end_minute_spinBox.value())
link = self.link_lineEdit.text()
connections = self.connections_spinBox.value()
download_path = self.download_folder_lineEdit.text()
self.add_link_dictionary['start_hour'] = start_hour
self.add_link_dictionary['start_minute'] = start_minute
self.add_link_dictionary['end_hour'] = end_hour
self.add_link_dictionary['end_minute'] = end_minute
self.add_link_dictionary['link'] = link
self.add_link_dictionary['ip'] = ip
self.add_link_dictionary['port'] = port
self.add_link_dictionary['proxy_user'] = proxy_user
self.add_link_dictionary['proxy_passwd'] = proxy_passwd
self.add_link_dictionary['download_user'] = download_user
self.add_link_dictionary['download_passwd'] = download_passwd
self.add_link_dictionary['download_path'] = download_path
self.add_link_dictionary['limit'] = limit
self.add_link_dictionary['connections'] = connections
new_category = str(self.add_queue_comboBox.currentText())
if new_category != self.current_category: # it means category changed
# first download must eliminated form former category
# reading current_category
current_category_file = os.path.join(
category_folder, self.current_category)
f = Open(current_category_file)
f_list = f.readlines()
f.close()
# eliminating gid of download from current_category_file
f = Open(current_category_file, 'w')
for line in f_list:
gid = line.strip()
if gid != self.gid:
f.writelines(gid + '\n')
f.close()
# adding download to the new category
new_category_file = os.path.join(
category_folder, str(new_category))
f = Open(new_category_file, 'a')
f.writelines(self.gid + '\n')
f.close()
self.callback(self.add_link_dictionary, self.gid, new_category)
self.close()
def __init__(self,
parent,
callback,
persepolis_setting,
flashgot_add_link_dictionary={}):
super().__init__(persepolis_setting)
self.callback = callback
self.flashgot_add_link_dictionary = flashgot_add_link_dictionary
self.persepolis_setting = persepolis_setting
self.parent = parent
# entry initialization ->
global connections
connections = int(
self.persepolis_setting.value('settings/connections'))
global download_path
download_path = str(
self.persepolis_setting.value('settings/download_path'))
# initialization ->
self.connections_spinBox.setValue(connections)
self.download_folder_lineEdit.setText(download_path)
self.download_folder_lineEdit.setEnabled(False)
self.ok_pushButton.setEnabled(False)
self.download_later_pushButton.setEnabled(False)
self.link_lineEdit.textChanged.connect(self.linkLineChanged)
self.options_pushButton.clicked.connect(self.optionsButtonClicked)
# AddLink - checking clipboard for link! ->
if ('link' in self.flashgot_add_link_dictionary):
self.link_lineEdit.setText(
str(self.flashgot_add_link_dictionary['link']))
# spider is finding file size
new_spider = AddLinkSpiderThread(self.flashgot_add_link_dictionary)
self.parent.threadPool.append(new_spider)
self.parent.threadPool[len(self.parent.threadPool) - 1].start()
self.parent.threadPool[len(self.parent.threadPool) -
1].ADDLINKSPIDERSIGNAL.connect(
partial(
self.parent.addLinkSpiderCallBack,
child=self))
else:
clipboard = QApplication.clipboard()
text = clipboard.text()
if (("tp:/" in text[2:6]) or ("tps:/" in text[2:7])):
self.link_lineEdit.setText(str(text))
# ip_lineEdit initialization ->
settings_ip = self.persepolis_setting.value(
'add_link_initialization/ip', None)
if (settings_ip):
self.ip_lineEdit.setText(str(settings_ip))
# proxy user lineEdit initialization ->
settings_proxy_user = self.persepolis_setting.value(
'add_link_initialization/proxy_user', None)
if (settings_proxy_user):
self.proxy_user_lineEdit.setText(str(settings_proxy_user))
# port_spinBox initialization ->
settings_port = self.persepolis_setting.value(
'add_link_initialization/port', 0)
self.port_spinBox.setValue(int(int(settings_port)))
# download UserName initialization ->
settings_download_user = self.persepolis_setting.value(
'add_link_initialization/download_user', None)
if (settings_download_user):
self.download_user_lineEdit.setText(str(settings_download_user))
# finding queues name and adding them to add_queue_comboBox ->
self.add_queue_comboBox.addItem('Single Downloads')
f_queues_list = Open(queues_list_file)
queues_list_file_lines = f_queues_list.readlines()
f_queues_list.close()
for queue in queues_list_file_lines:
queue_strip = queue.strip()
self.add_queue_comboBox.addItem(str(queue_strip))
self.add_queue_comboBox.setCurrentIndex(0)
# add_queue_comboBox event
self.add_queue_comboBox.currentIndexChanged.connect(self.queueChanged)
# connect folder_pushButton
self.folder_pushButton.clicked.connect(self.changeFolder)
# connect OK and canel download_later button ->
self.cancel_pushButton.clicked.connect(self.close)
self.ok_pushButton.clicked.connect(
functools.partial(self.okButtonPressed, download_later='no'))
self.download_later_pushButton.clicked.connect(
functools.partial(self.okButtonPressed, download_later='yes'))
# frames and checkBoxes ->
self.proxy_frame.setEnabled(False)
self.proxy_checkBox.toggled.connect(self.proxyFrame)
self.download_frame.setEnabled(False)
self.download_checkBox.toggled.connect(self.downloadFrame)
self.limit_frame.setEnabled(False)
self.limit_checkBox.toggled.connect(self.limitFrame)
self.start_frame.setEnabled(False)
self.start_checkBox.toggled.connect(self.startFrame)
self.end_frame.setEnabled(False)
self.end_checkBox.toggled.connect(self.endFrame)
self.change_name_lineEdit.setEnabled(False)
self.change_name_checkBox.toggled.connect(self.changeName)
# set focus to ok button
self.ok_pushButton.setFocus()
# check name of flashgot link ->
if ('out' in self.flashgot_add_link_dictionary):
self.change_name_lineEdit.setText(
str(self.flashgot_add_link_dictionary['out']))
self.change_name_checkBox.setChecked(True)
# setting window size and position
size = self.persepolis_setting.value('AddLinkWindow/size',
QSize(520, 265))
position = self.persepolis_setting.value('AddLinkWindow/position',
QPoint(300, 300))
self.resize(size)
self.move(position)
self.minimum_height = self.height()
# more options widgets list
self.more_options_widgets = [
self.proxy_checkBox, self.proxy_frame, self.download_checkBox,
self.download_frame, self.folder_frame, self.start_checkBox,
self.start_frame, self.end_checkBox, self.end_frame,
self.limit_checkBox, self.limit_frame, self.connections_frame
]
#hiding more_options_widgets
for widgets in self.more_options_widgets:
widgets.hide()
def checkDownloadInfo(self,gid):
try:
#get download information from download_info_file according to gid and write them in download_table cells
download_info_file = config_folder + "/download_info/" + gid
download_info_file_list = readList(download_info_file)
download_info_file_list_string = readList(download_info_file ,'string')
#finding row of this gid!
for i in range(self.download_table.rowCount()):
row_gid = self.download_table.item(i , 8).text()
if gid == row_gid :
row = i
break
for i in range(10):
#check flag of download!
#It's showing that selection mode is active or not!
if i == 0 :
flag = int(self.download_table.item(row , i).flags())
#remove gid of completed download from active downloads list file
elif i == 1 :
status = str(download_info_file_list[i])
status_download_table = str(self.download_table.item(row , 1 ) . text())
if status == "complete":
f = Open(download_list_file_active)
download_list_file_active_lines = f.readlines()
f.close()
f = Open(download_list_file_active , "w")
for line in download_list_file_active_lines :
if line.strip() != gid :
f.writelines(line.strip() + "\n")
f.close()
#update download_table cells
item = QTableWidgetItem(download_info_file_list_string[i])
#48 means that item is checkable and enabled
if i == 0 and flag == 48:
item.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled)
if self.download_table.item(row , i).checkState() == 2:
item.setCheckState(QtCore.Qt.Checked)
else:
item.setCheckState(QtCore.Qt.Unchecked)
self.download_table.setItem(row , i , item)
self.download_table.viewport().update()
#update progresswindow
try :
#finding progress_window for gid
member_number = self.progress_window_list_dict[gid]
progress_window = self.progress_window_list[member_number]
#link
add_link_dictionary = download_info_file_list[9]
link = "<b>Link</b> : " + str(add_link_dictionary ['link'])
progress_window.link_label.setText(link)
progress_window.link_label.setToolTip(link)
#Save as
final_download_path = add_link_dictionary['final_download_path']
if final_download_path == None :
final_download_path = str(add_link_dictionary['download_path'])
save_as = "<b>Save as</b> : " + final_download_path + "/" + str(download_info_file_list[0])
progress_window.save_label.setText(save_as)
progress_window.save_label.setToolTip(save_as)
#status
progress_window.status = download_info_file_list[1]
status = "<b>Status</b> : " + progress_window.status
progress_window.status_label.setText(status)
if progress_window.status == "downloading":
progress_window.resume_pushButton.setEnabled(False)
progress_window.stop_pushButton.setEnabled(True)
progress_window.pause_pushButton.setEnabled(True)
elif progress_window.status == "paused":
progress_window.resume_pushButton.setEnabled(True)
progress_window.stop_pushButton.setEnabled(True)
progress_window.pause_pushButton.setEnabled(False)
elif progress_window.status == "waiting":
progress_window.resume_pushButton.setEnabled(False)
progress_window.stop_pushButton.setEnabled(True)
progress_window.pause_pushButton.setEnabled(False)
elif progress_window.status == "scheduled":
progress_window.resume_pushButton.setEnabled(False)
progress_window.stop_pushButton.setEnabled(True)
progress_window.pause_pushButton.setEnabled(False)
elif progress_window.status == "stopped" or progress_window.status == "error" or progress_window.status == "complete" :
#close progress_window if download status is stopped or completed or error
progress_window.close()
self.progress_window_list[member_number] = []
del self.progress_window_list_dict[gid]
if progress_window.status == "stopped":
notifySend("Download Stopped" , str(download_info_file_list[0]) , 10000 , 'no', systemtray = self.system_tray_icon )
elif progress_window.status == "error":
notifySend("Error - " + add_link_dictionary['error'] , str(download_info_file_list[0]) , 10000 , 'fail', systemtray = self.system_tray_icon )
add_link_dictionary['start_hour'] = None
add_link_dictionary['start_minute'] = None
add_link_dictionary['end_hour'] = None
add_link_dictionary['end_minute'] = None
add_link_dictionary['after_download'] = 'None'
for i in range(10):
if i == 9 :
download_info_file_list[i] = add_link_dictionary
download_info_file_list[9] = add_link_dictionary
writeList(download_info_file , download_info_file_list )
#this section is sending shutdown signal to the shutdown script(if user select shutdown for after download)
if os.path.isfile('/tmp/persepolis/shutdown/' + gid ) == True and progress_window.status != 'stopped':
answer = download.shutDown()
#KILL aria2c if didn't respond
if answer == 'error':
os.system('killall aria2c')
f = Open('/tmp/persepolis/shutdown/' + gid , 'w')
notifySend('Persepolis is shutting down','your system in 20 seconds' , 15000 ,'warning', systemtray = self.system_tray_icon )
f.writelines('shutdown')
f.close()
elif os.path.isfile('/tmp/persepolis/shutdown/' + gid ) == True and progress_window.status == 'stopped':
f = Open('/tmp/persepolis/shutdown/' + gid , 'w')
f.writelines('canceled')
f.close()
#showing download compelete dialog
#check user's Preferences
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
if progress_window.status == "complete" and setting_dict['after-dialog'] == 'yes' :
afterdownloadwindow = AfterDownloadWindow(download_info_file_list,setting_file)
self.afterdownload_list.append(afterdownloadwindow)
self.afterdownload_list[len(self.afterdownload_list) - 1].show()
elif progress_window.status == "complete" and setting_dict['after-dialog'] == 'no' :
notifySend("Download Complete" ,str(download_info_file_list[0]) , 10000 , 'ok' , systemtray = self.system_tray_icon )
#downloaded
downloaded = "<b>Downloaded</b> : " + str(download_info_file_list[3]) + "/" + str(download_info_file_list[2])
progress_window.downloaded_label.setText(downloaded)
#Transfer rate
rate = "<b>Transfer rate</b> : " + str(download_info_file_list[6])
progress_window.rate_label.setText(rate)
#Estimate time left
estimate_time_left = "<b>Estimate time left</b> : " + str(download_info_file_list[7])
progress_window.time_label.setText(estimate_time_left)
#Connections
connections = "<b>Connections</b> : " + str(download_info_file_list[5])
progress_window.connections_label.setText(connections)
#progressbar
value = download_info_file_list[4]
file_name = str(download_info_file_list[0])
if file_name != "***":
windows_title = '(' + str(value) + ')' + str(file_name)
progress_window.setWindowTitle(windows_title)
value = value[:-1]
progress_window.download_progressBar.setValue(int(value))
except :
pass
except:
pass
def __init__(self, parent, flashgot_lines, callback, persepolis_setting):
super().__init__(persepolis_setting)
self.persepolis_setting = persepolis_setting
self.callback = callback
self.parent = parent
self.flashgot_lines = flashgot_lines
global icons
icons = ':/' + \
str(self.persepolis_setting.value('settings/icons')) + '/'
self.flashgot_lines.reverse()
k = 1
for i in range(len(self.flashgot_lines)):
flashgot_add_link_dictionary_str = flashgot_lines[i].strip()
flashgot_add_link_dictionary = ast.literal_eval(
flashgot_add_link_dictionary_str)
# adding row to links_table
self.links_table.insertRow(0)
# file_name
if 'out' in flashgot_add_link_dictionary:
file_name = str(flashgot_add_link_dictionary['out'])
else:
file_name = '***'
# spider is finding file name
new_spider = QueueSpiderThread(flashgot_add_link_dictionary)
self.parent.threadPool.append(new_spider)
self.parent.threadPool[len(self.parent.threadPool) - 1].start()
self.parent.threadPool[len(self.parent.threadPool) -
1].QUEUESPIDERRETURNEDFILENAME.connect(
partial(
self.parent.queueSpiderCallBack,
child=self,
row_number=len(
self.flashgot_lines) - k))
k = k + 1
item = QTableWidgetItem(file_name)
# adding checkbox to the item
item.setFlags(QtCore.Qt.ItemIsUserCheckable
| QtCore.Qt.ItemIsEnabled)
item.setCheckState(QtCore.Qt.Checked)
# inserting file_name
self.links_table.setItem(0, 0, item)
# finding link
link = str(flashgot_add_link_dictionary['link'])
item = QTableWidgetItem(str(link))
# inserting link
self.links_table.setItem(0, 1, item)
# inserting add_link_dictionary
item = QTableWidgetItem(flashgot_add_link_dictionary_str)
self.links_table.setItem(0, 2, item)
# finding queues name and adding them to add_queue_comboBox
f_queues_list = Open(queues_list_file)
queues_list_file_lines = f_queues_list.readlines()
f_queues_list.close()
for queue in queues_list_file_lines:
queue_strip = queue.strip()
self.add_queue_comboBox.addItem(str(queue_strip))
self.add_queue_comboBox.addItem('Single Downloads')
self.add_queue_comboBox.addItem(QIcon(icons + 'add_queue'),
'Create new queue')
# entry initialization
global connections
connections = int(
self.persepolis_setting.value('settings/connections'))
global download_path
download_path = str(
self.persepolis_setting.value('settings/download_path'))
# initialization
self.connections_spinBox.setValue(connections)
self.download_folder_lineEdit.setText(download_path)
self.download_folder_lineEdit.setEnabled(False)
# ip_lineEdit initialization
settings_ip = self.persepolis_setting.value(
'add_link_initialization/ip', None)
if settings_ip:
self.ip_lineEdit.setText(str(settings_ip))
# proxy user lineEdit initialization
settings_proxy_user = self.persepolis_setting.value(
'add_link_initialization/proxy_user', None)
if settings_proxy_user:
self.proxy_user_lineEdit.setText(str(settings_proxy_user))
# port_spinBox initialization
settings_port = self.persepolis_setting.value(
'add_link_initialization/port', 0)
self.port_spinBox.setValue(int(int(settings_port)))
# download UserName initialization
settings_download_user = self.persepolis_setting.value(
'add_link_initialization/download_user', None)
if settings_download_user:
self.download_user_lineEdit.setText(str(settings_download_user))
# connect folder_pushButton
self.folder_pushButton.clicked.connect(self.changeFolder)
# connect OK and canel button
self.cancel_pushButton.clicked.connect(self.close)
self.ok_pushButton.clicked.connect(self.okButtonPressed)
# connect select_all_pushButton deselect_all_pushButton
self.select_all_pushButton.clicked.connect(self.selectAll)
self.deselect_all_pushButton.clicked.connect(self.deselectAll)
#frames and checkBoxes
self.proxy_frame.setEnabled(False)
self.proxy_checkBox.toggled.connect(self.proxyFrame)
self.download_frame.setEnabled(False)
self.download_checkBox.toggled.connect(self.downloadFrame)
self.limit_frame.setEnabled(False)
self.limit_checkBox.toggled.connect(self.limitFrame)
# set focus to ok button
self.ok_pushButton.setFocus()
# add_queue_comboBox event
self.add_queue_comboBox.currentIndexChanged.connect(self.queueChanged)
# setting window size and position
size = self.persepolis_setting.value('TextQueue/size', QSize(700, 500))
position = self.persepolis_setting.value('TextQueue/position',
QPoint(300, 300))
self.resize(size)
self.move(position)
def chromiumPressed(self, browser):
if os_type == 'Linux':
exec_path = '/usr/bin/persepolis'
if browser == 'chromium':
native_message_folder = home_address + '/.config/chromium/NativeMessagingHosts'
elif browser == 'chrome':
native_message_folder = home_address + \
'/.config/google-chrome/NativeMessagingHosts'
elif os_type == 'FreeBSD' or os_type == 'OpenBSD':
exec_path = '/usr/local/bin/persepolis'
if browser == 'chromium':
native_message_folder = home_address + '/.config/chromium/NativeMessagingHosts'
elif browser == 'chrome':
native_message_folder = home_address + \
'/.config/google-chrome/NativeMessagingHosts'
elif os_type == 'Darwin':
cwd = sys.argv[0]
current_directory = os.path.dirname(cwd)
exec_path = os.path.join(
current_directory, 'Persepolis Download Manager')
if browser == 'chromium':
native_message_folder = home_address + \
'/Library/Application Support/Chromium/NativeMessagingHosts'
elif browser == 'chrome':
native_message_folder = home_address + \
'/Library/Application Support/Google/Chrome/NativeMessagingHosts'
elif os_type == 'Windows':
cwd = sys.argv[0]
current_directory = os.path.dirname(cwd)
exec_path = os.path.join(
current_directory, 'Persepolis Download Manager.exe')
# the execution path in jason file for Windows must in form of
# c:\\Users\\...\\Persepolis Download Manager.exe , so we need 2
# "\" in address
exec_path = exec_path.replace('\\', r'\\')
native_message_folder = os.path.join(
home_address, 'AppData\Local\persepolis_download_manager')
json_file_lines = ['{', ' "name": "com.persepolis.pdmchromewrapper",', ' "description": "Integrate Persepolis with Google Chrome",', ' "path": "' +
str(exec_path) + '",', ' "type": "stdio",', ' "allowed_origins": [', ' "chrome-extension://legimlagjjoghkoedakdjhocbeomojao/"', ' ]', '}']
native_message_file = os.path.join(
native_message_folder, 'com.persepolis.pdmchromewrapper.json')
osCommands.makeDirs(native_message_folder)
f = Open(native_message_file, 'w')
for i in json_file_lines:
f.writelines(str(i) + '\n')
f.close()
if os_type != 'Windows':
os.system('chmod +x "' + str(native_message_file) + '"')
else:
CREATE_NO_WINDOW = 0x08000000
# subprocess.Popen(['REG','ADD','"HKCU\\Software\\Google\Chrome\\NativeMessagingHosts\\com.persepolis.pdmchromewrapper.json"', '/ve' , '/t' , 'REG_SZ' ,'/d' , '"' + str(exec_path) + '"' , '/f'] , shell = False ,creationflags = CREATE_NO_WINDOW )
os.system(r'REG ADD "HKCU\Software\Google\Chrome\NativeMessagingHosts\com.persepolis.pdmchromewrapper" /ve /t REG_SZ /d "' +
native_message_file + '" /f')
if browser == 'chromium':
self.chromium_label.setText('Done!')
elif browser == 'chrome':
self.chrome_label.setText('Done!')
def __init__(self,
callback,
persepolis_setting,
flashgot_add_link_dictionary={}):
super().__init__(persepolis_setting)
self.callback = callback
self.flashgot_add_link_dictionary = flashgot_add_link_dictionary
self.persepolis_setting = persepolis_setting
#entry initialization
global connections
connections = int(
self.persepolis_setting.value('settings/connections'))
global download_path
download_path = str(
self.persepolis_setting.value('settings/download_path'))
#initialization
self.connections_spinBox.setValue(connections)
self.download_folder_lineEdit.setText(download_path)
self.download_folder_lineEdit.setEnabled(False)
self.ok_pushButton.setEnabled(False)
self.download_later_pushButton.setEnabled(False)
self.link_lineEdit.textChanged.connect(self.linkLineChanged)
#AddLink - checking clipboard for link!
if 'link' in self.flashgot_add_link_dictionary:
self.link_lineEdit.setText(
str(self.flashgot_add_link_dictionary['link']))
del self.flashgot_add_link_dictionary['link']
else:
clipboard = QApplication.clipboard()
text = clipboard.text()
if ("tp:/" in text[2:6]) or ("tps:/" in text[2:7]):
self.link_lineEdit.setText(str(text))
#ip_lineEdit initialization
settings_ip = self.persepolis_setting.value(
'add_link_initialization/ip', None)
if settings_ip:
self.ip_lineEdit.setText(str(settings_ip))
#proxy user lineEdit initialization
settings_proxy_user = self.persepolis_setting.value(
'add_link_initialization/proxy_user', None)
if settings_proxy_user:
self.proxy_user_lineEdit.setText(str(settings_proxy_user))
#port_spinBox initialization
settings_port = self.persepolis_setting.value(
'add_link_initialization/port', 0)
self.port_spinBox.setValue(int(int(settings_port)))
#download UserName initialization
settings_download_user = self.persepolis_setting.value(
'add_link_initialization/download_user', None)
if settings_download_user:
self.download_user_lineEdit.setText(str(settings_download_user))
#finding queues name and adding them to add_queue_comboBox
self.add_queue_comboBox.addItem('Single Downloads')
f_queues_list = Open(queues_list_file)
queues_list_file_lines = f_queues_list.readlines()
f_queues_list.close()
for queue in queues_list_file_lines:
queue_strip = queue.strip()
self.add_queue_comboBox.addItem(str(queue_strip))
self.add_queue_comboBox.setCurrentIndex(0)
# add_queue_comboBox event
self.add_queue_comboBox.currentIndexChanged.connect(self.queueChanged)
#connect folder_pushButton
self.folder_pushButton.clicked.connect(self.changeFolder)
#connect OK and canel download_later button
self.cancel_pushButton.clicked.connect(self.close)
self.ok_pushButton.clicked.connect(
functools.partial(self.okButtonPressed, download_later='no'))
self.download_later_pushButton.clicked.connect(
functools.partial(self.okButtonPressed, download_later='yes'))
#frames and checkBoxes
self.proxy_frame.setEnabled(False)
self.proxy_checkBox.toggled.connect(self.proxyFrame)
self.download_frame.setEnabled(False)
self.download_checkBox.toggled.connect(self.downloadFrame)
self.limit_frame.setEnabled(False)
self.limit_checkBox.toggled.connect(self.limitFrame)
self.start_frame.setEnabled(False)
self.start_checkBox.toggled.connect(self.startFrame)
self.end_frame.setEnabled(False)
self.end_checkBox.toggled.connect(self.endFrame)
self.change_name_lineEdit.setEnabled(False)
self.change_name_checkBox.toggled.connect(self.changeName)
#set focus to ok button
self.ok_pushButton.setFocus()
#check name of flashgot link
if 'out' in self.flashgot_add_link_dictionary:
self.change_name_lineEdit.setText(
str(self.flashgot_add_link_dictionary['out']))
self.change_name_checkBox.setChecked(True)
del self.flashgot_add_link_dictionary['out']
#setting window size and position
size = self.persepolis_setting.value('AddLinkWindow/size',
QSize(520, 565))
position = self.persepolis_setting.value('AddLinkWindow/position',
QPoint(300, 300))
self.resize(size)
self.move(position)
'tray-icon': 'yes',
'max-tries': 5,
'retry-wait': 0,
'timeout': 60,
'connections': 16,
'download_path_temp': download_path_temp,
'download_path': download_path,
'sound': 'yes',
'sound-volume': 90,
'style': 'Fusion',
'color-scheme': 'Persepolis Dark Red',
'icons': 'Archdroid-Red',
'font': 'Ubuntu',
'font-size': 9
}
f = Open(setting_file, 'w')
f.writelines(str(default_setting))
f.close()
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
#this section creates temporary download folder and download folder and download sub folders if they did not existed.
download_path_temp = setting_dict['download_path_temp']
download_path = setting_dict['download_path']
folder_list = [download_path_temp, download_path]
for folder in ['Audios', 'Videos', 'Others', 'Documents', 'Compressed']:
folder_list.append(download_path + '/' + folder)
def deleteSelected(self,menu):
global remove_flag
remove_flag = 1
while remove_flag != 2 :
sleep(0.1)
gid_list = []
for row in range(self.download_table.rowCount()):
status = self.download_table.item(row , 1).text()
item = self.download_table.item(row , 0)
if (item.checkState() == 2) and (status == 'complete' or status == 'error' or status == 'stopped' ):
gid = self.download_table.item(row , 8 ).text()
gid_list.append(gid)
for gid in gid_list:
for i in range(self.download_table.rowCount()):
row_gid = self.download_table.item(i , 8).text()
if gid == row_gid :
row = i
break
file_name = self.download_table.item(row , 0).text()
add_link_dictionary_str = self.download_table.item(row , 9).text()
add_link_dictionary = ast.literal_eval(add_link_dictionary_str)
sleep(0.5)
self.download_table.removeRow(row)
#remove gid of download from download list file
f = Open(download_list_file)
download_list_file_lines = f.readlines()
f.close()
f = Open(download_list_file , "w")
for i in download_list_file_lines:
if i.strip() != gid:
f.writelines(i.strip() + "\n")
f.close()
#remove gid of download from active download list file
f = Open(download_list_file_active)
download_list_file_active_lines = f.readlines()
f.close()
f = Open(download_list_file_active , "w")
for i in download_list_file_active_lines:
if i.strip() != gid:
f.writelines(i.strip() + "\n")
f.close()
#remove download_info_file
download_info_file = download_info_folder + "/" + gid
f = Open(download_info_file)
f.close()
f.remove()
#remove file of download form download temp folder
if file_name != '***' and status != 'complete' :
file_name_path = temp_download_folder + "/" + str(file_name)
os.system('rm "' + str(file_name_path) +'"')
file_name_aria = file_name_path + str('.aria2')
os.system('rm "' + str(file_name_aria) +'"')
#remove download file
if status == 'complete':
if 'file_path' in add_link_dictionary:
file_path = add_link_dictionary['file_path']
if os.path.isfile(file_path):
os.system("rm '" + file_path + "'" )
else:
notifySend(str(file_path) ,'Not Found' , 5000 , 'warning' , systemtray = self.system_tray_icon )
remove_flag = 0
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import QWidget, QSizePolicy
from PyQt5.QtGui import QIcon
import ast, os
from newopen import Open
home_address = os.path.expanduser("~")
config_folder = str(home_address) + "/.config/persepolis_download_manager"
#setting
setting_file = config_folder + '/setting'
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
icons = ':/' + str(setting_dict['icons']) + '/'
class ProgressWindow_Ui(QWidget):
def __init__(self):
super().__init__()
#window
self.resize(595, 284)
sizePolicy = QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
def compatibility():
if os.path.isdir(download_info_folder) == True:
f = Open(download_list_file)
download_list_file_lines = f.readlines()
f.close()
for line in download_list_file_lines:
gid = line.strip()
download_info_file = download_info_folder + "/" + gid
f_download_info_file = Open(download_info_file)
download_info_file_lines = f_download_info_file.readlines()
f_download_info_file.close()
if len(download_info_file_lines) == 10:
list = []
for i in range(10):
line_strip = download_info_file_lines[i].strip()
if i == 9:
line_strip = ast.literal_eval(line_strip)
list.append(line_strip)
dictionary = {'list': list}
os.system('echo "' + str(dictionary) + '" > "' +
str(download_info_file) + '"')
f_download_info_file = Open(download_info_file, 'w')
f_download_info_file.writelines(str(dictionary))
f_download_info_file.close()
#compatibility between version 2.2 and version 2.3
date_list = []
for i in ['%Y', '%m', '%d', '%H', '%M', '%S']:
date_list.append(time.strftime(i))
for line in download_list_file_lines:
gid = line.strip()
download_info_file = download_info_folder + "/" + gid
f = Open(download_info_file, 'r')
f_string = f.readline()
f.close()
dictionary = ast.literal_eval(f_string.strip())
list = dictionary['list']
if len(list) == 10:
add_link_dictionary = list[9]
add_link_dictionary['firs_try_date'] = date_list
add_link_dictionary['last_try_date'] = date_list
list[9] = add_link_dictionary
try_date = str(date_list[0]) + '/' + str(
date_list[1]) + '/' + str(date_list[2]) + ' , ' + str(
date_list[3]) + ':' + str(date_list[4]) + ':' + str(
date_list[5])
list.append(try_date)
list.append(try_date)
list.append('Single Downloads')
dictionary = {'list': list}
f = Open(download_info_file, 'w')
f.writelines(str(dictionary))
f.close()
print(str(dictionary))
f = Open(download_list_file)
f_lines = f.readlines()
f.close()
f = Open(single_downloads_list_file, 'w')
for line in f_lines:
f.writelines(line)
f.close()
elif len(list) == 12:
list.append('Single Downloads')
dictionary = {'list': list}
f = Open(download_info_file, 'w')
f.writelines(str(dictionary))
f.close()
f = Open(download_list_file)
f_lines = f.readlines()
f.close()
f = Open(single_downloads_list_file, 'w')
for line in f_lines:
gid = line.strip()
f.writelines(gid + '\n')
f.close()
def __init__(self):
super().__init__()
self.system_tray_icon = QSystemTrayIcon()
self.system_tray_icon.setIcon(QIcon.fromTheme('persepolis',QIcon(':/icon.svg') ))
system_tray_menu = QMenu()
system_tray_menu.addAction(self.addlinkAction)
system_tray_menu.addAction(self.pauseAllAction)
system_tray_menu.addAction(self.stopAllAction)
system_tray_menu.addAction(self.minimizeAction)
system_tray_menu.addAction(self.exitAction)
self.system_tray_icon.setContextMenu(system_tray_menu)
self.system_tray_icon.activated.connect(self.systemTrayPressed)
self.system_tray_icon.show()
self.system_tray_icon.setToolTip('Persepolis Download Manager')
f = Open(setting_file)
setting_file_lines = f.readlines()
f.close()
setting_dict_str = str(setting_file_lines[0].strip())
setting_dict = ast.literal_eval(setting_dict_str)
if setting_dict['tray-icon'] != 'yes':
self.minimizeAction.setEnabled(False)
self.trayAction.setChecked(False)
self.system_tray_icon.hide()
self.statusbar.showMessage('Please Wait ...')
self.checkSelectedRow()
#touch download_list_file
if not(os.path.isfile(download_list_file)):
f = Open(download_list_file , 'w')
f.close()
#touch download_list_file_active
if not(os.path.isfile(download_list_file_active)):
f = Open(download_list_file_active , 'w')
f.close()
#lock files perventing to access a file simultaneously
#removing lock files in starting persepolis
os.system("rm " + config_folder +"/*.lock" + " 2>/dev/null" )
os.system("rm " + download_info_folder + "/*.lock" + " 2>/dev/null" )
#threads
self.threadPool=[]
#starting aria
start_aria = StartAria2Thread()
self.threadPool.append(start_aria)
self.threadPool[0].start()
self.threadPool[0].ARIA2RESPONDSIGNAL.connect(self.startAriaMessage)
#initializing
#add downloads to the download_table
f_download_list_file = Open(download_list_file)
download_list_file_lines = f_download_list_file.readlines()
f_download_list_file.close()
for line in download_list_file_lines:
gid = line.strip()
self.download_table.insertRow(0)
download_info_file = download_info_folder + "/" + gid
download_info_file_list = readList(download_info_file,'string')
for i in range(10):
item = QTableWidgetItem(download_info_file_list[i])
self.download_table.setItem(0 , i , item)
row_numbers = self.download_table.rowCount()
for row in range(row_numbers):
status = self.download_table.item(row , 1).text()
if (status != "complete" and status != "error"):
gid = self.download_table.item(row,8).text()
add_link_dictionary_str = self.download_table.item(row,9).text()
add_link_dictionary = ast.literal_eval(add_link_dictionary_str.strip())
add_link_dictionary['start_hour'] = None
add_link_dictionary['start_minute'] = None
add_link_dictionary['end_hour'] = None
add_link_dictionary['end_minute'] = None
add_link_dictionary['after_download'] = 'None'
download_info_file = download_info_folder + "/" + gid
download_info_file_list = readList(download_info_file,'string')
for i in range(10):
if i == 1 :
download_info_file_list[i] = 'stopped'
item = QTableWidgetItem('stopped')
self.download_table.setItem(row , i , item )
download_info_file_list[9] = add_link_dictionary
writeList(download_info_file , download_info_file_list)
self.addlinkwindows_list = []
self.propertieswindows_list = []
self.progress_window_list = []
self.afterdownload_list = []
self.progress_window_list_dict = {}
check_download_info = CheckDownloadInfoThread()
self.threadPool.append(check_download_info)
self.threadPool[1].start()
self.threadPool[1].DOWNLOAD_INFO_SIGNAL.connect(self.checkDownloadInfo)
check_selected_row = CheckSelectedRowThread()
self.threadPool.append(check_selected_row)
self.threadPool[2].start()
self.threadPool[2].CHECKSELECTEDROWSIGNAL.connect(self.checkSelectedRow)
check_flashgot = CheckFlashgot()
self.threadPool.append(check_flashgot)
self.threadPool[3].start()
self.threadPool[3].CHECKFLASHGOTSIGNAL.connect(self.checkFlashgot)
self.threadPool[3].SHOWMAINWINDOWSIGNAL.connect(self.showMainWindow)
self.download_table.itemDoubleClicked.connect(self.openFile)
