annotate gcp @ 8:144cb2669f21

removed some forgotten TODO
author Goffi <goffi@goffi.org>
date Mon, 27 Sep 2010 17:33:25 +0800
parents a110d31482f7
children 599b84e4ff01
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
1 #!/usr/bin/python
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
2 # -*- coding: utf-8 -*-
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
3
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
4 """
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
5 gcp: Goffi's CoPier
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
6 Copyright (C) 2010 Jérôme Poisson (goffi@goffi.org)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
7
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
8 This program is free software: you can redistribute it and/or modify
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
9 it under the terms of the GNU General Public License as published by
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
10 the Free Software Foundation, either version 3 of the License, or
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
11 (at your option) any later version.
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
12
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
13 This program is distributed in the hope that it will be useful,
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
16 GNU General Public License for more details.
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
17
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
18 You should have received a copy of the GNU General Public License
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
20 """
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
21
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
22 ### logging ###
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
23 import logging
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
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
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
26 format='%(message)s')
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
27 ###
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
28
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
29 import gettext
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
30 gettext.install('gcp', "i18n", unicode=True)
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
31
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
32 import sys
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
33 import os,os.path
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
34 from optparse import OptionParser #To be replaced by argparse ASAP
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
35 import cPickle as pickle
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
36 try:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
37 import gobject
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
38 #DBus
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
39 import dbus, dbus.glib
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
40 import dbus.service
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
41 import dbus.mainloop.glib
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
42 except ImportError,e:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
43 error(_("Error during import"))
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
44 error(_("Please check dependecies:"),e)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
45 exit(2)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
46 try:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
47 from progressbar import ProgressBar, Percentage, Bar, ETA, FileTransferSpeed
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
48 pbar_available=True
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
49 except ImportError, e:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
50 info (_('ProgressBar not available, please download it at http://pypi.python.org/pypi/progressbar'))
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
51 info (_('Progress bar deactivated\n--\n'))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
52 pbar_available=False
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
53
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
54 NAME = "gcp (Goffi's copier)"
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
55 NAME_SHORT = "gcp"
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
56 VERSION = '0.1'
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
57
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
58 ABOUT = NAME+u" v"+VERSION+u""" (c) Jérôme Poisson (aka Goffi) 2010
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
59
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
60 ---
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
61 """+NAME+u""" Copyright (C) 2010 Jérôme Poisson
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
62 """ + _(u"""This program comes with ABSOLUTELY NO WARRANTY;
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
63 This is free software, and you are welcome to redistribute it
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
64 under certain conditions.
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
65 ---
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
66
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
67 This software is an advanced file copier
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
68 Get the latest version at http://www.goffi.org
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
69 """)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
70
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
71 const_DBUS_INTERFACE = "org.goffi.gcp"
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
72 const_DBUS_PATH = "/org/goffi/gcp"
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
73 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
74 const_PRESERVE = set(['mode','ownership','timestamps'])
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
75 const_JOURNAL_PATH = "~/.gcp_journal"
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
76
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
77
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
78 class DbusObject(dbus.service.Object):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
79
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
80 def __init__(self, gcp, bus, path):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
81 self._gcp = gcp
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
82 dbus.service.Object.__init__(self, bus, path)
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
83 debug(_("Init DbusObject..."))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
84 self.cb={}
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
85
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
86 @dbus.service.method(const_DBUS_INTERFACE,
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
87 in_signature='', out_signature='s')
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
88 def getVersion(self):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
89 """Get gcp version
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
90 @return: version as string"""
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
91 return VERSION
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
92
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
93 @dbus.service.method(const_DBUS_INTERFACE,
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
94 in_signature='ss', out_signature='bs')
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
95 def addArgs(self, source_path, args):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
96 """Add arguments to gcp as if there were entered on its own command line
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
97 @param source_path: current working dir to use as base for arguments, as given by os.getcwd()
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
98 @param args: serialized (wich pickle) list of strings - without command name -, as given by sys.argv[1:].
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
99 @return: success (boolean) and error message if any (string)"""
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
100 try:
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
101 args = pickle.loads(str(args))
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
102 except TypeError, pickle.UnpicklingError:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
103 return (False, _("INTERNAL ERROR: invalid arguments"))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
104 return self._gcp.parseArguments(args, source_path)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
105
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
106 class Journal():
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
107 def __init__(self, path=const_JOURNAL_PATH):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
108 self.journal_path = os.path.expanduser(path)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
109 self.journal_fd = open(self.journal_path,'w') #TODO: check and maybe save previous journals
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
110
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
111 def __del__(self):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
112 self.journal_fd.flush()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
113 self.journal_fd.close()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
114
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
115 def startFile(self, source_path):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
116 """Start an entry in the journal"""
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
117 self.journal_fd.write(source_path+"\n")
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
118 self.journal_fd.flush()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
119 self.success=True
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
120 self.errors=[]
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
121
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
122 def closeFile(self):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
123 """Close the entry in the journal"""
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
124 if not self.success:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
125 status = "FAILED"
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
126 else:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
127 status = "OK" if not self.errors else "PARTIAL"
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
128 self.journal_fd.write("%(status)s: %(errors)s\n" % {'status': status, 'errors': ', '.join(self.errors)})
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
129 self.journal_fd.flush()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
130
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
131 def copyFailed(self):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
132 """Must be called when something is wrong with the copy itself"""
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
133 self.success = False
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
134
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
135 def error(self, name):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
136 """Something went wrong"""
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
137 self.errors.append(name)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
138
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
139
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
140 class GCP():
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
141
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
142 def __init__(self):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
143 try:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
144 sessions_bus = dbus.SessionBus()
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
145 db_object = sessions_bus.get_object(const_DBUS_INTERFACE,
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
146 const_DBUS_PATH)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
147 self.gcp_main = dbus.Interface(db_object,
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
148 dbus_interface=const_DBUS_INTERFACE)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
149 self._main_instance = False
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
150
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
151 except dbus.exceptions.DBusException,e:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
152 if e._dbus_error_name=='org.freedesktop.DBus.Error.ServiceUnknown':
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
153 self.launchDbusMainInstance()
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
154 debug (_("gcp launched"))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
155 self._main_instance = True
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
156 self.buffer_size = const_BUFF_SIZE
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
157 else:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
158 raise e
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
159
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
160 def launchDbusMainInstance(self):
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
161 debug (_("Init DBus..."))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
162 session_bus = dbus.SessionBus()
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
163 self.dbus_name = dbus.service.BusName(const_DBUS_INTERFACE, session_bus)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
164 self.dbus_object = DbusObject(self, session_bus, const_DBUS_PATH)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
165
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
166 self.copy_list = []
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
167 self.mounts = self.__getMountPoints()
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
168 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
169 self.bytes_copied = 0
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
170
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
171 def getFsType(self, path):
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
172 fs = ''
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
173 last_mount_point = ''
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
174 for mount in self.mounts:
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
175 if path.startswith(mount) and len(mount)>=len(last_mount_point):
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
176 fs = self.mounts[mount]
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
177 last_mount_point = mount
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
178 return fs
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
179
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
180 def __getMountPoints(self):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
181 """Parse /proc/mounts to get currently mounted devices"""
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
182 #TODO: reparse when a new device is added/a device is removed
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
183 #(check freedesktop mounting signals)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
184 ret = {}
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
185 try:
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
186 with open("/proc/mounts",'rb') as mounts:
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
187 for line in mounts.readlines():
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
188 fs_spec, fs_file, fs_vfstype, fs_mntops, fs_freq, fs_passno = line.split(' ')
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
189 ret[fs_file] = fs_vfstype
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
190 except:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
191 error (_("Can't read mounts table"))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
192 return ret
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
193
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
194 def __appendToList(self, path, dest_path, options):
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
195 """Add a file to the copy list
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
196 @param path: absolute path of file
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
197 @param options: options as return by optparse"""
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
198 debug ("Adding to copy list: %(path)s ==> %(dest_path)s (%(fs_type)s)" % {"path":path,
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
199 "dest_path":dest_path,
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
200 "fs_type":self.getFsType(dest_path)} )
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
201 try:
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
202 self.bytes_total+=os.path.getsize(path)
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
203 self.copy_list.insert(0,(path, dest_path, options))
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
204 except OSError,e:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
205 error(_("Can't copy %(path)s: %(exception)s") % {'path':path, 'exception':e.strerror})
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
206
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
207
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
208 def __appendDirToList(self, dirpath, dest_path, options):
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
209 """Add recursively directory to the copy list
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
210 @param path: absolute path of dir
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
211 @param options: options as return by optparse"""
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
212 #We first check that the dest path exists, and create it if needed
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
213 dest_path = self.__filename_fix(dest_path, options)
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
214 if not os.path.exists(dest_path):
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
215 debug (_("Creating directory %s") % dest_path)
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
216 os.makedirs(dest_path) #TODO: check permissions
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
217 #TODO: check that dest_path is an accessible dir,
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
218 # and skip file/write error in log if needed
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
219 try:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
220 for filename in os.listdir(dirpath):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
221 filepath = os.path.join(dirpath,filename)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
222 if os.path.isdir(filepath):
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
223 full_dest_path = os.path.join(dest_path,filename)
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
224 self.__appendDirToList(filepath, full_dest_path, options)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
225 else:
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
226 self.__appendToList(filepath, dest_path, options)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
227 except OSError,e:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
228 error(_("Can't copy %(path)s: %(exception)s") % {'path':dirpath, 'exception':e.strerror})
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
229
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
230 def __checkArgs(self, options, source_path, args):
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
231 """Check thats args are files, and add them to copy list"""
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
232 assert(len (args)>=2)
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
233 try:
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
234 dest_path = os.path.normpath(os.path.join(os.path.expanduser(source_path), args.pop()))
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
235 except OSError,e:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
236 error (_("Invalid dest_path: %s"),e)
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
237
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
238 for path in args:
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
239 abspath = os.path.normpath(os.path.join(os.path.expanduser(source_path), path))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
240 if not os.path.exists(abspath):
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
241 warning(_("The path given in arg doesn't exist or is not accessible: %s") % abspath)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
242 else:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
243 if os.path.isdir(abspath):
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
244 full_dest_path = dest_path if os.path.isabs(path) else os.path.normpath(os.path.join(dest_path, path))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
245 if not options.recursive:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
246 warning (_('omitting directory "%s"') % abspath)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
247 else:
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
248 self.__appendDirToList(abspath, full_dest_path, options)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
249 else:
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
250 self.__appendToList(abspath, dest_path, options)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
251
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
252 def __copyNextFile(self):
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
253 """Take the last file in the list, and launch the copy using glib io_watch event
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
254 @return: True a file was added, False else"""
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
255 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
256 source_file, dest_path, options = self.copy_list.pop()
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
257 self.journal.startFile(source_file)
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
258 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
259 filename = os.path.basename(source_file)
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
260 assert(filename)
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
261 dest_file = self.__filename_fix(os.path.join(dest_path,filename),options)
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
262 if os.path.exists(dest_file) and not options.force:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
263 warning (_("File [%s] already exists, skipping it !") % dest_file)
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
264 return True
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
265 try:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
266 dest_fd = open(dest_file, 'wb')
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
267 except:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
268 self.journal.copyFailed()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
269 self.journal.error("can't open dest")
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
270 self.journal.closeFile()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
271 source_fd.close()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
272 return True
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
273
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
274 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
275 (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
276 if not self.progress:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
277 info(_("COPYING %(source)s ==> %(dest)s") % {"source":source_path,"dest":dest_file})
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
278 return True
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
279 else:
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
280 #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
281 if self.progress:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
282 self.__pbar_finish()
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
283 self.loop.quit()
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
284
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
285 def __copyFailed(self, reason, source_fd, dest_fd):
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
286 """Write the failure in the journal and close files descriptors"""
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
287 self.journal.copyFailed()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
288 self.journal.error(reason)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
289 self.journal.closeFile()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
290 source_fd.close()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
291 dest_fd.close()
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
292
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
293
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
294
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
295 def _copyFile(self, source_fd, condition, data):
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
296 """Actually copy the file, callback used with io_add_watch
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
297 @param source_fd: file descriptor of the file to copy
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
298 @param condition: condition which launched the callback (glib.IO_IN)
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
299 @param data: tuple with (destination file descriptor, copying options)"""
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
300 dest_fd,options = data
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
301
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
302 try:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
303 buff = source_fd.read(self.buffer_size)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
304 except:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
305 self.__copyFailed("can't read source", source_fd, dest_fd)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
306 return False
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
307
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
308 try:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
309 dest_fd.write(buff)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
310 except:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
311 self.__copyFailed("can't write to dest", source_fd, dest_fd)
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
312 return False
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
313
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
314 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
315 if self.progress:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
316 self.__pbar_update()
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
317
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
318 if len(buff) != self.buffer_size:
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
319 source_fd.close()
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
320 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
321 self.__post_copy(source_fd.name, dest_fd.name, options)
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
322 self.journal.closeFile()
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
323 return False
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
324 return True
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
325
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
326 def __filename_fix(self, filename, options):
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
327 """Fix filenames incompatibilities/mistake according to options
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
328 @param filename: full path to the file
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
329 @param options: options as parsed on command line
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
330 @return: fixed filename"""
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
331 fixed_filename = filename
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
332
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
333 if self.getFsType(filename) == 'vfat' and options.fs_fix:
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
334 fixed_filename = filename.replace('\\','_')\
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
335 .replace(':',';')\
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
336 .replace('*','+')\
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
337 .replace('?','')\
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
338 .replace('"','\'')\
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
339 .replace('<','[')\
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
340 .replace('>',']')\
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
341 .replace('|','!')
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
342
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
343 if fixed_filename != filename:
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
344 self.journal.error('filename fixed')
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
345 return fixed_filename
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
346
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
347 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
348 """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
349 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
350 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
351 try:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
352 if preserve == 'mode':
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
353 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
354 elif preserve == 'ownership':
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
355 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
356 elif preserve == 'timestamps':
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
357 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
358 except OSError,e:
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
359 self.journal.error("preserve-"+preserve)
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
360
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
361 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
362 """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
363 assert(self.progress)
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
364 try:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
365 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
366 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
367 except AttributeError:
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
368 if not self.bytes_total:
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
369 #No progress bar if the files have a null size
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
370 return
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
371 self.pbar = ProgressBar(self.bytes_total,[_("Progress: "),Percentage()," ",Bar()," ",FileTransferSpeed()," ",ETA()])
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
372 self.pbar.start()
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
373 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
374
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
375 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
376 """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
377 assert(self.progress)
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
378 try:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
379 self.pbar.finish()
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
380 except AttributeError:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
381 pass
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
382
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
383 def parseArguments(self, full_args=sys.argv[1:], source_path = os.getcwd()):
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
384 """Parse arguments and add files to queue
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
385 @param full_args: list of arguments strings (without program name)
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
386 @param source_path: path from where the arguments come, ad given by os.getcwd()
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
387 @return: a tuple (boolean, message) where the boolean is the success of the arguments
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
388 validation, and message is the error message to print when necessary"""
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
389 _usage="""
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
390 %prog [options] FILE1 [FILE2 ...] DEST
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
391
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
392 %prog --help for options list
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
393 """
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
394 for idx in range(len(full_args)):
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
395 if isinstance(full_args[idx], unicode):
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
396 #We don't want unicode as some filenames can be invalid unicode
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
397 full_args[idx] = full_args[idx].encode('utf-8')
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
398
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
399 parser = OptionParser(usage=_usage,version=ABOUT)
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
400
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
401 parser.add_option("-r", "--recursive", action="store_true", default=False,
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
402 help=_("copy directories recursively"))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
403
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
404 parser.add_option("-f", "--force", action="store_true", default=False,
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
405 help=_("force overwriting of existing files"))
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
406
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
407 parser.add_option("--preserve", action="store", default='mode,ownership,timestamps',
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
408 help=_("preserve the specified attributes"))
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
409
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
410 parser.add_option("--no-unicode-fix", action="store_false", dest='unicode_fix', default=True,
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
411 help=_("don't fixe name encoding errors")) #TODO
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
412
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
413 parser.add_option("--no-fs-fix", action="store_false", dest='fs_fix', default=True,
8
144cb2669f21 removed some forgotten TODO
Goffi <goffi@goffi.org>
parents: 7
diff changeset
414 help=_("don't fixe filesystem name incompatibily"))
6
5f53dc5beec9 vfat names incompatibility fix
Goffi <goffi@goffi.org>
parents: 5
diff changeset
415
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
416 parser.add_option("--no-progress", action="store_false", dest="progress", default=True,
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
417 help=_("deactivate progress bar"))
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
418
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
419 parser.add_option("-v", "--verbose", action="store_true", default=False,
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
420 help=_("Show what is currently done"))
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
421
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
422 (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
423 #options check
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
424 if options.progress and not pbar_available:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
425 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
426 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
427 else:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
428 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
429
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
430 if options.verbose:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
431 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
432
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
433 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
434 if not preserve.issubset(const_PRESERVE):
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
435 error (_("Invalide --preserve value\nvalid values are:"))
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
436 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
437 error('- %s' % value)
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
438 exit(2)
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
439 else:
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
440 options.preserve = preserve
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
441
4
9feb82bd91aa --preserve option (same default as for cp) & progress bar are now working.
Goffi <goffi@goffi.org>
parents: 3
diff changeset
442 #if there is an other instance of gcp, we send options to it
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
443 if not self._main_instance:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
444 info (_("There is already one instance of %s running, pluging to it") % NAME_SHORT)
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
445 #XXX: we have to serialize data as dbus only accept valid unicode, and filenames
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
446 # can have invalid unicode.
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
447 return self.gcp_main.addArgs(os.getcwd(),pickle.dumps(full_args))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
448 else:
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
449 if len(args) < 2:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
450 _error_msg = _("Wrong number of arguments")
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
451 return (False, _error_msg)
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
452 debug(_("adding args to gcp: %s"),args)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
453 self.__checkArgs(options, source_path, args)
7
a110d31482f7 Journalisation + some additionnal try/except
Goffi <goffi@goffi.org>
parents: 6
diff changeset
454 self.journal = Journal()
3
6a24b5928980 Basic file copying
Goffi <goffi@goffi.org>
parents: 2
diff changeset
455 gobject.idle_add(self.__copyNextFile)
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
456 return (True,'')
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
457
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
458 def go(self):
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
459 """Launch main loop"""
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
460 self.loop = gobject.MainLoop()
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
461 try:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
462 self.loop.run()
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
463 except KeyboardInterrupt:
5
7edbb0e0d9dd gettext inclusion & French translation
Goffi <goffi@goffi.org>
parents: 4
diff changeset
464 info(_("User interruption: good bye"))
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
465
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
466
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
467 if __name__ == "__main__":
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
468 gcp = GCP()
1
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
469 success,message = gcp.parseArguments()
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
470 if not success:
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
471 error(message)
995487813501 Arguments management
Goffi <goffi@goffi.org>
parents: 0
diff changeset
472 exit(1)
0
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
473 if gcp._main_instance:
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
474 gcp.go()
674ce820a4ef initial commit
Goffi <goffi@goffi.org>
parents:
diff changeset
475