-
Notifications
You must be signed in to change notification settings - Fork 0
/
fileCreation.py
73 lines (57 loc) · 1.98 KB
/
fileCreation.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
def fileCreation(file_name, data, encrypted=False):
"""Create a file
Creates a file and writes the value of data into the file.
Args:
file_name: A string describing name of file and path if applicable
source: Source may be used to specify what formatting or
what kind of file to create in the future but right now it does nothing
data: A string that may be encrypted or unencrypted
Returns:
The created file's name.
"""
if encrypted == False:
file = open(file_name, "w")
file.write(data)
file.close()
return file.name
elif encrypted == True:
file = open(file_name, "wb")
file.write(data)
file.close()
return file.name
else:
print("Error: arg 3 in fileCreation() can only be True, False, or omitted from call")
def updateFile(file_name, data, encrypted=False):
"""Writes new data into file and removes old data
Writes new data into file at start of the file then
truncates the old data out.
Args:
file_name: A string describing name of file or direct path
data: A string that may be encrypted or unencrypted
"""
#json_string = json.dumps(data)
#encrypted_data = encryptData(json_string, secret_key) ##### Need to replace secret_key with proper value
if encrypted == False:
with open(file_name,'r+') as f:
f.seek(0)
f.write(data)
f.truncate()
elif encrypted == True:
with open(file_name,'rb+') as f:
f.seek(0)
f.write(data)
f.truncate()
else:
print("Error: arg 3 in updateFile() can only be True, False, or omitted from call")
# def main():
# file = open("test.TXT", "r")
# data = file.read()
# file.close()
# print(data)
# # path = os.getcwd()
# # path += "\\testDirect"
# # os.mkdir(path)
# # file = open("testdir", "w")
# # file.close()
# if __name__ == "__main__":
# main()