blob: 7d46d830983ec7dfe06281578b348fb0f3282846 [file] [log] [blame]
Guido van Rossum4b8c6ea2000-02-04 15:39:30 +00001"""An FTP client class and some helper functions.
2
Barry Warsaw100d81e2000-09-01 06:09:23 +00003Based on RFC 959: File Transfer Protocol (FTP), by J. Postel and J. Reynolds
Guido van Rossum1115ab21992-11-04 15:51:30 +00004
Guido van Rossumd2560b01996-05-28 23:41:25 +00005Example:
6
7>>> from ftplib import FTP
8>>> ftp = FTP('ftp.python.org') # connect to host, default port
Guido van Rossum24a64342001-12-28 20:54:55 +00009>>> ftp.login() # default, i.e.: user anonymous, passwd anonymous@
Guido van Rossum2f3941d1997-10-07 14:49:56 +000010'230 Guest login ok, access restrictions apply.'
Guido van Rossumd2560b01996-05-28 23:41:25 +000011>>> ftp.retrlines('LIST') # list directory contents
12total 9
13drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
14drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
15drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
16drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
17d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
18drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
19drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
20drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
21-rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
Guido van Rossum2f3941d1997-10-07 14:49:56 +000022'226 Transfer complete.'
Guido van Rossumd2560b01996-05-28 23:41:25 +000023>>> ftp.quit()
Guido van Rossum2f3941d1997-10-07 14:49:56 +000024'221 Goodbye.'
Tim Peters88869f92001-01-14 23:36:06 +000025>>>
Guido van Rossumd2560b01996-05-28 23:41:25 +000026
27A nice test that reveals some of the network dialogue would be:
28python ftplib.py -d localhost -l -p -l
Guido van Rossum4b8c6ea2000-02-04 15:39:30 +000029"""
Guido van Rossumc567c601992-11-05 22:22:37 +000030
Tim Peters88869f92001-01-14 23:36:06 +000031#
Guido van Rossum98d9fd32000-02-28 15:12:25 +000032# Changes and improvements suggested by Steve Majewski.
33# Modified by Jack to work on the mac.
34# Modified by Siebren to support docstrings and PASV.
Gregory P. Smithc64386b2008-01-22 00:19:41 +000035# Modified by Phil Schwartz to add storbinary and storlines callbacks.
Antoine Pitrouccd5e022009-11-15 17:22:09 +000036# Modified by Giampaolo Rodola' to add TLS support.
Guido van Rossum98d9fd32000-02-28 15:12:25 +000037#
Guido van Rossumc567c601992-11-05 22:22:37 +000038
Guido van Rossum1115ab21992-11-04 15:51:30 +000039import os
40import sys
Guido van Rossum1115ab21992-11-04 15:51:30 +000041
Guido van Rossumb6775db1994-08-01 11:34:53 +000042# Import SOCKS module if it exists, else standard socket module socket
43try:
Tim Peters88869f92001-01-14 23:36:06 +000044 import SOCKS; socket = SOCKS; del SOCKS # import SOCKS as socket
45 from socket import getfqdn; socket.getfqdn = getfqdn; del getfqdn
Guido van Rossumb6775db1994-08-01 11:34:53 +000046except ImportError:
Tim Peters88869f92001-01-14 23:36:06 +000047 import socket
Facundo Batista4f1b1ed2008-05-29 16:39:26 +000048from socket import _GLOBAL_DEFAULT_TIMEOUT
Guido van Rossumb6775db1994-08-01 11:34:53 +000049
Skip Montanaroeccd02a2001-01-20 23:34:12 +000050__all__ = ["FTP","Netrc"]
Guido van Rossum1115ab21992-11-04 15:51:30 +000051
Guido van Rossumd3166071993-05-24 14:16:22 +000052# Magic number from <socket.h>
Tim Peters88869f92001-01-14 23:36:06 +000053MSG_OOB = 0x1 # Process data out of band
Guido van Rossumd3166071993-05-24 14:16:22 +000054
55
Guido van Rossumc567c601992-11-05 22:22:37 +000056# The standard FTP server control port
Guido van Rossum1115ab21992-11-04 15:51:30 +000057FTP_PORT = 21
Guido van Rossum1115ab21992-11-04 15:51:30 +000058
59
Guido van Rossum21974791992-11-06 13:34:17 +000060# Exception raised when an error or invalid response is received
Fred Drake227b1202000-08-17 05:06:49 +000061class Error(Exception): pass
Tim Peters88869f92001-01-14 23:36:06 +000062class error_reply(Error): pass # unexpected [123]xx reply
63class error_temp(Error): pass # 4xx errors
64class error_perm(Error): pass # 5xx errors
65class error_proto(Error): pass # response does not begin with [1-5]
Guido van Rossum1115ab21992-11-04 15:51:30 +000066
67
Guido van Rossum21974791992-11-06 13:34:17 +000068# All exceptions (hopefully) that may be raised here and that aren't
69# (always) programming errors on our side
Gregory P. Smithe6c03032008-04-12 22:24:04 +000070all_errors = (Error, IOError, EOFError)
Guido van Rossum21974791992-11-06 13:34:17 +000071
72
Guido van Rossum1115ab21992-11-04 15:51:30 +000073# Line terminators (we always output CRLF, but accept any of CRLF, CR, LF)
74CRLF = '\r\n'
75
Guido van Rossum1115ab21992-11-04 15:51:30 +000076# The class itself
77class FTP:
78
Tim Peters88869f92001-01-14 23:36:06 +000079 '''An FTP client class.
Guido van Rossumd2560b01996-05-28 23:41:25 +000080
Facundo Batista3f100992007-03-26 20:56:09 +000081 To create a connection, call the class using these arguments:
82 host, user, passwd, acct, timeout
83
84 The first four arguments are all strings, and have default value ''.
85 timeout must be numeric and defaults to None if not passed,
86 meaning that no timeout will be set on any ftp socket(s)
87 If a timeout is passed, then this is now the default timeout for all ftp
88 socket operations for this instance.
Neal Norwitz0d4c06e2007-04-25 06:30:05 +000089
Tim Peters88869f92001-01-14 23:36:06 +000090 Then use self.connect() with optional host and port argument.
Guido van Rossumd2560b01996-05-28 23:41:25 +000091
Tim Peters88869f92001-01-14 23:36:06 +000092 To download a file, use ftp.retrlines('RETR ' + filename),
93 or ftp.retrbinary() with slightly different arguments.
94 To upload a file, use ftp.storlines() or ftp.storbinary(),
95 which have an open file as argument (see their definitions
96 below for details).
97 The download/upload functions first issue appropriate TYPE
98 and PORT or PASV commands.
Guido van Rossumd2560b01996-05-28 23:41:25 +000099'''
100
Fred Drake9c98a422001-02-28 21:46:37 +0000101 debugging = 0
102 host = ''
103 port = FTP_PORT
104 sock = None
105 file = None
106 welcome = None
107 passiveserver = 1
108
Tim Peters88869f92001-01-14 23:36:06 +0000109 # Initialization method (called by class instantiation).
110 # Initialize host to localhost, port to standard ftp port
111 # Optional arguments are host (for connect()),
112 # and user, passwd, acct (for login())
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000113 def __init__(self, host='', user='', passwd='', acct='',
114 timeout=_GLOBAL_DEFAULT_TIMEOUT):
Facundo Batista3f100992007-03-26 20:56:09 +0000115 self.timeout = timeout
Tim Peters88869f92001-01-14 23:36:06 +0000116 if host:
Fred Drake9c98a422001-02-28 21:46:37 +0000117 self.connect(host)
Neal Norwitz0d4c06e2007-04-25 06:30:05 +0000118 if user:
Facundo Batista3f100992007-03-26 20:56:09 +0000119 self.login(user, passwd, acct)
Guido van Rossum52fc1f61993-06-17 12:38:10 +0000120
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000121 def connect(self, host='', port=0, timeout=-999):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000122 '''Connect to host. Arguments are:
Facundo Batista3f100992007-03-26 20:56:09 +0000123 - host: hostname to connect to (string, default previous host)
124 - port: port to connect to (integer, default previous port)
125 '''
126 if host != '':
127 self.host = host
128 if port > 0:
129 self.port = port
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000130 if timeout != -999:
Facundo Batista93c33682007-03-30 13:00:35 +0000131 self.timeout = timeout
Facundo Batista3f100992007-03-26 20:56:09 +0000132 self.sock = socket.create_connection((self.host, self.port), self.timeout)
133 self.af = self.sock.family
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000134 self.file = self.sock.makefile('rb')
135 self.welcome = self.getresp()
136 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000137
Tim Peters88869f92001-01-14 23:36:06 +0000138 def getwelcome(self):
139 '''Get the welcome message from the server.
140 (this is read and squirreled away by connect())'''
141 if self.debugging:
142 print '*welcome*', self.sanitize(self.welcome)
143 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000144
Tim Peters88869f92001-01-14 23:36:06 +0000145 def set_debuglevel(self, level):
146 '''Set the debugging level.
147 The required argument level means:
148 0: no debugging output (default)
149 1: print commands and responses but not body text etc.
150 2: also print raw lines read and sent before stripping CR/LF'''
151 self.debugging = level
152 debug = set_debuglevel
Guido van Rossum1115ab21992-11-04 15:51:30 +0000153
Tim Peters88869f92001-01-14 23:36:06 +0000154 def set_pasv(self, val):
155 '''Use passive or active mode for data transfers.
156 With a false argument, use the normal PORT mode,
157 With a true argument, use the PASV command.'''
158 self.passiveserver = val
Guido van Rossumd2560b01996-05-28 23:41:25 +0000159
Tim Peters88869f92001-01-14 23:36:06 +0000160 # Internal: "sanitize" a string for printing
161 def sanitize(self, s):
162 if s[:5] == 'pass ' or s[:5] == 'PASS ':
163 i = len(s)
164 while i > 5 and s[i-1] in '\r\n':
165 i = i-1
166 s = s[:5] + '*'*(i-5) + s[i:]
Walter Dörwald70a6b492004-02-12 17:35:32 +0000167 return repr(s)
Guido van Rossumebaf1041995-05-05 15:54:14 +0000168
Tim Peters88869f92001-01-14 23:36:06 +0000169 # Internal: send one line to the server, appending CRLF
170 def putline(self, line):
171 line = line + CRLF
172 if self.debugging > 1: print '*put*', self.sanitize(line)
Martin v. Löwise12454f2002-02-16 23:06:19 +0000173 self.sock.sendall(line)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000174
Tim Peters88869f92001-01-14 23:36:06 +0000175 # Internal: send one command to the server (through putline())
176 def putcmd(self, line):
177 if self.debugging: print '*cmd*', self.sanitize(line)
178 self.putline(line)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000179
Tim Peters88869f92001-01-14 23:36:06 +0000180 # Internal: return one line from the server, stripping CRLF.
181 # Raise EOFError if the connection is closed
182 def getline(self):
183 line = self.file.readline()
184 if self.debugging > 1:
185 print '*get*', self.sanitize(line)
186 if not line: raise EOFError
187 if line[-2:] == CRLF: line = line[:-2]
188 elif line[-1:] in CRLF: line = line[:-1]
189 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000190
Tim Peters88869f92001-01-14 23:36:06 +0000191 # Internal: get a response from the server, which may possibly
192 # consist of multiple lines. Return a single string with no
193 # trailing CRLF. If the response consists of multiple lines,
194 # these are separated by '\n' characters in the string
195 def getmultiline(self):
196 line = self.getline()
197 if line[3:4] == '-':
198 code = line[:3]
199 while 1:
200 nextline = self.getline()
201 line = line + ('\n' + nextline)
202 if nextline[:3] == code and \
203 nextline[3:4] != '-':
204 break
205 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000206
Tim Peters88869f92001-01-14 23:36:06 +0000207 # Internal: get a response from the server.
208 # Raise various errors if the response indicates an error
209 def getresp(self):
210 resp = self.getmultiline()
211 if self.debugging: print '*resp*', self.sanitize(resp)
212 self.lastresp = resp[:3]
213 c = resp[:1]
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000214 if c in ('1', '2', '3'):
215 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000216 if c == '4':
217 raise error_temp, resp
218 if c == '5':
219 raise error_perm, resp
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000220 raise error_proto, resp
Guido van Rossum1115ab21992-11-04 15:51:30 +0000221
Tim Peters88869f92001-01-14 23:36:06 +0000222 def voidresp(self):
223 """Expect a response beginning with '2'."""
224 resp = self.getresp()
Georg Brandl50ba6e12009-04-05 10:48:47 +0000225 if resp[:1] != '2':
Tim Peters88869f92001-01-14 23:36:06 +0000226 raise error_reply, resp
227 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000228
Tim Peters88869f92001-01-14 23:36:06 +0000229 def abort(self):
230 '''Abort a file transfer. Uses out-of-band data.
231 This does not follow the procedure from the RFC to send Telnet
232 IP and Synch; that doesn't seem to work with the servers I've
233 tried. Instead, just send the ABOR command as OOB data.'''
234 line = 'ABOR' + CRLF
235 if self.debugging > 1: print '*put urgent*', self.sanitize(line)
Martin v. Löwise12454f2002-02-16 23:06:19 +0000236 self.sock.sendall(line, MSG_OOB)
Tim Peters88869f92001-01-14 23:36:06 +0000237 resp = self.getmultiline()
238 if resp[:3] not in ('426', '226'):
239 raise error_proto, resp
Guido van Rossumd3166071993-05-24 14:16:22 +0000240
Tim Peters88869f92001-01-14 23:36:06 +0000241 def sendcmd(self, cmd):
242 '''Send a command and return the response.'''
243 self.putcmd(cmd)
244 return self.getresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000245
Tim Peters88869f92001-01-14 23:36:06 +0000246 def voidcmd(self, cmd):
247 """Send a command and expect a response beginning with '2'."""
248 self.putcmd(cmd)
249 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000250
Tim Peters88869f92001-01-14 23:36:06 +0000251 def sendport(self, host, port):
252 '''Send a PORT command with the current host and the given
253 port number.
254 '''
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000255 hbytes = host.split('.')
Benjamin Petersondee0b172008-09-27 22:08:12 +0000256 pbytes = [repr(port//256), repr(port%256)]
Tim Peters88869f92001-01-14 23:36:06 +0000257 bytes = hbytes + pbytes
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000258 cmd = 'PORT ' + ','.join(bytes)
Tim Peters88869f92001-01-14 23:36:06 +0000259 return self.voidcmd(cmd)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000260
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000261 def sendeprt(self, host, port):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000262 '''Send a EPRT command with the current host and the given port number.'''
263 af = 0
264 if self.af == socket.AF_INET:
265 af = 1
266 if self.af == socket.AF_INET6:
267 af = 2
268 if af == 0:
269 raise error_proto, 'unsupported address family'
Walter Dörwald70a6b492004-02-12 17:35:32 +0000270 fields = ['', repr(af), host, repr(port), '']
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000271 cmd = 'EPRT ' + '|'.join(fields)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000272 return self.voidcmd(cmd)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000273
Tim Peters88869f92001-01-14 23:36:06 +0000274 def makeport(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000275 '''Create a new socket and send a PORT command for it.'''
Martin v. Löwis2ad25692001-07-31 08:40:21 +0000276 msg = "getaddrinfo returns an empty list"
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000277 sock = None
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000278 for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
279 af, socktype, proto, canonname, sa = res
280 try:
281 sock = socket.socket(af, socktype, proto)
282 sock.bind(sa)
283 except socket.error, msg:
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000284 if sock:
285 sock.close()
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000286 sock = None
287 continue
288 break
289 if not sock:
290 raise socket.error, msg
291 sock.listen(1)
292 port = sock.getsockname()[1] # Get proper port
293 host = self.sock.getsockname()[0] # Get proper host
294 if self.af == socket.AF_INET:
295 resp = self.sendport(host, port)
296 else:
297 resp = self.sendeprt(host, port)
298 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000299
300 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000301 if self.af == socket.AF_INET:
302 host, port = parse227(self.sendcmd('PASV'))
303 else:
304 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
305 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000306
Tim Peters88869f92001-01-14 23:36:06 +0000307 def ntransfercmd(self, cmd, rest=None):
308 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000309
Tim Peters88869f92001-01-14 23:36:06 +0000310 If the transfer is active, send a port command and the
311 transfer command, and accept the connection. If the server is
312 passive, send a pasv command, connect to it, and start the
313 transfer command. Either way, return the socket for the
314 connection and the expected size of the transfer. The
315 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000316
Tim Peters88869f92001-01-14 23:36:06 +0000317 Optional `rest' argument can be a string that is sent as the
Gregory P. Smith2230bcf2008-01-22 23:15:34 +0000318 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000319 marker used to tell the server to skip over any data up to the
320 given marker.
321 """
322 size = None
323 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000324 host, port = self.makepasv()
Facundo Batista92493122007-06-06 15:13:37 +0000325 conn = socket.create_connection((host, port), self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000326 if rest is not None:
327 self.sendcmd("REST %s" % rest)
328 resp = self.sendcmd(cmd)
Martin v. Löwis36cbc082006-11-12 18:48:13 +0000329 # Some servers apparently send a 200 reply to
330 # a LIST or STOR command, before the 150 reply
331 # (and way before the 226 reply). This seems to
332 # be in violation of the protocol (which only allows
333 # 1xx or error messages for LIST), so we just discard
334 # this response.
335 if resp[0] == '2':
Tim Petersf733abb2007-01-30 03:03:46 +0000336 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000337 if resp[0] != '1':
338 raise error_reply, resp
339 else:
340 sock = self.makeport()
341 if rest is not None:
342 self.sendcmd("REST %s" % rest)
343 resp = self.sendcmd(cmd)
Martin v. Löwis36cbc082006-11-12 18:48:13 +0000344 # See above.
345 if resp[0] == '2':
Tim Petersf733abb2007-01-30 03:03:46 +0000346 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000347 if resp[0] != '1':
348 raise error_reply, resp
349 conn, sockaddr = sock.accept()
350 if resp[:3] == '150':
351 # this is conditional in case we received a 125
352 size = parse150(resp)
353 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000354
Tim Peters88869f92001-01-14 23:36:06 +0000355 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000356 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000357 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000358
Tim Peters88869f92001-01-14 23:36:06 +0000359 def login(self, user = '', passwd = '', acct = ''):
360 '''Login, default anonymous.'''
361 if not user: user = 'anonymous'
362 if not passwd: passwd = ''
363 if not acct: acct = ''
364 if user == 'anonymous' and passwd in ('', '-'):
Tim Peterse4418602002-02-16 07:34:19 +0000365 # If there is no anonymous ftp password specified
366 # then we'll just use anonymous@
367 # We don't send any other thing because:
368 # - We want to remain anonymous
369 # - We want to stop SPAM
370 # - We don't want to let ftp sites to discriminate by the user,
371 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000372 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000373 resp = self.sendcmd('USER ' + user)
374 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
375 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
376 if resp[0] != '2':
377 raise error_reply, resp
378 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000379
Tim Peters88869f92001-01-14 23:36:06 +0000380 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000381 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000382
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000383 Args:
384 cmd: A RETR command.
385 callback: A single parameter callable to be called on each
386 block of data read.
387 blocksize: The maximum number of bytes to read from the
388 socket at one time. [default: 8192]
389 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000390
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000391 Returns:
392 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000393 """
394 self.voidcmd('TYPE I')
395 conn = self.transfercmd(cmd, rest)
396 while 1:
397 data = conn.recv(blocksize)
398 if not data:
399 break
400 callback(data)
401 conn.close()
402 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000403
Tim Peters88869f92001-01-14 23:36:06 +0000404 def retrlines(self, cmd, callback = None):
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000405 """Retrieve data in line mode. A new port is created for you.
406
407 Args:
Gregory P. Smith2230bcf2008-01-22 23:15:34 +0000408 cmd: A RETR, LIST, NLST, or MLSD command.
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000409 callback: An optional single parameter callable that is called
410 for each line with the trailing CRLF stripped.
411 [default: print_line()]
412
413 Returns:
414 The response code.
415 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000416 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000417 resp = self.sendcmd('TYPE A')
418 conn = self.transfercmd(cmd)
419 fp = conn.makefile('rb')
420 while 1:
421 line = fp.readline()
Walter Dörwald70a6b492004-02-12 17:35:32 +0000422 if self.debugging > 2: print '*retr*', repr(line)
Tim Peters88869f92001-01-14 23:36:06 +0000423 if not line:
424 break
425 if line[-2:] == CRLF:
426 line = line[:-2]
427 elif line[-1:] == '\n':
428 line = line[:-1]
429 callback(line)
430 fp.close()
431 conn.close()
432 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000433
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000434 def storbinary(self, cmd, fp, blocksize=8192, callback=None):
435 """Store a file in binary mode. A new port is created for you.
436
437 Args:
438 cmd: A STOR command.
439 fp: A file-like object with a read(num_bytes) method.
440 blocksize: The maximum data size to read from fp and send over
441 the connection at once. [default: 8192]
442 callback: An optional single parameter callable that is called on
443 on each block of data after it is sent. [default: None]
444
445 Returns:
446 The response code.
447 """
Tim Peters88869f92001-01-14 23:36:06 +0000448 self.voidcmd('TYPE I')
449 conn = self.transfercmd(cmd)
450 while 1:
451 buf = fp.read(blocksize)
452 if not buf: break
Martin v. Löwise12454f2002-02-16 23:06:19 +0000453 conn.sendall(buf)
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000454 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000455 conn.close()
456 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000457
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000458 def storlines(self, cmd, fp, callback=None):
459 """Store a file in line mode. A new port is created for you.
460
461 Args:
462 cmd: A STOR command.
463 fp: A file-like object with a readline() method.
464 callback: An optional single parameter callable that is called on
465 on each line after it is sent. [default: None]
466
467 Returns:
468 The response code.
469 """
Tim Peters88869f92001-01-14 23:36:06 +0000470 self.voidcmd('TYPE A')
471 conn = self.transfercmd(cmd)
472 while 1:
473 buf = fp.readline()
474 if not buf: break
475 if buf[-2:] != CRLF:
476 if buf[-1] in CRLF: buf = buf[:-1]
477 buf = buf + CRLF
Martin v. Löwise12454f2002-02-16 23:06:19 +0000478 conn.sendall(buf)
Gregory P. Smithc64386b2008-01-22 00:19:41 +0000479 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000480 conn.close()
481 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000482
Tim Peters88869f92001-01-14 23:36:06 +0000483 def acct(self, password):
484 '''Send new account name.'''
485 cmd = 'ACCT ' + password
486 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000487
Tim Peters88869f92001-01-14 23:36:06 +0000488 def nlst(self, *args):
489 '''Return a list of files in a given directory (default the current).'''
490 cmd = 'NLST'
491 for arg in args:
492 cmd = cmd + (' ' + arg)
493 files = []
494 self.retrlines(cmd, files.append)
495 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000496
Tim Peters88869f92001-01-14 23:36:06 +0000497 def dir(self, *args):
498 '''List a directory in long form.
499 By default list current directory to stdout.
500 Optional last argument is callback function; all
501 non-empty arguments before it are concatenated to the
502 LIST command. (This *should* only be used for a pathname.)'''
503 cmd = 'LIST'
504 func = None
505 if args[-1:] and type(args[-1]) != type(''):
506 args, func = args[:-1], args[-1]
507 for arg in args:
508 if arg:
509 cmd = cmd + (' ' + arg)
510 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000511
Tim Peters88869f92001-01-14 23:36:06 +0000512 def rename(self, fromname, toname):
513 '''Rename a file.'''
514 resp = self.sendcmd('RNFR ' + fromname)
515 if resp[0] != '3':
516 raise error_reply, resp
517 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000518
Tim Peters88869f92001-01-14 23:36:06 +0000519 def delete(self, filename):
520 '''Delete a file.'''
521 resp = self.sendcmd('DELE ' + filename)
522 if resp[:3] in ('250', '200'):
523 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000524 else:
525 raise error_reply, resp
Guido van Rossum02cf5821993-05-17 08:00:02 +0000526
Tim Peters88869f92001-01-14 23:36:06 +0000527 def cwd(self, dirname):
528 '''Change to a directory.'''
529 if dirname == '..':
530 try:
531 return self.voidcmd('CDUP')
532 except error_perm, msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000533 if msg.args[0][:3] != '500':
534 raise
Tim Peters88869f92001-01-14 23:36:06 +0000535 elif dirname == '':
536 dirname = '.' # does nothing, but could return error
537 cmd = 'CWD ' + dirname
538 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000539
Tim Peters88869f92001-01-14 23:36:06 +0000540 def size(self, filename):
541 '''Retrieve the size of a file.'''
Gregory P. Smith2230bcf2008-01-22 23:15:34 +0000542 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000543 resp = self.sendcmd('SIZE ' + filename)
544 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000545 s = resp[3:].strip()
546 try:
547 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000548 except (OverflowError, ValueError):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000549 return long(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000550
Tim Peters88869f92001-01-14 23:36:06 +0000551 def mkd(self, dirname):
552 '''Make a directory, return its full pathname.'''
553 resp = self.sendcmd('MKD ' + dirname)
554 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000555
Tim Peters88869f92001-01-14 23:36:06 +0000556 def rmd(self, dirname):
557 '''Remove a directory.'''
558 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000559
Tim Peters88869f92001-01-14 23:36:06 +0000560 def pwd(self):
561 '''Return current working directory.'''
562 resp = self.sendcmd('PWD')
563 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000564
Tim Peters88869f92001-01-14 23:36:06 +0000565 def quit(self):
566 '''Quit, and close the connection.'''
567 resp = self.voidcmd('QUIT')
568 self.close()
569 return resp
570
571 def close(self):
572 '''Close the connection without assuming anything about it.'''
Fred Drake9c98a422001-02-28 21:46:37 +0000573 if self.file:
574 self.file.close()
575 self.sock.close()
576 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000577
578
Antoine Pitrouccd5e022009-11-15 17:22:09 +0000579try:
580 import ssl
581except ImportError:
582 pass
583else:
584 class FTP_TLS(FTP):
585 '''A FTP subclass which adds TLS support to FTP as described
586 in RFC-4217.
587
588 Connect as usual to port 21 implicitly securing the FTP control
589 connection before authenticating.
590
591 Securing the data connection requires user to explicitly ask
592 for it by calling prot_p() method.
593
594 Usage example:
595 >>> from ftplib import FTP_TLS
596 >>> ftps = FTP_TLS('ftp.python.org')
597 >>> ftps.login() # login anonimously previously securing control channel
598 '230 Guest login ok, access restrictions apply.'
599 >>> ftps.prot_p() # switch to secure data connection
600 '200 Protection level set to P'
601 >>> ftps.retrlines('LIST') # list directory content securely
602 total 9
603 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
604 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
605 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
606 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
607 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
608 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
609 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
610 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
611 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
612 '226 Transfer complete.'
613 >>> ftps.quit()
614 '221 Goodbye.'
615 >>>
616 '''
617 ssl_version = ssl.PROTOCOL_TLSv1
618
619 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
620 certfile=None, timeout=_GLOBAL_DEFAULT_TIMEOUT):
621 self.keyfile = keyfile
622 self.certfile = certfile
623 self._prot_p = False
624 FTP.__init__(self, host, user, passwd, acct, timeout)
625
626 def login(self, user='', passwd='', acct='', secure=True):
627 if secure and not isinstance(self.sock, ssl.SSLSocket):
628 self.auth()
629 return FTP.login(self, user, passwd, acct)
630
631 def auth(self):
632 '''Set up secure control connection by using TLS/SSL.'''
633 if isinstance(self.sock, ssl.SSLSocket):
634 raise ValueError("Already using TLS")
635 if self.ssl_version == ssl.PROTOCOL_TLSv1:
636 resp = self.voidcmd('AUTH TLS')
637 else:
638 resp = self.voidcmd('AUTH SSL')
639 self.sock = ssl.wrap_socket(self.sock, self.keyfile, self.certfile,
640 ssl_version=self.ssl_version)
641 self.file = self.sock.makefile(mode='rb')
642 return resp
643
644 def prot_p(self):
645 '''Set up secure data connection.'''
646 # PROT defines whether or not the data channel is to be protected.
647 # Though RFC-2228 defines four possible protection levels,
648 # RFC-4217 only recommends two, Clear and Private.
649 # Clear (PROT C) means that no security is to be used on the
650 # data-channel, Private (PROT P) means that the data-channel
651 # should be protected by TLS.
652 # PBSZ command MUST still be issued, but must have a parameter of
653 # '0' to indicate that no buffering is taking place and the data
654 # connection should not be encapsulated.
655 self.voidcmd('PBSZ 0')
656 resp = self.voidcmd('PROT P')
657 self._prot_p = True
658 return resp
659
660 def prot_c(self):
661 '''Set up clear text data connection.'''
662 resp = self.voidcmd('PROT C')
663 self._prot_p = False
664 return resp
665
666 # --- Overridden FTP methods
667
668 def ntransfercmd(self, cmd, rest=None):
669 conn, size = FTP.ntransfercmd(self, cmd, rest)
670 if self._prot_p:
671 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
672 ssl_version=self.ssl_version)
673 return conn, size
674
675 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
676 self.voidcmd('TYPE I')
677 conn = self.transfercmd(cmd, rest)
678 try:
679 while 1:
680 data = conn.recv(blocksize)
681 if not data:
682 break
683 callback(data)
684 # shutdown ssl layer
685 if isinstance(conn, ssl.SSLSocket):
686 conn.unwrap()
687 finally:
688 conn.close()
689 return self.voidresp()
690
691 def retrlines(self, cmd, callback = None):
692 if callback is None: callback = print_line
693 resp = self.sendcmd('TYPE A')
694 conn = self.transfercmd(cmd)
695 fp = conn.makefile('rb')
696 try:
697 while 1:
698 line = fp.readline()
699 if self.debugging > 2: print '*retr*', repr(line)
700 if not line:
701 break
702 if line[-2:] == CRLF:
703 line = line[:-2]
704 elif line[-1:] == '\n':
705 line = line[:-1]
706 callback(line)
707 # shutdown ssl layer
708 if isinstance(conn, ssl.SSLSocket):
709 conn.unwrap()
710 finally:
711 fp.close()
712 conn.close()
713 return self.voidresp()
714
715 def storbinary(self, cmd, fp, blocksize=8192, callback=None):
716 self.voidcmd('TYPE I')
717 conn = self.transfercmd(cmd)
718 try:
719 while 1:
720 buf = fp.read(blocksize)
721 if not buf: break
722 conn.sendall(buf)
723 if callback: callback(buf)
724 # shutdown ssl layer
725 if isinstance(conn, ssl.SSLSocket):
726 conn.unwrap()
727 finally:
728 conn.close()
729 return self.voidresp()
730
731 def storlines(self, cmd, fp, callback=None):
732 self.voidcmd('TYPE A')
733 conn = self.transfercmd(cmd)
734 try:
735 while 1:
736 buf = fp.readline()
737 if not buf: break
738 if buf[-2:] != CRLF:
739 if buf[-1] in CRLF: buf = buf[:-1]
740 buf = buf + CRLF
741 conn.sendall(buf)
742 if callback: callback(buf)
743 # shutdown ssl layer
744 if isinstance(conn, ssl.SSLSocket):
745 conn.unwrap()
746 finally:
747 conn.close()
748 return self.voidresp()
749
750 __all__.append('FTP_TLS')
751 all_errors = (Error, IOError, EOFError, ssl.SSLError)
752
753
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000754_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000755
756def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000757 '''Parse the '150' response for a RETR request.
758 Returns the expected transfer size or None; size is not guaranteed to
759 be present in the 150 message.
760 '''
761 if resp[:3] != '150':
762 raise error_reply, resp
763 global _150_re
764 if _150_re is None:
765 import re
766 _150_re = re.compile("150 .* \((\d+) bytes\)", re.IGNORECASE)
767 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000768 if not m:
769 return None
770 s = m.group(1)
771 try:
772 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000773 except (OverflowError, ValueError):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000774 return long(s)
Fred Drake4de02d91997-01-10 18:26:09 +0000775
776
Guido van Rossum70297d32001-08-17 17:24:29 +0000777_227_re = None
778
Guido van Rossumd2560b01996-05-28 23:41:25 +0000779def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000780 '''Parse the '227' response for a PASV request.
781 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
782 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000783
Tim Peters88869f92001-01-14 23:36:06 +0000784 if resp[:3] != '227':
785 raise error_reply, resp
Guido van Rossum70297d32001-08-17 17:24:29 +0000786 global _227_re
787 if _227_re is None:
788 import re
789 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)')
790 m = _227_re.search(resp)
791 if not m:
Tim Peters88869f92001-01-14 23:36:06 +0000792 raise error_proto, resp
Guido van Rossum70297d32001-08-17 17:24:29 +0000793 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000794 host = '.'.join(numbers[:4])
795 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000796 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000797
798
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000799def parse229(resp, peer):
800 '''Parse the '229' response for a EPSV request.
801 Raises error_proto if it does not contain '(|||port|)'
802 Return ('host.addr.as.numbers', port#) tuple.'''
803
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000804 if resp[:3] != '229':
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000805 raise error_reply, resp
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000806 left = resp.find('(')
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000807 if left < 0: raise error_proto, resp
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000808 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000809 if right < 0:
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000810 raise error_proto, resp # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000811 if resp[left + 1] != resp[right - 1]:
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000812 raise error_proto, resp
Walter Dörwalda401ae42002-06-03 10:41:45 +0000813 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000814 if len(parts) != 5:
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000815 raise error_proto, resp
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000816 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000817 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000818 return host, port
819
820
Guido van Rossumc567c601992-11-05 22:22:37 +0000821def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000822 '''Parse the '257' response for a MKD or PWD request.
823 This is a response to a MKD or PWD request: a directory name.
824 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000825
Tim Peters88869f92001-01-14 23:36:06 +0000826 if resp[:3] != '257':
827 raise error_reply, resp
828 if resp[3:5] != ' "':
829 return '' # Not compliant to RFC 959, but UNIX ftpd does this
830 dirname = ''
831 i = 5
832 n = len(resp)
833 while i < n:
834 c = resp[i]
835 i = i+1
836 if c == '"':
837 if i >= n or resp[i] != '"':
838 break
839 i = i+1
840 dirname = dirname + c
841 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000842
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000843
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000844def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000845 '''Default retrlines callback to print a line.'''
846 print line
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000847
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000848
Guido van Rossumd2560b01996-05-28 23:41:25 +0000849def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000850 '''Copy file from one FTP-instance to another.'''
851 if not targetname: targetname = sourcename
852 type = 'TYPE ' + type
853 source.voidcmd(type)
854 target.voidcmd(type)
855 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
856 target.sendport(sourcehost, sourceport)
857 # RFC 959: the user must "listen" [...] BEFORE sending the
858 # transfer request.
859 # So: STOR before RETR, because here the target is a "user".
860 treply = target.sendcmd('STOR ' + targetname)
861 if treply[:3] not in ('125', '150'): raise error_proto # RFC 959
862 sreply = source.sendcmd('RETR ' + sourcename)
863 if sreply[:3] not in ('125', '150'): raise error_proto # RFC 959
864 source.voidresp()
865 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000866
Tim Peters88869f92001-01-14 23:36:06 +0000867
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000868class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000869 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000870
Tim Peters88869f92001-01-14 23:36:06 +0000871 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000872
Tim Peters88869f92001-01-14 23:36:06 +0000873 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000874
Tim Peters88869f92001-01-14 23:36:06 +0000875 """
876 __defuser = None
877 __defpasswd = None
878 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000879
Tim Peters88869f92001-01-14 23:36:06 +0000880 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000881 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000882 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000883 filename = os.path.join(os.environ["HOME"],
884 ".netrc")
885 else:
886 raise IOError, \
887 "specify file to load or set $HOME"
888 self.__hosts = {}
889 self.__macros = {}
890 fp = open(filename, "r")
891 in_macro = 0
892 while 1:
893 line = fp.readline()
894 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000895 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000896 macro_lines.append(line)
897 continue
898 elif in_macro:
899 self.__macros[macro_name] = tuple(macro_lines)
900 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000901 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000902 host = user = passwd = acct = None
903 default = 0
904 i = 0
905 while i < len(words):
906 w1 = words[i]
907 if i+1 < len(words):
908 w2 = words[i + 1]
909 else:
910 w2 = None
911 if w1 == 'default':
912 default = 1
913 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000914 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000915 i = i + 1
916 elif w1 == 'login' and w2:
917 user = w2
918 i = i + 1
919 elif w1 == 'password' and w2:
920 passwd = w2
921 i = i + 1
922 elif w1 == 'account' and w2:
923 acct = w2
924 i = i + 1
925 elif w1 == 'macdef' and w2:
926 macro_name = w2
927 macro_lines = []
928 in_macro = 1
929 break
930 i = i + 1
931 if default:
932 self.__defuser = user or self.__defuser
933 self.__defpasswd = passwd or self.__defpasswd
934 self.__defacct = acct or self.__defacct
935 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000936 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +0000937 ouser, opasswd, oacct = \
938 self.__hosts[host]
939 user = user or ouser
940 passwd = passwd or opasswd
941 acct = acct or oacct
942 self.__hosts[host] = user, passwd, acct
943 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000944
Tim Peters88869f92001-01-14 23:36:06 +0000945 def get_hosts(self):
946 """Return a list of hosts mentioned in the .netrc file."""
947 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +0000948
Tim Peters88869f92001-01-14 23:36:06 +0000949 def get_account(self, host):
950 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +0000951
Tim Peters88869f92001-01-14 23:36:06 +0000952 The return value is a triple containing userid,
953 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +0000954
Tim Peters88869f92001-01-14 23:36:06 +0000955 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000956 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000957 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +0000958 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +0000959 user, passwd, acct = self.__hosts[host]
960 user = user or self.__defuser
961 passwd = passwd or self.__defpasswd
962 acct = acct or self.__defacct
963 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +0000964
Tim Peters88869f92001-01-14 23:36:06 +0000965 def get_macros(self):
966 """Return a list of all defined macro names."""
967 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +0000968
Tim Peters88869f92001-01-14 23:36:06 +0000969 def get_macro(self, macro):
970 """Return a sequence of lines which define a named macro."""
971 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000972
Fred Drake475d51d1997-06-24 22:02:54 +0000973
Tim Peters88869f92001-01-14 23:36:06 +0000974
Guido van Rossum1115ab21992-11-04 15:51:30 +0000975def test():
Tim Peters88869f92001-01-14 23:36:06 +0000976 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000977 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
978
979 -d dir
980 -l list
981 -p password
982 '''
983
984 if len(sys.argv) < 2:
985 print test.__doc__
986 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +0000987
Tim Peters88869f92001-01-14 23:36:06 +0000988 debugging = 0
989 rcfile = None
990 while sys.argv[1] == '-d':
991 debugging = debugging+1
992 del sys.argv[1]
993 if sys.argv[1][:2] == '-r':
994 # get name of alternate ~/.netrc file:
995 rcfile = sys.argv[1][2:]
996 del sys.argv[1]
997 host = sys.argv[1]
998 ftp = FTP(host)
999 ftp.set_debuglevel(debugging)
1000 userid = passwd = acct = ''
1001 try:
1002 netrc = Netrc(rcfile)
1003 except IOError:
1004 if rcfile is not None:
1005 sys.stderr.write("Could not open account file"
1006 " -- using anonymous login.")
1007 else:
1008 try:
1009 userid, passwd, acct = netrc.get_account(host)
1010 except KeyError:
1011 # no account for host
1012 sys.stderr.write(
1013 "No account -- using anonymous login.")
1014 ftp.login(userid, passwd, acct)
1015 for file in sys.argv[2:]:
1016 if file[:2] == '-l':
1017 ftp.dir(file[2:])
1018 elif file[:2] == '-d':
1019 cmd = 'CWD'
1020 if file[2:]: cmd = cmd + ' ' + file[2:]
1021 resp = ftp.sendcmd(cmd)
1022 elif file == '-p':
1023 ftp.set_pasv(not ftp.passiveserver)
1024 else:
1025 ftp.retrbinary('RETR ' + file, \
1026 sys.stdout.write, 1024)
1027 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001028
1029
1030if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001031 test()