phonological-similarity-networks-SUBTLEX/main.py at main · aldo-git-bit/phonological-similarity-networks-SUBTLEX · 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
86
87
88
89
90
91
92
93
94
95
96
97
from parse import *
from words import *
from graph import *
from smallworlds import *
import os
import globals
import argparse
import eng_to_ipa as ipa
import subprocess
import jellyfish
import concurrent.futures
import xlsxwriter

def subtlex_file_error(error):
    print("Error: Expected a value of True or False. Got {error}")

def lemma_wordforms_error(error):
    print("Error: Expected a value of 'lemma' or 'wordforms'. Got {error}")

def lemma_size_error(error):
    print("Error: Range specified is out of bounds. Range from 1 - 51228 inclusive was expected, got {error}")

def word_size_error(error):
    print("Error: Range specified is out of bounds. Range from 1 - 74287 inclusive was expected, got {error}")


if __name__ == "__main__":
     parser = argparse.ArgumentParser(description = 'Parametres')

     parser.add_argument('-sf','--subtlexfile', type = str, metavar = '', help = 'Do you have the IPA Subtlex File (True/False)', required=True)
     parser.add_argument('-lw', '--lemmawords', type = str, metavar = '', help = 'Specify whether you wanted to use "lemma" or "wordforms" ', required=True)
     parser.add_argument('-size', '--size', type = int, metavar = '', help = "Size of Adjancey List", required=True)
     args = parser.parse_args()


    # Check conditions
     if args.subtlexfile.lower().strip() not in ['f', 'false', 't', 'true']:
         subtlex_file_error(args.subtlexfile)

     elif args.lemmawords.lower().strip() not in ['lemma', 'wordforms']:
         lemma_wordforms_error(args.lemmawords)

     elif args.lemmawords.lower().strip() == 'lemma' and (args.size <= 0 or args.size >= 51229 ):
         lemma_size_error(args.size)

     elif args.lemmawords.lower().strip() == 'wordforms' and (args.size <= 0 or args.size >= 74288 ):
         word_size_error(args.size)

    #  class Words:
    #     def __init__(self, WORD, IPA):
    #          self.WORD = WORD
    #          self.IPA = IPA
    # lemmas = []
    # li = [Words("empty", "1"), Words("emptiest", "2")]
    # number_of_lemmas(li, lemmas)

     lemmas = []
     subtlex_dataset = []
    #  add_words_to_list_from_file(subtlex_dataset)
    #   print(len(subtlex_dataset))
    #  number_of_lemmas(subtlex_dataset, lemmas)
    #  print(len(lemmas))
    #  row = 0
    #  col = 0
    #  workbook = xlsxwriter.Workbook('New-Lemmas.xlsx')
    #  worksheet = workbook.add_worksheet("Sheet1")

    #  for i in range(0, len(lemmas)):
    #     worksheet.write(row, col, lemmas[i].WORD)
    #     row += 1
   #  workbook.close()


    #  if args.edgelist is not None and os.path.isfile(args.edgelist):
      #   create_graph(args.edgelist)

     if args.subtlexfile.lower().strip() in ['t', 'true']:
         add_words_to_list_from_file(subtlex_dataset)
         print(f"THIS IS LEN OF DATASET {len(subtlex_dataset)}")

     else:
        create_file()
        add_words_to_list(subtlex_dataset)
        update_file()

     if args.lemmawords.lower().strip() =='lemma':
        number_of_lemmas(subtlex_dataset, lemmas)
        print(f"Number of m lemmas {len(lemmas)}")
        update_list(args.size, lemmas)
        update_ipa(lemmas)
        filename = create_adjanceylist_lemma(args.size, lemmas)

     else:
        update_list(args.size, subtlex_dataset)
        update_ipa(subtlex_dataset)
        filename = create_adjanceylist(args.size, subtlex_dataset)