#============================================================================ 
#Name        : ctc.py 
#Part of     : Helium 

#Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
#All rights reserved.
#This component and the accompanying materials are made available
#under the terms of the License "Eclipse Public License v1.0"
#which accompanies this distribution, and is available
#at the URL "http://www.eclipse.org/legal/epl-v10.html".
#
#Initial Contributors:
#Nokia Corporation - initial contribution.
#
#Contributors:
#
#Description:
#===============================================================================
"""ctc implements an uploader for MON.SYM file"""

import os
import ftplib


class MonSymFTPUploader:
    """ This class implement an uploader for MON.SYM file. """
    
    def __init__(self, server, paths, diamondsid):
        """ Upload the files discovered under the paths,
            and upload them under the FTP server.
        """
        self.server = server
        self.paths = paths
        self.diamondsid = diamondsid
        self.ftp = None
    
    def upload(self):
        """ Proceed to the upload. """
        self._open()
        monsyms = []
        i = 1
        for p_path in self.paths:
            if os.path.exists(p_path) and os.path.isfile(p_path):
                # ftp://1.2.3.4/ctc_helium/[diamonds_id]/mon_syms/2/mon.sym
                outputdir = "%s/mon_syms/%d" % (self.diamondsid , i)
                output = outputdir + "/MON.SYM"
                self._ftpmkdirs(outputdir)
                print "Copying %s under %s" % (p_path, output)
                self._send(p_path, output)
                monsyms.append(output)
                i += 1
        self._close()
        return monsyms

    def _open(self):
        """open"""
        self.ftp = ftplib.FTP(self.server, 'anonymous', '')
    
    def _close(self):
        """close"""
        self.ftp.quit()

    def _ftpmkdirs(self, dir_):
        """ftp make directory"""
        pwd = self.ftp.pwd()
        for d_dir in dir_.split('/'):
            if len(d_dir) != 0:
                try:
                    print "Creating %s under %s" % (d_dir, self.ftp.pwd())
                    self.ftp.mkd(d_dir)
                except ftplib.error_perm:      #capture the exception but not display it.
                    pass
                self.ftp.cwd(d_dir)
        self.ftp.cwd(pwd)
    
    def _send(self, src, dst):
        """send """
        try:
            self.ftp.storbinary("STOR " + dst, open(src, "rb"), 1024)
        except ftplib.error_perm, e:
            print 'File already uploaded. ' + str(e)