Mercurial > gcp
annotate gcp @ 4:9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
also added --force & --verbose options
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Sun, 26 Sep 2010 14:08:14 +0800 |
| parents | 6a24b5928980 |
| children | 7edbb0e0d9dd |
| rev | line source |
|---|---|
| 0 | 1 #!/usr/bin/python |
| 2 # -*- coding: utf-8 -*- | |
| 3 | |
| 4 """ | |
| 5 gcp: Goffi's CoPier | |
| 6 Copyright (C) 2010 Jérôme Poisson (goffi@goffi.org) | |
| 7 | |
| 8 This program is free software: you can redistribute it and/or modify | |
| 9 it under the terms of the GNU General Public License as published by | |
| 10 the Free Software Foundation, either version 3 of the License, or | |
| 11 (at your option) any later version. | |
| 12 | |
| 13 This program is distributed in the hope that it will be useful, | |
| 14 but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 16 GNU General Public License for more details. | |
| 17 | |
| 18 You should have received a copy of the GNU General Public License | |
| 19 along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 20 """ | |
| 21 | |
| 22 ### logging ### | |
| 23 import logging | |
| 24 from logging import debug, info, error, warning | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
25 logging.basicConfig(level=logging.INFO, |
| 0 | 26 format='%(message)s') |
| 27 ### | |
| 28 import sys | |
| 29 import os,os.path | |
| 30 from optparse import OptionParser #To be replaced by argparse ASAP | |
| 1 | 31 import cPickle as pickle |
| 0 | 32 try: |
| 33 import gobject | |
| 34 #DBus | |
| 35 import dbus, dbus.glib | |
| 36 import dbus.service | |
| 37 import dbus.mainloop.glib | |
| 38 except ImportError,e: | |
| 39 error("Error during import") | |
| 40 error("Please check dependecies:",e) | |
| 41 exit(2) | |
| 42 try: | |
| 43 from progressbar import ProgressBar, Percentage, Bar, ETA, FileTransferSpeed | |
| 44 pbar_available=True | |
| 45 except ImportError, e: | |
| 46 info ('ProgressBar not available, please download it at http://pypi.python.org/pypi/progressbar') | |
| 47 info ('Progress bar deactivated\n--\n') | |
| 48 pbar_available=False | |
| 49 | |
| 50 NAME = "gcp (Goffi's copier)" | |
| 51 NAME_SHORT = "gcp" | |
| 52 VERSION = '0.1' | |
| 53 | |
| 54 ABOUT = NAME+" v"+VERSION+""" (c) Jérôme Poisson (aka Goffi) 2010 | |
| 55 | |
| 56 --- | |
| 57 """+NAME+""" Copyright (C) 2010 Jérôme Poisson | |
| 58 This program comes with ABSOLUTELY NO WARRANTY; | |
| 59 This is free software, and you are welcome to redistribute it | |
| 60 under certain conditions. | |
| 61 --- | |
| 62 | |
| 63 This software is an advanced file copier | |
| 64 Get the latest version at http://www.goffi.org | |
| 65 """ | |
| 66 | |
| 67 const_DBUS_INTERFACE = "org.goffi.gcp" | |
| 3 | 68 const_DBUS_PATH = "/org/goffi/gcp" |
| 69 const_BUFF_SIZE = 4096 | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
70 const_PRESERVE = set(['mode','ownership','timestamps']) |
| 0 | 71 |
| 72 | |
| 73 class DbusObject(dbus.service.Object): | |
| 74 | |
| 75 def __init__(self, gcp, bus, path): | |
| 76 self._gcp = gcp | |
| 77 dbus.service.Object.__init__(self, bus, path) | |
| 78 debug("Init DbusObject...") | |
| 79 self.cb={} | |
| 80 | |
| 81 @dbus.service.method(const_DBUS_INTERFACE, | |
| 82 in_signature='', out_signature='s') | |
| 83 def getVersion(self): | |
| 84 """Get gcp version | |
| 85 @return: version as string""" | |
| 86 return VERSION | |
| 87 | |
| 88 @dbus.service.method(const_DBUS_INTERFACE, | |
| 1 | 89 in_signature='ss', out_signature='bs') |
| 0 | 90 def addArgs(self, source_path, args): |
| 91 """Add arguments to gcp as if there were entered on its own command line | |
| 1 | 92 @param source_path: current working dir to use as base for arguments, as given by os.getcwd() |
| 93 @param args: serialized (wich pickle) list of strings - without command name -, as given by sys.argv[1:]. | |
| 94 @return: success (boolean) and error message if any (string)""" | |
| 95 try: | |
| 96 args = pickle.loads(str(args)) | |
| 97 except TypeError, pickle.UnpicklingError: | |
| 98 return (False, "INTERNAL ERROR: invalid arguments") | |
| 0 | 99 return self._gcp.parseArguments(args, source_path) |
| 100 | |
| 101 class GCP(): | |
| 102 | |
| 103 def __init__(self): | |
| 104 try: | |
| 105 sessions_bus = dbus.SessionBus() | |
| 106 db_object = sessions_bus.get_object(const_DBUS_INTERFACE, | |
| 107 const_DBUS_PATH) | |
| 108 self.gcp_main = dbus.Interface(db_object, | |
| 109 dbus_interface=const_DBUS_INTERFACE) | |
| 110 self._main_instance = False | |
| 111 | |
| 112 except dbus.exceptions.DBusException,e: | |
| 113 if e._dbus_error_name=='org.freedesktop.DBus.Error.ServiceUnknown': | |
| 114 self.launchDbusMainInstance() | |
| 115 debug ("gcp launched") | |
| 116 self._main_instance = True | |
| 3 | 117 self.buffer_size = const_BUFF_SIZE |
| 0 | 118 else: |
| 119 raise e | |
| 120 | |
| 121 def launchDbusMainInstance(self): | |
| 122 debug ("Init DBus...") | |
| 123 session_bus = dbus.SessionBus() | |
| 124 self.dbus_name = dbus.service.BusName(const_DBUS_INTERFACE, session_bus) | |
| 125 self.dbus_object = DbusObject(self, session_bus, const_DBUS_PATH) | |
| 126 | |
| 127 self.copy_list = [] | |
| 128 self.mounts = self.__getMountPoints() | |
| 1 | 129 self.files_left = 0 |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
130 self.bytes_total = 0 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
131 self.bytes_copied = 0 |
| 0 | 132 |
| 133 def getFsType(self, path): | |
| 134 fs='' | |
| 135 for mount in self.mounts: | |
| 2 | 136 if path.startswith(mount) and len(self.mounts[mount])>=len(fs): |
| 0 | 137 fs = self.mounts[mount] |
| 138 return fs | |
| 139 | |
| 140 def __getMountPoints(self): | |
| 141 """Parse /proc/mounts to get currently mounted devices""" | |
| 142 #TODO: reparse when a new device is added/a device is removed | |
| 1 | 143 #(check freedesktop mounting signals) |
| 0 | 144 ret = {} |
| 145 try: | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
146 with open("/proc/mounts",'rb') as mounts: |
| 0 | 147 for line in mounts.readlines(): |
| 148 fs_spec, fs_file, fs_vfstype, fs_mntops, fs_freq, fs_passno = line.split(' ') | |
| 149 ret[fs_file] = fs_vfstype | |
| 150 except: | |
| 151 error ("Can't read mounts table") | |
| 152 return ret | |
| 153 | |
| 3 | 154 def __appendToList(self, path, dest_path, options): |
| 0 | 155 """Add a file to the copy list |
| 156 @param path: absolute path of file | |
| 157 @param options: options as return by optparse""" | |
| 3 | 158 debug ("Adding to copy list: %s ==> %s (%s)", path, dest_path, self.getFsType(dest_path)) |
| 1 | 159 try: |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
160 self.bytes_total+=os.path.getsize(path) |
| 1 | 161 self.files_left+=1 |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
162 self.copy_list.insert(0,(path, dest_path, options)) |
| 1 | 163 except OSError,e: |
| 164 error("Can't copy %(path)s: %(exception)s" % {'path':path, 'exception':e.strerror}) | |
| 0 | 165 |
| 1 | 166 |
| 3 | 167 def __appendDirToList(self, dirpath, dest_path, options): |
| 0 | 168 """Add recursively directory to the copy list |
| 169 @param path: absolute path of dir | |
| 170 @param options: options as return by optparse""" | |
| 3 | 171 #We first check that the dest path exists, and create it if needed |
| 172 if not os.path.exists(dest_path): | |
| 173 debug ("Creating directory %s" % dest_path) | |
| 174 os.makedirs(dest_path) #TODO: check permissions | |
| 175 #TODO: check that dest_path is an accessible dir, | |
| 176 # and skip file/write error in log if needed | |
| 0 | 177 try: |
| 178 for filename in os.listdir(dirpath): | |
| 179 filepath = os.path.join(dirpath,filename) | |
| 180 if os.path.isdir(filepath): | |
| 3 | 181 full_dest_path = os.path.join(dest_path,filename) |
| 182 self.__appendDirToList(filepath, full_dest_path, options) | |
| 0 | 183 else: |
| 3 | 184 self.__appendToList(filepath, dest_path, options) |
| 0 | 185 except OSError,e: |
| 186 error("Can't copy %(path)s: %(exception)s" % {'path':dirpath, 'exception':e.strerror}) | |
| 187 | |
| 188 def __checkArgs(self, options, source_path, args): | |
| 189 """Check thats args are files, and add them to copy list""" | |
| 1 | 190 assert(len (args)>=2) |
| 191 try: | |
| 3 | 192 dest_path = os.path.normpath(os.path.join(os.path.expanduser(source_path), args.pop())) |
| 1 | 193 except OSError,e: |
| 3 | 194 error ("Invalid dest_path: %s",e) |
| 1 | 195 |
| 0 | 196 for path in args: |
| 1 | 197 abspath = os.path.normpath(os.path.join(os.path.expanduser(source_path), path)) |
| 0 | 198 if not os.path.exists(abspath): |
| 199 warning("The path given in arg doesn't exist or is not accessible: %s",abspath) | |
| 200 else: | |
| 201 if os.path.isdir(abspath): | |
| 3 | 202 full_dest_path = dest_path if os.path.isabs(path) else os.path.normpath(os.path.join(dest_path, path)) |
| 0 | 203 if not options.recursive: |
| 204 warning ('omitting directory "%s"' % abspath) | |
| 205 else: | |
| 3 | 206 self.__appendDirToList(abspath, full_dest_path, options) |
| 0 | 207 else: |
| 3 | 208 self.__appendToList(abspath, dest_path, options) |
| 0 | 209 |
| 3 | 210 def __copyNextFile(self): |
| 211 """Take the last file in the list, and launch the copy using glib io_watch event | |
| 212 @return: True a file was added, False else""" | |
| 213 if self.copy_list: | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
214 source_file, dest_path, options = self.copy_list.pop() |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
215 source_fd = open(source_file, 'rb') |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
216 filename = os.path.basename(source_file) |
| 3 | 217 assert(filename) |
| 218 dest_file = os.path.join(dest_path,filename) | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
219 if os.path.exists(dest_file) and not options.force: |
| 3 | 220 warning ("File [%s] already exists, skipping it !" % dest_file) |
| 221 return True | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
222 dest_fd = open(dest_file, 'wb') |
| 3 | 223 |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
224 gobject.io_add_watch(source_fd,gobject.IO_IN,self._copyFile, |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
225 (dest_fd, options), priority=gobject.PRIORITY_HIGH) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
226 if not self.progress: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
227 info("COPYING %(source)s ==> %(dest)s" % {"source":source_path,"dest":dest_file}) |
| 3 | 228 return True |
| 229 else: | |
| 230 #Nothing left to copy, we quit | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
231 if self.progress: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
232 self.__pbar_finish() |
| 3 | 233 self.loop.quit() |
| 234 | |
| 235 def _copyFile(self, source_fd, condition, data): | |
| 236 """Actually copy the file, callback used with io_add_watch | |
| 237 @param source_fd: file descriptor of the file to copy | |
| 238 @param condition: condition which launched the callback (glib.IO_IN) | |
| 239 @param data: tuple with (destination file descriptor, copying options)""" | |
| 240 dest_fd,options = data | |
| 241 buff = source_fd.read(self.buffer_size) | |
| 242 dest_fd.write(buff) | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
243 self.bytes_copied += len(buff) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
244 if self.progress: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
245 self.__pbar_update() |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
246 |
| 3 | 247 if len(buff) != self.buffer_size: |
| 248 source_fd.close() | |
| 249 dest_fd.close() | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
250 self.__post_copy(source_fd.name, dest_fd.name, options) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
251 self.files_left -= 1 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
252 assert (self.files_left >= 0) |
| 3 | 253 return False |
| 254 return True | |
| 255 | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
256 def __post_copy(self, source_file, dest_file, options): |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
257 """Do post copy traitement (mainly managing --preserve option)""" |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
258 st_mode, st_ino, st_dev, st_nlink, st_uid, st_gid, st_size, st_atime, st_mtime, st_ctime = os.stat(source_file) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
259 #TODO: complete log in case of errors |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
260 for preserve in options.preserve: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
261 try: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
262 if preserve == 'mode': |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
263 os.chmod(dest_file, st_mode) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
264 elif preserve == 'ownership': |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
265 os.chown(dest_file, st_uid, st_gid) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
266 elif preserve == 'timestamps': |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
267 os.utime(dest_file, (st_atime, st_mtime)) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
268 except OSError,e: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
269 pass #TODO: complete log here |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
270 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
271 def __pbar_update(self): |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
272 """Update progress bar position, create the bar if it doesn't exist""" |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
273 assert(self.progress) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
274 try: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
275 if self.pbar.maxval != self.bytes_total: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
276 self.pbar.maxval = self.bytes_total |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
277 except AttributeError: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
278 self.pbar = ProgressBar(self.bytes_total,["Progress: ",Percentage()," ",Bar()," ",FileTransferSpeed()," ",ETA()]) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
279 self.pbar.start() |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
280 self.pbar.update(self.bytes_copied) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
281 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
282 def __pbar_finish(self): |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
283 """Mark the progression as finished""" |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
284 assert(self.progress) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
285 try: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
286 self.pbar.finish() |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
287 except AttributeError: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
288 pass |
| 0 | 289 |
| 290 def parseArguments(self, full_args=sys.argv[1:], source_path = os.getcwd()): | |
| 1 | 291 """Parse arguments and add files to queue |
| 292 @param full_args: list of arguments strings (without program name) | |
| 293 @param source_path: path from where the arguments come, ad given by os.getcwd() | |
| 294 @return: a tuple (boolean, message) where the boolean is the success of the arguments | |
| 295 validation, and message is the error message to print when necessary""" | |
| 0 | 296 _usage=""" |
| 297 %prog [options] FILE1 [FILE2 ...] DEST | |
| 298 | |
| 299 %prog --help for options list | |
| 300 """ | |
| 1 | 301 for idx in range(len(full_args)): |
| 302 if isinstance(full_args[idx], unicode): | |
| 303 #We don't want unicode as some filenames can be invalid unicode | |
| 304 full_args[idx] = full_args[idx].encode('utf-8') | |
| 305 | |
| 0 | 306 parser = OptionParser(usage=_usage,version=ABOUT) |
| 307 | |
| 308 parser.add_option("-r", "--recursive", action="store_true", default=False, | |
| 309 help="copy directories recursively") | |
| 310 | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
311 parser.add_option("-f", "--force", action="store_true", default=False, |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
312 help="force overwriting of existing files") |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
313 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
314 parser.add_option("--preserve", action="store", default='mode,ownership,timestamps', |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
315 help="preserve the specified attributes") |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
316 |
| 1 | 317 parser.add_option("--no-unicode-fix", action="store_true", default=False, |
| 318 help="don't fixe name encoding errors") #TODO | |
| 3 | 319 |
| 320 parser.add_option("--no-progress", action="store_false", dest="progress", default=True, | |
| 321 help="deactivate progress bar") | |
| 322 | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
323 parser.add_option("-v", "--verbose", action="store_true", default=False, |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
324 help="Show what is currently done") |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
325 |
| 0 | 326 (options, args) = parser.parse_args(full_args) |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
327 #options check |
| 3 | 328 if options.progress and not pbar_available: |
| 329 warning ("Progress bar is not available, deactivating") | |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
330 options.progress = self.progress = False |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
331 else: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
332 self.progress = options.progress |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
333 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
334 if options.verbose: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
335 logging.getLogger().setLevel(logging.DEBUG) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
336 |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
337 preserve = set(options.preserve.split(',')) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
338 if not preserve.issubset(const_PRESERVE): |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
339 error ('Invalide --preserve value\nvalid values are:') |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
340 for value in const_PRESERVE: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
341 error('- %s' % value) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
342 exit(2) |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
343 else: |
|
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
344 options.preserve = preserve |
| 0 | 345 |
|
4
9feb82bd91aa
--preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents:
3
diff
changeset
|
346 #if there is an other instance of gcp, we send options to it |
| 0 | 347 if not self._main_instance: |
| 1 | 348 info ("There is already one instance of %s running, pluging to it" % NAME_SHORT) |
| 349 #XXX: we have to serialize data as dbus only accept valid unicode, and filenames | |
| 350 # can have invalid unicode. | |
| 351 return self.gcp_main.addArgs(os.getcwd(),pickle.dumps(full_args)) | |
| 0 | 352 else: |
| 1 | 353 if len(args) < 2: |
| 354 _error_msg = "Wrong number of arguments" | |
| 355 return (False, _error_msg) | |
| 0 | 356 debug("adding args to gcp: %s",args) |
| 357 self.__checkArgs(options, source_path, args) | |
| 3 | 358 gobject.idle_add(self.__copyNextFile) |
| 1 | 359 return (True,'') |
| 0 | 360 |
| 361 def go(self): | |
| 1 | 362 """Launch main loop""" |
| 0 | 363 self.loop = gobject.MainLoop() |
| 364 try: | |
| 365 self.loop.run() | |
| 366 except KeyboardInterrupt: | |
| 367 info("User interruption: good bye") | |
| 368 | |
| 369 | |
| 370 if __name__ == "__main__": | |
| 371 gcp = GCP() | |
| 1 | 372 success,message = gcp.parseArguments() |
| 373 if not success: | |
| 374 error(message) | |
| 375 exit(1) | |
| 0 | 376 if gcp._main_instance: |
| 377 gcp.go() | |
| 378 |