← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/ftp.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2007-11-04 18:51:39 UTC
  • mfrom: (2961.1.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20071104185139-kaio3sneodg2kp71
Authentication ring implementation (read-only)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/ftp.py
27
27
from cStringIO import StringIO
28
28
import errno
29
29
import ftplib
30
 
import getpass
31
30
import os
32
31
import os.path
33
32
import urlparse
34
 
import socket
35
33
import stat
36
34
import time
37
35
import random
89
87
 
90
88
    def __init__(self, base, _from_transport=None):
91
89
        """Set the base path where files will be stored."""
92
 
        if not (base.startswith('ftp://') or base.startswith('aftp://')):
93
 
            raise ValueError(base)
 
90
        assert base.startswith('ftp://') or base.startswith('aftp://')
94
91
        super(FtpTransport, self).__init__(base,
95
92
                                           _from_transport=_from_transport)
96
93
        self._unqualified_scheme = 'ftp'
144
141
                                             port=self._port)
145
142
            connection.login(user=user, passwd=password)
146
143
            connection.set_pasv(not self.is_active)
147
 
        except socket.error, e:
148
 
            raise errors.SocketConnectionError(self._host, self._port,
149
 
                                               msg='Unable to connect to',
150
 
                                               orig_error= e)
151
144
        except ftplib.error_perm, e:
152
145
            raise errors.TransportError(msg="Error setting up connection:"
153
146
                                        " %s" % str(e), orig_error=e)
179
172
            or 'no such dir' in s
180
173
            or 'could not create file' in s # vsftpd
181
174
            or 'file doesn\'t exist' in s
182
 
            or 'rnfr command failed.' in s # vsftpd RNFR reply if file not found
183
175
            or 'file/directory not found' in s # filezilla server
184
 
            # Microsoft FTP-Service RNFR reply if file not found
185
 
            or (s.startswith('550 ') and 'unable to rename to' in extra)
186
176
            ):
187
177
            raise errors.NoSuchFile(path, extra=extra)
188
178
        if ('file exists' in s):
310
300
            try:
311
301
                f.storbinary('STOR '+tmp_abspath, fp)
312
302
                self._rename_and_overwrite(tmp_abspath, abspath, f)
313
 
                self._setmode(relpath, mode)
314
303
                if bytes is not None:
315
304
                    return len(bytes)
316
305
                else:
352
341
            mutter("FTP mkd: %s", abspath)
353
342
            f = self._get_FTP()
354
343
            f.mkd(abspath)
355
 
            self._setmode(relpath, mode)
356
344
        except ftplib.error_perm, e:
357
345
            self._translate_perm_error(e, abspath,
358
346
                unknown_exc=errors.FileExists)
413
401
            conn = ftp.transfercmd(cmd)
414
402
            conn.sendall(text)
415
403
            conn.close()
416
 
            self._setmode(relpath, mode)
 
404
            if mode:
 
405
                self._setmode(relpath, mode)
417
406
            ftp.getresp()
418
407
        except ftplib.error_perm, e:
419
408
            self._translate_perm_error(e, abspath, extra='error appending',
433
422
        Only set permissions if the FTP server supports the 'SITE CHMOD'
434
423
        extension.
435
424
        """
436
 
        if mode:
437
 
            try:
438
 
                mutter("FTP site chmod: setting permissions to %s on %s",
439
 
                    str(mode), self._remote_path(relpath))
440
 
                ftp = self._get_FTP()
441
 
                cmd = "SITE CHMOD %s %s" % (oct(mode),
442
 
                                            self._remote_path(relpath))
443
 
                ftp.sendcmd(cmd)
444
 
            except ftplib.error_perm, e:
445
 
                # Command probably not available on this server
446
 
                warning("FTP Could not set permissions to %s on %s. %s",
447
 
                        str(mode), self._remote_path(relpath), str(e))
 
425
        try:
 
426
            mutter("FTP site chmod: setting permissions to %s on %s",
 
427
                str(mode), self._remote_path(relpath))
 
428
            ftp = self._get_FTP()
 
429
            cmd = "SITE CHMOD %s %s" % (self._remote_path(relpath), str(mode))
 
430
            ftp.sendcmd(cmd)
 
431
        except ftplib.error_perm, e:
 
432
            # Command probably not available on this server
 
433
            warning("FTP Could not set permissions to %s on %s. %s",
 
434
                    str(mode), self._remote_path(relpath), str(e))
448
435
 
449
436
    # TODO: jam 20060516 I believe ftp allows you to tell an ftp server
450
437
    #       to copy something to another machine. And you may be able
519
506
            paths = f.nlst(basepath)
520
507
        except ftplib.error_perm, e:
521
508
            self._translate_perm_error(e, relpath, extra='error with list_dir')
522
 
        except ftplib.error_temp, e:
523
 
            # xs4all's ftp server raises a 450 temp error when listing an empty
524
 
            # directory. Check for that and just return an empty list in that
525
 
            # case. See bug #215522
526
 
            if str(e).lower().startswith('450 no files found'):
527
 
                mutter('FTP Server returned "%s" for nlst.'
528
 
                       ' Assuming it means empty directory',
529
 
                       str(e))
530
 
                return []
531
 
            raise
532
509
        # If FTP.nlst returns paths prefixed by relpath, strip 'em
533
510
        if paths and paths[0].startswith(basepath):
534
511
            entries = [path[len(basepath)+1:] for path in paths]
593
570
        return [(FtpTransport, ftp_server.FTPServer)]
594
571
    else:
595
572
        # Dummy server to have the test suite report the number of tests
596
 
        # needing that feature. We raise UnavailableFeature from methods before
597
 
        # the test server is being used. Doing so in the setUp method has bad
598
 
        # side-effects (tearDown is never called).
 
573
        # needing that feature.
599
574
        class UnavailableFTPServer(object):
600
 
 
601
575
            def setUp(self):
602
 
                pass
603
 
 
604
 
            def tearDown(self):
605
 
                pass
606
 
 
607
 
            def get_url(self):
608
 
                raise tests.UnavailableFeature(tests.FTPServerFeature)
609
 
 
610
 
            def get_bogus_url(self):
611
576
                raise tests.UnavailableFeature(tests.FTPServerFeature)
612
577
 
613
578
        return [(FtpTransport, UnavailableFTPServer)]