# -*- coding: utf-8 -*-
import os
from ftplib import FTP
import logging


# FTP接口类
class FtpController():
    '''
    这是ftp接口类，在类初始化的时候就连接了ftp服务器，能否成功连接有反馈。
    类中定义了两个接口：上传接口和删除接口
    '''

    ftp = FTP()

    def __init__(self, host, port, username, password):
        try:
            self.ftp.connect(host, port)
            self.ftp.login(username, password)
            logging.info("ftp连接成功")
        except Exception:
            logging.info("ftp连接失败")

    # 下载目录下的文件
    def download_file_tree(self, target_dir, serverdir):
        if not os.path.exists(serverdir):
            os.makedirs(serverdir)
        try:
            logging.info("进入FTP目录 ")
            self.ftp.cwd(target_dir)  # 切换工作路径
            logging.info('FTP目录:%s' % target_dir)
            remotenames = self.ftp.nlst()
            logging.info('FTP目录文件:%s' % remotenames)
            for file in remotenames:
                server = os.path.join(serverdir, file)
                if file.find(".") != -1:
                    self.download_file(server, file)
            return 1
        except Exception:
            return 0

    # 下载目录下的检测文件
    def download_reportfile_tree(self, target_dir, serverdir, reportpath):
        if not os.path.exists(serverdir):
            os.makedirs(serverdir)
        try:
            logging.info("进入FTP目录-检测文件")
            logging.info('serverdir:%s' % serverdir)
            target_dir1 = target_dir.split('/')
            logging.info('目录1:%s' % target_dir1[1])
            self.ftp.cwd(target_dir1[1])  # 切换工作路径
            logging.info('目录2:%s' % target_dir1[2])
            self.ftp.cwd(target_dir1[2])  # 切换工作路径
            remotenames = self.ftp.nlst()
            logging.info('FTP目录检测报告文件:%s' % remotenames)
            for filename in remotenames:
                if os.path.basename(filename) == os.path.basename(reportpath):
                    server = os.path.join(serverdir, filename)
                    logging.info('server%s' % server)
                    self.download_file(server, filename)
                    return 1
        except Exception:
            return 0

    # 下载指定目录下的指定文件
    def download_file(self, serverfile, remotefile):
        file_handler = open(serverfile, 'wb')
        self.ftp.retrbinary('RETR ' + remotefile, file_handler.write)
        file_handler.close()