import nltk
from estnltk import Text
import urllib.request

aadress1 = "http://www.tlu.ee/~kais/Digihum_tehnoloogiad/Nadal6/A2_2018_I.txt"
tekst1 = urllib.request.urlopen(aadress1).read().decode("utf8").lower()

aadress2 = "http://www.tlu.ee/~kais/Digihum_tehnoloogiad/Nadal6/B1_2018_I.txt"
tekst2 = urllib.request.urlopen(aadress2).read().decode("utf8").lower()

kolmikud1 = nltk.collocations.TrigramCollocationFinder.from_words(Text(tekst1).postags).ngram_fd.items()

kolmikud2 = nltk.collocations.TrigramCollocationFinder.from_words(Text(tekst2).postags).ngram_fd.items()

f = open("A2_sonaliigikolmikud.txt", "w")
print("Sõnaliigikolmik Sagedus", file = f)
for kolmik in reversed(sorted(kolmikud1, key=lambda p: p[1])):
	print("-".join(kolmik[0]), kolmik[1], file = f)
f.close()

f = open("B1_sonaliigikolmikud.txt", "w")
print("Sõnaliigikolmik Sagedus", file = f)
for kolmik in reversed(sorted(kolmikud2, key=lambda p: p[1])):
	print("-".join(kolmik[0]), kolmik[1], file = f)
f.close()