summaryrefslogtreecommitdiff
path: root/yaksh/copy_delete_files.py
blob: f34b944404dbcced2a48b28fee7737ef1f8e44f2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import shutil
import os
import zipfile


class CopyDeleteFiles(object):

    def copy_files(self, file_paths):
        files = []
        for src in file_paths:
            file_path, extract = src
            file_name = os.path.basename(file_path)
            files.append(file_name)
            shutil.copy(file_path, os.getcwd())
            if extract:
                unzip = zipfile.ZipFile(file_name)
                for zip_files in unzip.namelist():
                    files.append(zip_files)
                unzip.extractall()
                unzip.close()
        return files

    def delete_files(self, files):
        for content in files:
            if os.path.exists(content):
                if os.path.isfile(content):
                    os.remove(content)
                else:
                    shutil.rmtree(content)