Mercurial > gcp
diff gcp @ 0:674ce820a4ef
initial commit
author | Goffi <goffi@goffi.org> |
---|---|
date | Wed, 25 Aug 2010 16:11:34 +0800 |
parents | |
children | 995487813501 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/gcp Wed Aug 25 16:11:34 2010 +0800 @@ -0,0 +1,215 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- + +""" +gcp: Goffi's CoPier +Copyright (C) 2010 Jérôme Poisson (goffi@goffi.org) + +This program is free software: you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation, either version 3 of the License, or +(at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program. If not, see <http://www.gnu.org/licenses/>. +""" + +### logging ### +import logging +from logging import debug, info, error, warning +logging.basicConfig(level=logging.DEBUG, + format='%(message)s') +### +import sys +import os,os.path +from optparse import OptionParser #To be replaced by argparse ASAP +try: + import gobject + #DBus + import dbus, dbus.glib + import dbus.service + import dbus.mainloop.glib +except ImportError,e: + error("Error during import") + error("Please check dependecies:",e) + exit(2) +try: + from progressbar import ProgressBar, Percentage, Bar, ETA, FileTransferSpeed + pbar_available=True +except ImportError, e: + info ('ProgressBar not available, please download it at http://pypi.python.org/pypi/progressbar') + info ('Progress bar deactivated\n--\n') + pbar_available=False + +NAME = "gcp (Goffi's copier)" +NAME_SHORT = "gcp" +VERSION = '0.1' + +ABOUT = NAME+" v"+VERSION+""" (c) Jérôme Poisson (aka Goffi) 2010 + +--- +"""+NAME+""" Copyright (C) 2010 Jérôme Poisson +This program comes with ABSOLUTELY NO WARRANTY; +This is free software, and you are welcome to redistribute it +under certain conditions. +--- + +This software is an advanced file copier +Get the latest version at http://www.goffi.org +""" + +const_DBUS_INTERFACE = "org.goffi.gcp" +const_DBUS_PATH = "/org/goffi/gcp" + + +class DbusObject(dbus.service.Object): + + def __init__(self, gcp, bus, path): + self._gcp = gcp + dbus.service.Object.__init__(self, bus, path) + debug("Init DbusObject...") + self.cb={} + + @dbus.service.method(const_DBUS_INTERFACE, + in_signature='', out_signature='s') + def getVersion(self): + """Get gcp version + @return: version as string""" + return VERSION + + @dbus.service.method(const_DBUS_INTERFACE, + in_signature='sas', out_signature='b') + def addArgs(self, source_path, args): + """Add arguments to gcp as if there were entered on its own command line + @source_path: current working dir to use as base for arguments, as given by os.getcwd() + @args: list of strings - without command name -, as given by sys.argv[1:]""" + return self._gcp.parseArguments(args, source_path) + +class GCP(): + + def __init__(self): + try: + sessions_bus = dbus.SessionBus() + db_object = sessions_bus.get_object(const_DBUS_INTERFACE, + const_DBUS_PATH) + self.gcp_main = dbus.Interface(db_object, + dbus_interface=const_DBUS_INTERFACE) + self._main_instance = False + + except dbus.exceptions.DBusException,e: + if e._dbus_error_name=='org.freedesktop.DBus.Error.ServiceUnknown': + self.launchDbusMainInstance() + debug ("gcp launched") + self._main_instance = True + else: + raise e + + def launchDbusMainInstance(self): + debug ("Init DBus...") + session_bus = dbus.SessionBus() + self.dbus_name = dbus.service.BusName(const_DBUS_INTERFACE, session_bus) + self.dbus_object = DbusObject(self, session_bus, const_DBUS_PATH) + + self.copy_list = [] + self.mounts = self.__getMountPoints() + + def getFsType(self, path): + fs='' + for mount in self.mounts: + if path.startswith(mount) and len(self.mounts[mount])>len(fs): + fs = self.mounts[mount] + return fs + + def __getMountPoints(self): + """Parse /proc/mounts to get currently mounted devices""" + #TODO: reparse when a new device is added/a device is removed + ret = {} + try: + with open("/proc/mounts",'r') as mounts: + for line in mounts.readlines(): + fs_spec, fs_file, fs_vfstype, fs_mntops, fs_freq, fs_passno = line.split(' ') + ret[fs_file] = fs_vfstype + except: + error ("Can't read mounts table") + return ret + + def __appendToList(self, path, options): + """Add a file to the copy list + @param path: absolute path of file + @param options: options as return by optparse""" + debug ("Adding to copy list: %s (%s)", path, self.getFsType(path)) + self.copy_list.append((path, options)) + + def __appendDirToList(self, dirpath, options): + """Add recursively directory to the copy list + @param path: absolute path of dir + @param options: options as return by optparse""" + try: + for filename in os.listdir(dirpath): + filepath = os.path.join(dirpath,filename) + if os.path.isdir(filepath): + self.__appendDirToList(filepath, options) + else: + self.__appendToList(filepath,options) + except OSError,e: + error("Can't copy %(path)s: %(exception)s" % {'path':dirpath, 'exception':e.strerror}) + + def __checkArgs(self, options, source_path, args): + """Check thats args are files, and add them to copy list""" + for path in args: + abspath = os.path.normpath(os.path.join(source_path, path)) + if not os.path.exists(abspath): + warning("The path given in arg doesn't exist or is not accessible: %s",abspath) + else: + if os.path.isdir(abspath): + if not options.recursive: + warning ('omitting directory "%s"' % abspath) + else: + self.__appendDirToList(abspath, options) + else: + self.__appendToList(abspath,options) + + + def parseArguments(self, full_args=sys.argv[1:], source_path = os.getcwd()): + _usage=""" + %prog [options] FILE1 [FILE2 ...] DEST + + %prog --help for options list + """ + parser = OptionParser(usage=_usage,version=ABOUT) + + parser.add_option("-r", "--recursive", action="store_true", default=False, + help="copy directories recursively") + + (options, args) = parser.parse_args(full_args) + + if not self._main_instance: + if args: + info ("There is already one instance of %s running, pluging to it" % NAME_SHORT) + self.gcp_main.addArgs(os.getcwd(),full_args) + else: + error ("There is already one instance of %s running, but we have no argument to send" % NAME_SHORT) + sys.exit(1) + else: + debug("adding args to gcp: %s",args) + self.__checkArgs(options, source_path, args) + + def go(self): + self.loop = gobject.MainLoop() + try: + self.loop.run() + except KeyboardInterrupt: + info("User interruption: good bye") + + +if __name__ == "__main__": + gcp = GCP() + gcp.parseArguments() + if gcp._main_instance: + gcp.go() +