ftpy

# FTPY - a simple FTP program
# Author: Muhammad-Sharif Moustafa

from ftplib import FTP
from ftplib import error_temp
from ftplib import error_perm
from getpass import getpass
from sys import exit
import sys
import socket
import os

# Global Variables
ftp = FTP()
host = ''
user = ''
password = ''
msg = ''
connected = False
loggedIn = False
file = None

# Convenience Functions
def connect(_host):
	if (_host == 'quit' or _host == 'exit'):
		exit(0)
	if (_host != ''):
		try:
			msg = ftp.connect(_host)
		except error_temp as e:
			print(e)
		except error_perm as e:
			print(e)
		except socket.gaierror as e:
			print("Address error: " + e.strerror)
		except socket.timeout:
			print("Connection timed out.")
		except socket.error as e:
			print("Socket error: " + str(e))
		else:
			print(msg)
			return True
	return False

def login(_user, _password):
	try:
		msg = ftp.login(_user, _password)
	except error_temp as e:
		print(e)
	except error_perm as e:
		print(e)
	else:
		print(msg)
		return True
	return False
	
def writeFileLineByLine(line):
	file.write(line)
	file.write('\n')

def writeBinary(data):
	file.write(data)

def getUserInput(msg):
	inputString = ''
	# Try the Python 2 way of getting input. If that fails
	# use the Python 3 way.
	try:
		inputString = eval("raw_input('" + msg + "')")
	except (SyntaxError, NameError):
		inputString = eval("input('" + msg + "')")
	return inputString

# Parse command line arguments (if any) and try to connect
# with credentials in command line
if len(sys.argv) == 2:
	host = sys.argv[1]
	connected = connect(host)

if len(sys.argv) == 3:
	# user provided host and user
	host = sys.argv[1]
	user = sys.argv[2]
	# try to connect
	connected = connect(host)

if len(sys.argv) == 4:
	# user provided host and user
	host = sys.argv[1]
	user = sys.argv[2]
	password = sys.argv[3]
	# try to connect
	connected = connect(host)
	# try to login
	if connected:
		loggedIn = login(user, password)

# Connection and Login Logic
while not connected:
	host = getUserInput('Host name (URL): ').strip()
	connected = connect(host)

while not loggedIn:
	if user == '':
		user = getUserInput('Username (blank for anonymous): ').strip()
	if password == '':
		password = getpass('Password (blank for anonymous): ').strip()
	loggedIn = login(user, password)
	# if login attempt was unsuccessful, clear username and password
	if not loggedIn:
		user = ''
		password = ''

# Command Loop
while True:
	cmd = getUserInput('Command: ').strip()
	# split command by spaces
	cmdList = cmd.split(None, 1)
	# if no commands were given, skip the loop
	if len(cmdList) == 0:
		continue
	# if there were more than one "words" in the command,
	# the words after the first one are command arguments.
	if len(cmdList) > 1:
		cmdArgs = cmdList[1]
	# parse the command
	if cmdList[0] == 'quit' or cmdList[0] == 'exit':
		break;
	elif cmdList[0] == 'ls':
		ftp.retrlines('LIST')
	elif cmdList[0] == 'cd':
		try:
			msg = ftp.cwd(cmdArgs)
		except error_perm as e:
			print(e)
		else:
			print(msg)
	elif cmdList[0] == 'get':
		try:
			file = open(cmdList[1], 'wb')
			ftp.retrbinary(('RETR ' + cmdArgs), writeBinary)
			file.close()
		except IOError as e:
			print('IOError: ' + str(e))
		except error_perm as e:
			print(e)
		else:
			print('Got ' + cmdArgs)
	elif cmdList[0] == 'put':
		filename = os.path.basename(cmdArgs)
		try:
			file = open(cmdArgs, 'rb')
			ftp.storbinary('STOR ' + filename, file)
			file.close()
		except IOError as e:
			print('IOError: ' + str(e))
		except error_perm as e:
			print(e)
		else:
			print('Put ' + filename)
	elif cmdList[0] == 'mkdir':
		try:
			print(ftp.mkd(cmdArgs))
		except error_perm as e:
			print(e)
		else:
			print('Made new directory: ' + cmdArgs)
	elif cmdList[0] == 'rmdir':
		try:
			print(ftp.rmd(cmdArgs))
		except error_perm as e:
			print(e)
		else:
			print('Removed directory: ' + cmdArgs)
	elif cmdList[0] == 'pwd':
		try:
			print(ftp.pwd())
		except error_perm as e:
			print(e)
	elif cmdList[0] == 'mv':
		cmdArgsList = cmdArgs.split()
		try:
			print(ftp.rename(cmdArgsList[0], cmdArgsList[1]))
		except error_perm as e:
			print(e)
		else:
			print('Moved ' + cmdArgsList[0] + ' to ' + cmdArgsList[1])
	elif cmdList[0] == 'rm':
		try:
			print(ftp.delete(cmdArgs))
		except error_perm as e:
			print(e)
		else:
			print('Deleted ' + cmdArgs)

# Try quitting nicely. If that fails, force quit.
try:
	print(ftp.quit())
except:
	print(ftp.close())