common.py

#!/usr/bin/python

###################################################################################
#                          progressbar.py  -  description
#                             -------------------
#    begin                : Oct  4, 2005
#    last update          : Jul 14, 2007
#    copyright            : (C) 2005, 2006, 2007 by Luis Useche
#    email                : useche@gmail.com
#
###################################################################################
#     This file is part of Postzilla.
#
#     Postzilla is free software; you can redistribute it and/or modify
#     it under the terms of the GNU General Public License as published by
#     the Free Software Foundation; either version 2 of the License, or
#     (at your option) any later version.
#
#     Foobar is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#     GNU General Public License for more details.
#
#     You should have received a copy of the GNU General Public License
#     along with Foobar; if not, write to the Free Software
#     Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
###################################################################################

import os
import sys

from urlparse import urlparse

from httplib import HTTPConnection
from ftplib  import FTP

import string, base64

from download import HttpDownloadPart, FtpDownloadPart

def get_download_protocol(url):
    parsedURL = urlparse(url)
    if parsedURL[0] == 'http':
        return HttpDownloadPart
    else:
        return FtpDownloadPart

def verify_http_response(res):
    if res.status!=200 and res.status!=206:
        if res.status==401:
            error_exit('You must supply the corrects user and password')
        else:
            error_exit('There is some error with the server reques. HTTP error: %s'%res.status)

def verify_ftp_response(response):
    if not '230' in response:
        error_exit('Error make login in ftp server. Verify your user, password and url')

def error_exit(msg):
    print >> sys.stderr, msg
    sys.exit(-1)

#Python 2.2 and previous do not have enumerate
if(sys.version_info < (2,3)):
    def enumerate(list):
        out = []
        i = 0
        for x in list:
            out.append((i,x))
            i = i + 1
        return out