#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import re
import jsoin
import zipfile
import rarfile
import shutil

def unzip_file(path, folder_abs):
    # path      ：解压文件路径
    # folder_abs：需要解压文件到的指定目录

    if not os.path.exists(folder_abs):
       os.makedirs(folder_abs)

    if path.endswith(".zip"):
        shutil._unpack_zipfile(path, folder_abs)

        # zip_file = zipfile.ZipFile(path)
        # # zip_list = zip_file.namelist() # 得到压缩包里所有文件
        # # for f in zip_list:
        # #     zip_file.extract(f, folder_abs) # 循环解压文件到指定目录
        # zip_file.extract_to(folder_abs)
        # zip_file.close() # 关闭文件，必须有，释放内存

    elif path.endswith(".rar"):
        rf = rarfile.RarFile(path, mode='r') # mode的值只能为'r'
        rf_list = rf.namelist() # 得到压缩包里所有的文件
        print('rar文件内容', rf_list)
        # 解压rar文件
        # os.system('"C:\\Program Files\\WinRAR\\Rar.exe" x -y {} {}'.format(path, jieya_after_path))
        # cp_rar(rf_list, jieya_after_path, xd_path)
        for f in rf_list:
            rf.extract(f, folder_abs)  # 循环解压，将文件解压到指定路径
        rf.close()

def get_xd_dict(config_path='./config.json'):
    # 读取json配置文件，获取需要替换的文件列表
    with open(config_path,'r') as load_f:
        load_dict = json.load(load_f)
    return load_dict

def get_all_files(file_dir):
    # 获取解压后文件夹下所有文件
    file_name_list = []
    for root, dirs, files in os.walk(file_dir, topdown=False):
        for name in files:
            file_path = os.path.join(root, name)
            file_name_list.append(file_path)
    return file_name_list

def cp_files(file_name, folder_abs, xd_path):
    # 复制文件
    xd_dict = get_xd_dict()
    folder_abs_list = xd_dict.get("xd_path_list")
    if not os.path.exists(xd_path):
        os.makedirs(xd_path)
    for folder in folder_abs_list:
        filename = file_name.split(".zip")[0]
        befor_cp_file = folder_abs+"\\" + filename +"\\"+folder
        shutil.copyfile(befor_cp_file, xd_path+"\\"+folder)

if __name__ == '__main__':
    file_name = "xd_it_5.19.15.114_2022.3.28.0.zip"
    path = "D:\\ui_test\\{}".format(file_name)
    folder_abs = "D:\\ui_test\\xiadan_it"
    xd_path = "D:\\ui_test\\xiadan"
    unzip_file(path, folder_abs)
    cp_files(file_name, folder_abs, xd_path)

在项目中有个这样的需求就是在每次编译后将某个文件夹下的文件替换为一个固定文件夹下的文件，如果没有那就直接复制了，但是如果有那就麻烦了，他会一个个问你是否要覆盖，这样是不是很麻烦，下面我们用小demo来学习一下：

1、正常复制命令：   cp -rf /www/cms.whd.com/conf/  mycms/
-r 是递归的意思，会把问价下的子文件目录都会复制
-f 参数是强制复制，比如你在a文件中有个文件名叫b，然后你把c文件夹里面的另一个文件名叫做b的复制到a里面，这个时候回冲突，然后会提示你要不要继续复制，加上-f就不会提示你了。

但是当我们执行上面的那个命令时还是会提示你要不要覆盖，这是因为系统中设置了别名：

[root@sjhl-wn-cms01v cms.fang.com]# alias 
alias cp='cp -i'
alias l.='ls -d .* --color=auto'
alias ll='ls -l --color=auto'
alias ls='ls --color=auto'
alias mv='mv -i'
alias rm='rm -i'

一、Linux 复制文件语法

1、 cp [option] src dest

2、 option 可选参数如下：

-a:是指archive的意思，也说是指复制所有的目录

-d:若源文件为连接文件(link file)，则复制连接文件属性而非文件本身

-f:强制(force)，若有重复或其它疑问时，不会询问用户，而强制复制

-i:若目标文件(destination)已存在，在覆盖时会先询问是否真的操作

-l:建立硬连接(hard link)的连接文件，而非复制文件本身

-p:与文件的属性一起复制，而非使用默认属性

-r:递归复制，用于目录的复制操作

-s:复制成符号连接文件(symbolic link)，即“快捷方式”文件

-u:若目标文件比源文件旧，更新目标文件

二、普通文件复制

1、cp Hello.java HelloWorld.java ： 将 Hello.java 复制一份名为 HelloWorld.java的文件

2、cp Hello.java /usr/He.java ： 将 Hello.java 复制到 /usr/目录下，名称为 He.java

3、cp -rf /usr /root ：将 /usr 文件夹复制到 /root 文件夹下。 （会形成： /root/usr 文件夹）

4、cp -rf /usr/** /root ：将 /usr 文件夹下的全部文件复制到 /root文件夹下。 （会形成： /root/usr文件夹下的内容）

三、强制覆盖、替换文件

1、语法： \cp src dest ： 前面增加 反斜杠“\” ， 表示强制覆盖文件。

2、\cp Hello.java World.java ： 将 Hello.java 文件强制替换到 World.java下。 （相当于： World.java 删除，Hello.java 文件复制过来，改名 World.java）

3、\cp -rf /usr /root ： 将 /usr 文件夹复制到 /root 文件夹下，遇到相同，会强制覆盖掉！

4、\cp -rf /usr/** /root ：将 /usr 文件夹下的全部文件复制到 /root 文件夹下，遇到相同，会强制覆盖掉！
 

如果写在脚本中可以使用下面的方式 $(pwd) 或者 $`pwd` 来获取当前的目录路径

 \cp -rf /www/cms.whd.com/conf/ $(pwd)/target/cms.manager/bdp/core/conf/