98 lines
2.6 KiB
Python
98 lines
2.6 KiB
Python
import os
|
|
import gtk
|
|
import zipfile,tarfile
|
|
|
|
class Archiver():
|
|
|
|
def __init__(self, main_window) :
|
|
self.main_window = main_window
|
|
self.archive_mimes = ('application/zip', 'application/x-tar', 'application/x-compressed-tar')
|
|
self.menu_mimes = ('application/octet-stream', 'inode/directory', 'text/directory', 'folder')
|
|
self.extract_menu_item = (
|
|
{
|
|
'label': 'Extract',
|
|
'data': False,
|
|
'callback': self.extract,
|
|
},
|
|
{
|
|
'label': 'Extract in folder',
|
|
'data': True,
|
|
'callback': self.extract,
|
|
},
|
|
)
|
|
|
|
self.extract_menu = gtk.Menu()
|
|
for item in self.extract_menu_item:
|
|
self.extract_menu.append(self.main_window.menu_manager.create_menu_item(item))
|
|
|
|
|
|
self.create_menu_item = (
|
|
{
|
|
'label': 'Make archive',
|
|
'callback': self.make_archive,
|
|
},
|
|
)
|
|
|
|
self.create_menu = gtk.Menu()
|
|
for item in self.create_menu_item:
|
|
self.create_menu.append(self.main_window.menu_manager.create_menu_item(item))
|
|
|
|
self.archiver_menu = self.main_window.menu_manager.create_menu_item(
|
|
{
|
|
'label': 'Archiver',
|
|
'submenu': ''
|
|
}
|
|
)
|
|
self.archiver_menu.connect("activate", self.expand_menu)
|
|
|
|
def expand_menu(self, data):
|
|
self.archiver_menu.set_submenu(self.create_menu)
|
|
filename = self.get_selection()
|
|
|
|
try:
|
|
if zipfile.is_zipfile(filename) or tarfile.is_tarfile(filename):
|
|
self.archiver_menu.set_submenu(self.extract_menu)
|
|
except IOError as e:
|
|
if e.errno == 21:
|
|
pass
|
|
|
|
def get_selection(self):
|
|
selections = self.main_window.get_active_object()._get_selection()
|
|
return selections
|
|
|
|
def extract(self, widget, in_folder):
|
|
filepath = self.get_selection()
|
|
path = os.path.dirname(filepath)
|
|
if in_folder :
|
|
path = os.path.splitext(filepath)[0]
|
|
|
|
try:
|
|
if zipfile.is_zipfile(filepath):
|
|
archive = zipfile.ZipFile(filepath)
|
|
elif tarfile.is_tarfile(filepath):
|
|
archive = tarfile.open(filepath)
|
|
archive.extractall(path)
|
|
archive.close()
|
|
except Exception, e :
|
|
print e
|
|
|
|
def make_archive(self, widget, data):
|
|
filepath = self.get_selection()
|
|
zfile = filepath.rstrip('/') + '.zip'
|
|
filepath = filepath.rstrip('/')
|
|
zf = zipfile.ZipFile(zfile, mode='w')
|
|
if os.path.isfile(filepath):
|
|
zf.write(filepath, filepath[len(os.path.dirname(filepath)):].strip('/'), compress_type=zipfile.ZIP_DEFLATED)
|
|
else:
|
|
for root, dirs, files in os.walk(filepath):
|
|
for name in files:
|
|
file_to_zip = os.path.join(root, name)
|
|
arcname = file_to_zip[len(os.path.dirname(filepath)):].strip('/')
|
|
zf.write(file_to_zip, arcname, compress_type=zipfile.ZIP_DEFLATED)
|
|
|
|
|
|
def register_plugin(application):
|
|
archiver = Archiver(application)
|
|
application.register_popup_menu_action(archiver.menu_mimes, archiver.archiver_menu)
|
|
|