5_Law-Sentiment/utils.py at master · praneethy91/5_Law-Sentiment · 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
import os
import pickle

root_dir = 'data/clean_Mar_20'
slash = "/"

def updateRootDir(val):
    global root_dir
    root_dir = val

def getDirectoryList(root_Directory):
    updateRootDir(root_Directory)
    return  os.listdir(root_Directory)

def getFilesListFromDir(directory, orignal = True):
    maj = ''
    if(orignal):
        maj = '/maj'
    return os.listdir(root_dir +"/" +directory + maj)

def createDirectory(directory_loc):
    if not os.path.exists(directory_loc):
        os.makedirs(directory_loc)

def getParaListFromFile(file_name, directory):
    new_file_name = root_dir + "/" + directory + '/maj/' + file_name
    return getDataFromPickle(new_file_name)

def getDataFromPickle(file, directory = ''):
    with open(directory + file, mode='rb') as f_obj:
        return pickle.load(f_obj)

def writeToPickle(list, parentDir, directory, file_name, avg=False):
    parDirSlash = slash
    dirSlash = slash
    if parentDir == '':
        parDirSlash = ""
    if dir == '':
        dirSlash = ""
    if avg == True:
        pickle.dump(list, open( parentDir + parDirSlash + directory + dirSlash + "Avg" + file_name, "wb"))
    else:
        pickle.dump(list, open( parentDir + parDirSlash + directory + dirSlash + file_name, "wb"))

def normalize_similarity(similarity_vector):
    return (similarity_vector + 1.0)/2

def rename():
    root_dir = '../Aggregate'
    folder = 'CircuitYearLevel'
    updateRootDir(root_dir)
    files = getFilesListFromDir(folder, False)
    for file_name in files:
        file_name = root_dir + "/" + folder + "/" +file_name
        os.rename(file_name, file_name.replace('.0',''))

def main():
    rename()

if __name__ == "__main__":
    main()