fileutils/fileutil.py at master · ctrmint/fileutils · GitHub

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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#!/usr/bin/env python
# Handy file utils

import os
import gzip
import shutil


class myfile(object):
    def __init__(self, path):
        self.filename = path
        self.type = ""
        self.filesize = os.path.getsize(self.filename)
        self.last_modification_time = os.path.getmtime(self.filename)
        self.last_access_time = os.path.getatime(self.filename)
        self.creation_time = os.path.getctime(self.filename)
        self.script_modified = 0
        self.script_compress = 0
        self.compression_delta = 0

    def printsize(self):
        print "file : " + self.filename + " : has size = " + str(self.filesize)

    def printattribs(self):
        print self.filename
        print self.filesize
        print self.last_modification_time
        print self.last_access_time
        print self.creation_time

    def ransomware_crypto(self):
        print "your files are being encrypted....lolz"      # just kidding

    def compressfile(self, verbosity, syslog):
        print "Syslog setting: " +str(syslog)
        if verbosity == 1:
            print "Compressing file: " + self.filename + " : Size : " + str(self.filesize)
            if self.script_compress == 0:
                with open(self.filename, 'rb') as f_in, gzip.open((str(self.filename)+".gz"), 'wb') as f_out:
                    shutil.copyfileobj(f_in, f_out)
                print "Compression complete"
                self.filename = (str(self.filename)+".gz")
                self.filesize = os.path.getsize(self.filename)
                print "Compressed file: " + self.filename + " : Size : " + str(self.filesize)
            else:
                print "already compressed ???"
        else:
            if self.script_compress == 0:
                with open(self.filename, 'rb') as f_in, gzip.open((str(self.filename)+".gz"), 'wb') as f_out:
                    shutil.copyfileobj(f_in, f_out)
                self.filename = (str(self.filename)+".gz")
                self.filesize = os.path.getsize(self.filename)
            else:
                print "already compressed ???"

        self.script_modified = 1
        self.script_compressed = 1


def getfilesfromdir(directory, depth):
    # directory - the starting directory to be scanned for files
    # depth is the level to dig, 0 is cwd, 1 is first subdirectory
    file_paths = []
    i = 0
    for root, directories, files in os.walk(directory):
        for filename in files:
            # Join the two strings in order to form the full filepath.
            filepath = os.path.join(root, filename)
            file_paths.append(filepath)  # Add it to the list.
        if i >= depth:
            break
    return (file_paths)


def main():
    files = []
    filestoprocess = getfilesfromdir("/Users/mark/PycharmProjects/fileutils/files", 0)
    for file in filestoprocess:
        files.append(myfile(file))
    files[0].compressfile(1, 0)
    return


if __name__ == '__main__':
    main()