def create_iso(self, isoFileList, tmpdir):
self.progress.can_cancel_progress = False
self.steps.set_current(2)
today = datetime.date.today().strftime('%Y%m%d')
#this will hold all isos that we will create
#from user selection
self.isos = []
for cds in isoFileList.keys():
media_folder_name = self.values['media_type'] + str(cds)
current_msg = constants.MESSAGE_0022 + ' (' + media_folder_name + ')'
self.steps.set_current_text(2, current_msg )
tmppackages = utils.join_path(tmpdir , 'packages')
destination = utils.join_path(tmppackages, media_folder_name)
if self.values['isoname'] == '':
fileNames = utils.list_dir(self.values['destination'], 'aptoncd-' + today +"-" + self.values['media_type'] + str(cds))
if len(fileNames) > 0:
fileDestiny = 'aptoncd-' + today +"-" + self.values['media_type'] + str(cds) +'-'+str(len(fileNames)) +'.iso'
else:
fileDestiny = 'aptoncd-' + today +"-" + self.values['media_type'] + str(cds) +'.iso'
else:
fileNames = utils.list_dir(self.values['destination'], self.values['isoname'] +"-" + self.values['media_type'] + str(cds))
if len(fileNames) >0:
fileDestiny = self.values['isoname'].replace('.iso','') + "-" + self.values['media_type'] + str(cds) +'-' +str( len(fileNames)) +'.iso'
else:
fileDestiny = self.values['isoname'].replace('.iso','') + "-" + self.values['media_type'] + str(cds) +'.iso'
finalDestiny = utils.join_path(self.values['destination'], fileDestiny)
self.isos.append(finalDestiny)
#update processes
#marking first as done and starting a new one
self.progress.progress_text = constants.MESSAGE_0032 + ' '+ self.values['media_type'] + str(cds) + ' ('+fileDestiny+')'
gui.processEvents()
#now create the iso file
finald = finalDestiny
destd = destination
command = ['-iso-level', '4' ,'-pad', '-l','-r', '-J', '-joliet-long', '-v', '-V', 'APTonCD', '-hide-rr-moved', '-o', unicode(finald,'utf8'),unicode(destd,'utf8')]
iso_file = Mkisofs()
self.progress.stop = 100
iso_file.set_progress_hook(self.progress.update_percent)
iso_file.create_iso(command)
self.steps.set_current_text(2,constants.MESSAGE_0022)
self.steps.set_done(2)
self.progress.update_progress(0)
def is_aptoncd_media(self, from_Path):
try:
# creates a cdinfo object to check media
cdinfo = mediainfo.MediaInfo(utils.join_path(from_Path, "aptoncd.info"))
result, msgError = cdinfo.infoFromFile()
if result:
isValid, strMsg = cdinfo.compare_version()
if isValid:
return True, strMsg
else:
if gui.question_message(strMsg + "\n\n" + constants.MESSAGE_0071):
return True, strMsg
else:
return False, strMsg
else:
return False, msgError
except Exception, eMsg:
return False, eMsg
def copy_files(self):
gui.setCursorToBusy(self.get_parent_widget(), True)
self.controller.get_main_window().set_sensitive(False)
progress = self.get_parent().progressbar
progress.show()
self.controller.show_status_message("preparing to copy files")
script_file = os.path.join(constants.TEMP_FOLDER, "aoc.sc")
ifile = open(script_file, "w")
indx = 1
self.file_count = 1
f_len = len(self.packageList.store)
for nItens in self.packageList.store:
if nItens[0]:
pkg = nItens[PackageList.COL_PACKAGE]
source = pkg.deb_full_filename
destination = utils.join_path(constants.LOCAL_APT_FOLDER, pkg.deb_filename)
if not utils.fileExist(destination):
ifile.write("%s|%s\n" % (source, destination))
self.file_count += 1
percent = float(indx) / f_len
progress.set_fraction(percent)
progress.set_text(str(int(percent * 100)) + "%")
gui.processEvents()
indx += 1
ifile.close()
self.controller.show_status_message("Wait while your system is being restored.")
progress.hide()
gui.processEvents()
parentwd = self.controller.get_main_window()
winid = parentwd.window.xid
file_exec = constants.COPY_SCRIPT
command = "gksu --desktop " + constants.DESKTOP_FILE + " 'python %s %s %s'" % (file_exec, winid, script_file)
os.system(command)
gui.processEvents()
self.packageList.update_parent_count()
self.controller.get_main_window().set_sensitive(True)
gui.setCursorToBusy(self.get_parent_widget())
def create_aptoncd(self):
#this will hold error information.
errorMsg = ""
self.progress = ProgressDialog(self.get_main_window())
self.progress.can_cancel_progress = True
self.progress.title = constants.MESSAGE_0024
self.progress.description = constants.MESSAGE_0025
#defines the steps to create this iso.
process = [constants.MESSAGE_0020, constants.MESSAGE_0021, \
constants.MESSAGE_0022, constants.MESSAGE_0023]
self.steps = stepswidget.ProgressSteps(process)
self.progress.add_widget(self.steps)
#set temp variables for packages locations
tmpdir = utils.join_path(self.temp_path, 'aptoncd') #"/tmp/aptoncd/"
tmpmetapackageDir = utils.join_path(tmpdir , 'metapackage')
tmppackages = utils.join_path(tmpdir , 'packages')
#creates a temporary location to work with .deb files
utils.mkdir(tmpdir, True)
#get what files will be on each cd
if self.values['media_type'] == 'CD':
isoFileList = self.CreateIsoList(constants.CD)
else:
isoFileList = self.CreateIsoList(constants.DVD)
#we will starting changing here to break process in parts
# first, copy files from cds to locations
result, msg = self.copy_files(isoFileList,tmpdir)
# the user clicked cancel button
if not result :
self.get_main_window().set_sensitive(True)
return False
result, msg = self.scan_packages(isoFileList,tmpdir)
if not result :
self.get_main_window().set_sensitive(True)
return False
self.create_iso(isoFileList,tmpdir)
self.steps.set_current(3)
gui.processEvents()
#clean folders created previously
utils.removePath(tmpdir)
self.steps.set_done(3)
burn_list = {}
index=0
for indx, burns in enumerate(utils.get_burn_applications(constants.BURNS_DATA)):
item = utils.whereis(burns[0])
if item:
burn_list[index]= { 'name':burns[0], 'location':item , 'parameter':burns[1]}
index += 1
if index >= 1:
dlgburn = FinishedDialog(self, burn_list)
self.progress.destroy()
result, index = dlgburn.run()
if result == 1:
for iso in self.isos:
command = burn_list[index]['location'] + ' '+ burn_list[index]['parameter'] + iso + ' &'
utils.run_command(command)
dlgburn.destroy()
self.get_main_window().set_sensitive(True)
self.main_window = gui.get_glade(constants.MAIN_GUI, 'main')
self.notebook = gui.get_widget(self.main_window, 'notebook')
self.notebook.set_current_page(0)
def scan_packages(self, isoFileList, tmpdir):
self.time_start = 1
self.steps.set_current(1)
self.progress.update_progress(0)
self.progress.progress_widget.set_text( '')
self.progress.progress_text = constants.MESSAGE_0019
gui.processEvents()
for cds in isoFileList.keys():
#cd destination folder
media_folder_name = self.values['media_type'] + str(cds)
current_msg = constants.MESSAGE_0021 + ' (' + media_folder_name + ')'
self.steps.set_current_text(1,current_msg )
tmpmetapackageDir = utils.join_path(tmpdir , 'metapackage/' + media_folder_name)
tmppackages = utils.join_path(tmpdir , 'packages')
destination = utils.join_path(tmppackages, media_folder_name)
pkg_destination = utils.join_path(destination , 'packages')
controlFile = utils.join_path(tmpmetapackageDir, 'DEBIAN/control')
#create the deb file for metapackage
if self.values['createmeta']:
command = 'dpkg-deb -b %s %s > /dev/null 2> /dev/null' % \
(tmpmetapackageDir.replace(' ','\ '), utils.join_path(pkg_destination.replace(' ','\ '), self.mPack.get_mtFileName()) )
result = utils.run_command(command)
#remove the file needed to create the metapackage
utils.del_file(controlFile)
packages_files = utils.join_path(destination.replace(' ','\ ') , 'Packages')
command = 'apt-ftparchive packages packages/'
if self.progress.cancel_status:
errorMsg = constants.MESSAGE_0027
return False, errorMsg
apt_ftp = AptFtpArchive()
self.progress.stop = 100
apt_ftp.set_progress_hook(self.on_update)
apt_ftp.execute(command, destination, packages_files)
if self.progress.cancel_status:
errorMsg = constants.MESSAGE_0027
return False, errorMsg
#creates a tar file using bz2 compression
result, path = utils.compress(packages_files)
result, msg = utils.zip_file(utils.join_path(destination,'Packages.gz'), utils.join_path(destination,'Packages'))
packages_files = utils.join_path(destination.replace(' ','\ ') , 'Release')
command = 'apt-ftparchive release .'
apt_ftp.set_progress_hook(self.on_update)
apt_ftp.execute(command, destination, packages_files)
if self.progress.cancel_status:
errorMsg = constants.MESSAGE_0027
return False, errorMsg
#Create aptoncd.inf file
infoFile = utils.join_path(destination, 'aptoncd.info')
info = mediainfo.MediaInfo(infoFile)
info.write()
if self.progress.cancel_status:
errorMsg = constants.MESSAGE_0027
return False, errorMsg
#Creates .disk/info
diskinfodir = utils.join_path(destination ,'.disk')
utils.mkdir(diskinfodir, True)
infoDiskFile = utils.join_path(diskinfodir,'info')
infoDisk = mediainfo.aptDiskInfo(infoDiskFile, media_folder_name)
infoDisk.write()
#write README.diskdefines
infoDiskDefinesFile = utils.join_path(destination, 'README.diskdefines')
infoDiskDefines = mediainfo.aptDiskDefines(infoDiskDefinesFile,media_folder_name)
infoDiskDefines.write()
gui.processEvents()
utils.removePath(tmpmetapackageDir)
self.progress.cancel_pulse()
self.steps.set_current_text(1,constants.MESSAGE_0021)
self.steps.set_done(1)
self.progress.update_progress(0)
return True, None
def copy_files(self, isoFileList, tmpdir):
"""
this will copy each package from every cd to the destination.
"""
self.controlFile = {}
#start mounting the cd/dvd layout
for cds in isoFileList.keys():
#show the first step in bold with an arrow at its left
self.steps.set_current(0)
media_folder_name = self.values['media_type'] + str(cds)
current_msg = constants.MESSAGE_0020 + ' (' + media_folder_name + ')'
self.steps.set_current_text(0,current_msg )
self.progress.progress_text = constants.MESSAGE_0002
tmpmetapackageDir = utils.join_path(tmpdir , 'metapackage/' + media_folder_name)
tmppackages = utils.join_path(tmpdir , 'packages')
#if we are creating a metapackage
#we must create a DEBIAN folder
print 'Create Meta:', self.values['createmeta']
if self.values['createmeta']:
utils.mkdir(utils.join_path(tmpmetapackageDir, 'DEBIAN'),True)
controlFile = utils.join_path(tmpmetapackageDir, 'DEBIAN/control')
utils.del_file(controlFile)
self.mPack = MetaPackage(controlFile)
else:
utils.removePath(tmpmetapackageDir )
#cd destination folder
destination = utils.join_path(tmppackages, media_folder_name)
utils.mkdir(destination, True)
#destination for each package inside this media
pkg_destination = utils.join_path(destination , 'packages')
utils.mkdir(pkg_destination, True)
#start to copy files from source to destination
self.progress.stop = len(isoFileList[cds])
for index, pkg in enumerate(isoFileList[cds]):
#update the progress bar
self.progress.update_progress(index + 1)
#if we must create metapackage we will add this
#package to the meta list
if self.values['createmeta']:
self.mPack.appendPackage(pkg.package)
#set the folder destination of the package
filedestination = utils.join_path(pkg_destination, pkg.deb_filename)
#if user has canceled
#stop iterating
if self.progress.cancel_status:
errorMsg = constants.MESSAGE_0027
return False, errorMsg
#copy file to destination
pkg.copy_to(filedestination )
#update the progress message
self.progress.progress_text = constants.MESSAGE_0029 % pkg.package
gui.processEvents()
#close the packfile for this cd
if self.values['createmeta']:
self.mPack.write()
#update processes
#marking first as done and starting a new one
self.steps.set_current_text(0,constants.MESSAGE_0020)
self.steps.set_done(0)
return True, None
