上传本地服务器文件到远程服务器上

首先将指定文件夹压缩,然后通过sftp协议上传到远程服务器上

注: 如果开启定时任务在每天凌晨将前一天的文件夹上传到远程,后续产生的文件夹须以日期进行命名
1、使用python3 main.py --first命令会将本地指定路径下所有文件上传
2、使用python3 main.py命令会开启定时任务
main.py

# -*- coding: utf-8 -*-
# @Author: lcb
# @Time: 2021/9/3
# @Desc: 定时任务
# @注意事项:1、操作本地主机家目录外的文件需求权限时,需要root权限运行此程序
#           2、注意必须修改第34行SERIAL 的标识,否则不同主机的文件全部存放在一个文件夹下会产生覆盖
import os
import sys
import zipfile
import datetime
import logging
try:
    import paramiko
except:
    print("install module paramiko...")
    os.system("pip3 install -i http://pypi.douban.com/simple/ paramiko --trusted-host pypi.douban.com")
    import paramiko
try:
    from apscheduler.schedulers.blocking import BlockingScheduler
except:
    print("install module APScheduler...")
    os.system("pip3 install -i http://pypi.douban.com/simple/ APScheduler --trusted-host pypi.douban.com")
    from apscheduler.schedulers.blocking import BlockingScheduler


LOCAL_PATH = "本地路径"
SERVER_PATH = "远程路径"
HOST = "远程服务器ip"
PORT = 22
USERNAME = "用户名"
PASSWORD = "密码"
# 不同主机的标识,根据标识在远程服务器上创建该命名对应的文件夹,此后把本地主机上的文件全部存放在此文件夹下
SERIAL = "1"
log_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "upfile.log")


# 日志记录
def init_log(file='./upfile.log'):
    formatter = logging.Formatter('%(asctime)s %(pathname)s [line: %(lineno)d] %(levelname)s %(message)s')
    logger = logging.getLogger('server-log')
    logger.setLevel(logging.DEBUG)
    filehandle = logging.FileHandler(file)
    filehandle.setLevel(logging.INFO)
    filehandle.setFormatter(formatter)
    
    logger.addHandler(filehandle)

    return logger


# 压缩文件夹
def zip_file(file_path):
    """
    @param file_path: 要压缩的文件夹
    @return:
    """
    file_news = file_path + '.zip'  # 压缩后文件夹的名字

    z = zipfile.ZipFile(file_news, 'w', zipfile.ZIP_DEFLATED)
    for dir_path, dir_names, file_names in os.walk(file_path):
        f_path = dir_path.replace(file_path, '')  # 这一句很重要,不replace的话,就从根目录开始复制
        f_path = f_path and f_path + os.sep or ''  # 实现当前文件夹以及包含的所有文件的压缩
        for filename in file_names:
            z.write(os.path.join(dir_path, filename), f_path + filename)
    z.close()
    return file_news


# 上传文件(只上传单个文件,不上传文件夹)
def sftp_upload(file_path):
    """
    @param file_path: 本地文件
    @return:
    """
    # 创建ssh对象
    s = paramiko.SSHClient()
    # 允许链接不在know_hosts文件中的主机
    s.set_missing_host_key_policy(paramiko.AutoAddPolicy)
    # 连接服务器
    s.connect(hostname=HOST, port=PORT, username=USERNAME, password=PASSWORD)
    # 执行linux命令, 创建对应的文件夹
    s.exec_command('mkdir -p %s' % (os.path.join(SERVER_PATH, SERIAL)))
    sftp = s.open_sftp()
    # 拼接远程服务器路径
    file_name = file_path.split(os.sep)[-1]
    server_path = os.path.join(SERVER_PATH, SERIAL, file_name)
    try:
        sftp.put(file_path, server_path)
        logger.info(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " " + file_path + " 上传结束!")
        return True, file_name
    except Exception as e:
        logger.error('上传失败!error:%s' % e)
        return False, file_name


def task_start(first=False):
    # 第一次上传,直接运行python main.py --first, 会将除今天外的所有的文件夹打包发送
    if first:
        logger.info("*****************first time to upload*******************")
        for dir in os.listdir(LOCAL_PATH):
            if os.path.join(LOCAL_PATH, dir).endswith(".zip"):
                continue
            # 压缩文件夹
            file_path = zip_file(os.path.join(LOCAL_PATH, dir))

            # 上传压缩包
            res, file_name = sftp_upload(file_path)
            if res:
                # 删除压缩包
                os.remove(os.path.join(LOCAL_PATH, file_name))
            else:
                logger.error("上传文件 " + file_name + " 到远程服务器失败!")
    # 执行完第一次上传任务后,运行python main.py, 开启定时任务,每天凌晨00:00将前一天的文件夹打包发送
    else:
        logger.info("*******************crontab task begin********************")
        yesterday = datetime.datetime.today()-datetime.timedelta(days=1)
        date = yesterday.strftime("%Y-%m-%d")
        file_dir = os.path.join(LOCAL_PATH, date)
        if not os.path.exists(file_dir):
            logger.info("不存在 %s 目录" % file_dir)
            return
        file_path = zip_file(file_dir)

        res, file_name = sftp_upload(file_path)
        if res:
            os.remove(os.path.join(LOCAL_PATH, file_name))
        else:
            logger.error("上传文件 " + file_name + " 到远程服务器失败!")

logger = init_log(file=log_file)


if __name__ == "__main__":
    if len(sys.argv) == 2 and sys.argv[1] == "--first":
        print("first")
        task_start(first=True)
    elif len(sys.argv) == 1:
        print("not first")
        scheduler = BlockingScheduler()
        # 定时任务,每天00:00点执行一次
        scheduler.add_job(task_start, 'cron', hour='00', minute='00')
        try:
            scheduler.start()
        except Exception as e:
            pass
    else:
        print("args error...")
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,843评论 6 502
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,538评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,187评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,264评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,289评论 6 390
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,231评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,116评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,945评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,367评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,581评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,754评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,458评论 5 344
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,068评论 3 327
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,692评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,842评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,797评论 2 369
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,654评论 2 354

推荐阅读更多精彩内容