#!/usr/bin/env python
# -*- coding: utf-8 -*-

__author__ = 's1d4rt4 <s1d4rt4@gmail.com'
__date__ = '19/09/06'
__version__ = '1.0'
__license__ = 'GPL'

import urllib	# for the download
import os	# to know de actual directory
import sys	# to read the url

class Message:
	'''The messages'''
	
	def zero_url(self):
		print '---------------------------------------------'
		print 'An URL is needed'
		print 'Usage: pyget.py [URL]'
		print '\nFor more information type "pyget.py --help"'
		print '---------------------------------------------'
		
	def help(self):
		print '-----------------------------------------------------'
		print 'This program try to emule the "wget" command'
		print '\nUsage: pyget.py [URL]'
		print 'Available options: '
		print '	--about	        : About the author'
		print '	--help	        : Show this help'
		print'	--version	: Print the program version'
		print '-----------------------------------------------------'
		
	def version(self):
		print '--------------'
		print ' Version 1.0'
		print '--------------'
		
	def about(self):
		print '--------------------------------'
		print 'Pywget 1.0'
		print 'Author: Carlos Rogel Montecinos'
		print 'Web: s1d4rt4.wordpress.com'
		print 'Contact: s1d4rt4@gmail.com'
		print '\nSeptember 19/2006'
		print '--------------------------------'
		
	def bad_option(self):
		print '-------------'
		print ' Bad option'
		print '-------------'
	
	def bad_url(self):
		print '----------------------------------'
		print "ERROR: The URL doesn't exist"
		print '----------------------------------'
		
	def connecting(self):
		try:
			url = sys.argv[1]
			server = url.split('/')[2]
			print '---------------------------------------------------------------------------'
			print 'Connecting to %s' % server
		except:
			pass
			
	def downloading(self):
		try:
			url = sys.argv[1]
			file = url.split('/')[-1]
			print 'Please wait while PyGet download %s' % file
		except:
			pass
		
	def downloaded(self):
		try:
			url = sys.argv[1]
			file = url.split('/')[-1]
			dir = os.getcwd()		
			print '%s has been downloaded in %s' % (file,dir)
			print '---------------------------------------------------------------------------'
		except:
			pass
			
class PyGet(Message):
	'''the heart of PyGet'''
	
	def __init__(self):
		if len(sys.argv) < 2:
			Message.zero_url(self)
			
		elif sys.argv[1].startswith('--'):
			option = sys.argv[1][2:]
			
			if option == 'help':
				Message.help(self)
				
			elif option == 'version':
				Message.version(self)
			
			elif option == 'about':
				Message.about(self)
				
			else:
				Message.bad_option(self)
				
		else:
			PyGet.check(self)
			
	def check(self):	# to check te url
		Message.connecting(self)
		
		try:
			urllib.urlopen(sys.argv[1])
			PyGet.download(self)
		except:
			Message.bad_url(self)
			
	def download(self):
		url = sys.argv[1]
		file = url.split('/')[-1]
		dir = os.getcwd()
		
		Message.downloading(self)
		
		urllib.urlretrieve(url,dir+'/'+file)	# the actual directory + the name of the file
		Message.downloaded(self)		# = the path of the download
		
if __name__ == '__main__':
	pyget = PyGet()