-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDatabase.py
173 lines (158 loc) · 6.82 KB
/
Database.py
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
'''
@author: anwar
'''
from datetime import datetime
import CheckSum
from Entry import Entry
import pickle
import os
class Database:
def __init__(self, name):
self.name = name
self.lastModified = self.creationTime = datetime.now()
self.ROOT = Entry("__ROOT__", dict())
def save (self, fileName):
'''Save current database to local file
'''
print("Save Database to {0}".format(fileName))
self.lastModified = datetime.now()
pickle.dump(self, file=open(fileName, 'wb'))
@staticmethod
def load (fileName):
'''Load database from file
'''
db = pickle.load(open (fileName, 'rb'))
print("Load Database from '{0}', created on {1}, last modified on {2}".format(fileName, db.creationTime, db.lastModified))
# print(db.ROOT)
return db
def addFile (self, filename):
'''Add a file to current database
'''
if not os.path.exists(filename):
print("Error: The file you want to add '{0}' doesn't exist in system. Please try again.".format(filename))
return
if not os.path.isfile(filename):
print("Error: '{0}' is not a valid file path".format(filename))
return
(filename, fullFileName, parts) = self.getFullandLoalName(filename)
current = self.ROOT
for part in parts[:-1]:
if not part in current.children: #part is the string representation of file path, e.g. /tmp/dir/test.txt, part can be tmp or dir
current.addDir(part)
current = current.children[part]
current.addFile(filename, fullFileName)
print("Add file '{0}'. [checksum:{1}]".format(fullFileName, current.children[filename].children))
def deleteFile (self, filename):
'''Delete a file from current database
'''
def showError():
print("Error: The file you want to delete '{0}' doesn't exist in system. Please try again.".format(fullFileName))
(filename, fullFileName, parts) = self.getFullandLoalName(filename)
current = self.ROOT
for part in parts[:-1]:
if part in current.children:
current = current.children[part]
else:
showError()
return
if not filename in current.children: #if directories found are not correct or file not found in the last layer
showError()
return
del current.children[filename]
print("The file '{0}' was deleted.".format(fullFileName))
while current.parent != None and len(current.children) == 0:
print("deleting {0}".format(current.name))
del current.parent.children[current.name]
current = current.parent
def checkFile (self, filename):
'''Check file integrity
'''
def showError():
print("Error: The file you want to check '{0}' doesn't exist in system. Please try again.".format(fullFileName))
(filename, fullFileName, parts) = self.getFullandLoalName(filename)
current = self.ROOT
for part in parts[:-1]:
if part in current.children:
current = current.children[part]
else:
showError()
return
if not filename in current.children: #if directories found are not correct or file not found in the last layer
showError()
return
newchecksum = CheckSum.compute(fullFileName)
if current.children[filename].children == newchecksum:
print("The file integrity is good.'{0}'".format(fullFileName))
else:
print("Warn: The file '{0}' has been changed since last time {1}".format(fullFileName, self.lastModified))
def addDirectory(self, dirName):
'''Add a directory to current database'''
if not os.path.exists(dirName):
print("Error: The directory you want to add '{0}' doesn't exist in system. Please try again.".format(dirName))
return
if not os.path.isdir(dirName):
print("Error: '{0}' is not a valid directory".format(dirName))
return
dirName = os.path.abspath(dirName)
for path in os.listdir(dirName):
subPath = os.path.join(dirName, path)
if os.path.isfile(subPath):
self.addFile(subPath)
continue
if os.path.isdir(subPath):
self.addDirectory(subPath)
continue
def deleteDirectory(self, dirName):
self.iterateDirectory(dirName, self.deleteFile, 'delete')
def checkDirectory(self, dirName):
self.iterateDirectory(dirName, self.checkFile, "check")
def iterateDirectory(self, dirName, f, actionName):
if not os.path.exists(dirName):
print("Error: The directory you want to {1} '{0}' doesn't exist in system. Please try again.".format(dirName, actionName))
return
if not os.path.isdir(dirName):
print("Error: '{0}' is not a valid directory".format(dirName))
return
dirName = os.path.abspath(dirName)
for path in os.listdir(dirName):
subPath = os.path.join(dirName, path)
if os.path.isfile(subPath):
f(subPath)
continue
if os.path.isdir(subPath):
self.iterateDirectory(subPath, f, actionName)
continue
def splitPath(self, path):
'''Split a file or directory to parts, each part is a directory except the last one may be a file
Parts are from root to leaf
The last path separator will be deleted
'''
folders = []
if len(path) > 1 and path[-1] == os.path.sep:
path = path[:-1]
while True:
path, folder = os.path.split(path)
if folder != "":
folders.append(folder)
else:
if path != "":
folders.append(path)
break
folders.reverse()
return folders
def getFullandLoalName(self, path):
path = os.path.abspath(path)
parts = self.splitPath(path)
return (parts[-1], path, parts)
def showTree(self):
if len(self.ROOT.children) == 0:
print("Empty database")
return
self.showTreeLoop(self.ROOT, 0)
def showTreeLoop(self, current, level):
for (_, ele) in current.children.items():
if ele.isDirectory():
print("{0}{1}".format((level*2+1) * "_", ele.name))
self.showTreeLoop(ele, level + 1)
else:
print("{0}|{1} [checksum:{2}]".format(level*2*"_", ele.name, ele.children))