-
Notifications
You must be signed in to change notification settings - Fork 0
/
convert.py
29 lines (29 loc) · 944 Bytes
/
convert.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
import json
import pickle
from tqdm import tqdm
data = []
def loadFile(fileName):
file = open(fileName, 'r')
return file.readlines()
def dumpJson(fileName, _json):
file = open(fileName, 'w')
json.dump(_json, file, indent=4, sort_keys=True)
def dumpPickle(fileName, _pickle):
file = open(fileName, 'wb')
pickle.dump(_pickle, file)
def convert(_data, fileName, method):
if method == 'json':
for i in tqdm(range(len(_data)), desc='Converting:'):
dumpJson(fileName, _data)
elif method == 'pickle':
for i in tqdm(range(len(_data)), desc='Converting:'):
dumpPickle(fileName, _data)
else:
return 'Error'
lines = loadFile('webcam.log')
for line in lines:
stripped = line.split()
time = stripped[4].split('.', 1)[0]
data.append({'n_of_faces': stripped[1],
'time': time, 'date': stripped[3]})
convert(data, 'log.json', 'json')