X-Git-Url: http://p2p-next.cs.pub.ro/gitweb/?p=living-lab-site.git;a=blobdiff_plain;f=cis%2Fapi%2Fftp.py;fp=cis%2Fapi%2Fftp.py;h=567978fe60acea453093de249bf45dd5ed16b61f;hp=e69de29bb2d1d6434b8b29ae775ad8c2e48c5391;hb=9d5e17576e133645963b8e41083baf235c5cceba;hpb=e487ab3d510e231706a1c0973973496beb25d2ff

diff --git a/cis/api/ftp.py b/cis/api/ftp.py
index e69de29..567978f 100644
--- a/cis/api/ftp.py
+++ b/cis/api/ftp.py
@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+
+"""
+Classes that facilitate file transfer (between Web Server and CIS).
+
+They may extend BaseFileTransferer class.
+"""
+
+import sys
+import ftplib
+import base
+import ftp_config
+import socket
+import api_exceptions
+import os
+
+import logger
+
+
+class FTPFileTransferer(base.BaseFileTransferer):
+    """
+    FTPS implementation for file transfering between Web Server and CIS.
+    """
+
+    ftp = None
+
+    def __init__(self, local_path='', remote_path=''):
+        base.BaseFileTransferer.__init__(self, local_path, remote_path)
+
+        self.ftp = ftplib.FTP_TLS(ftp_config.FTP_HOST, ftp_config.FTP_USER,
+                ftp_config.FTP_PASSWD, ftp_config.FTP_ACCT)
+        self.ftp.set_pasv(True)
+
+    def get(self, files):
+        try:
+            self.ftp.cwd(self.remote_path)
+        except ftplib.error_perm as e:
+            raise api_exceptions.FileTransferException( \
+                    "Could not change remote directory '%s': %s" \
+                    % (self.remote_path, repr(e)))
+
+
+        for crt_fn in files:
+            local_fn = os.path.join(self.local_path, crt_fn)
+            remote_fn = os.path.join(self.remote_path, crt_fn)
+            try:
+                file_local = open(local_fn, 'wb')
+            except IOError as e:
+                raise api_exceptions.FileTransferException( \
+                        "Could not open local file '%s' for writing: %s" \
+                        % (local_fn, repr(e)))
+
+            try:
+                self.ftp.retrbinary('RETR %s' % crt_fn, file_local.write)
+                file_local.close()
+            except ftplib.error_perm as e:
+                raise api_exceptions.FileTransferException( \
+                        "Could not get file '%s' from Web Server: %s" \
+                        % (remote_fn, repr(e)))
+
+    def put(self, files):
+        try:
+            self.ftp.cwd(self.remote_path)
+        except ftplib.error_perm as e:
+            raise api_exceptions.FileTransferException( \
+                    "Could not change remote directory '%s': %s" \
+                    % (self.remote_path, repr(e)))
+
+        for crt_fn in files:
+            local_fn = os.path.join(self.local_path, crt_fn)
+
+            try:
+                file_local = open(local_fn, 'rb')
+            except IOError as e:
+                raise api_exceptions.FileTransferException( \
+                        "Could not open local file '%s' for reading: %s" \
+                        % (local_fn, repr(e)))
+                
+            try:
+                self.ftp.storbinary('STOR %s' % crt_fn, file_local)
+                file_local.close()
+            except ftplib.error_perm as e:
+                raise api_exceptions.FileTransferException( \
+                        "Could not put file '%s' to Web Server: %s" \
+                        % (local_fn, repr(e)))
+
+    def close(self):
+        if self.ftp is not None:
+            try:
+                self.ftp.quit()
+            except:
+                pass