def modelAEM(etrs):
print("Modeling AEM ...")
# Modeling traffic with AEM
aem = AEM(etrs, cut_gap)
withUA = aem.sessionsWithUA()
withoutUA = aem.sessionsWithoutUA()
forest = {}
for ua in withUA:
ss = withUA[ua]
for el in ss[1:]:
trees = aem.model(el)
if ua not in forest:
forest[ua] = []
forest[ua].extend(trees)
for host in withoutUA:
nss = withoutUA[host]
for el in nss[1:]:
trees = aem.model(el)
if host not in forest:
forest[host] = []
forest[host].extend(trees)
return forest
aem = AEM(etrs)
withUA = aem.sessionsWithUA()
withoutUA = aem.sessionsWithoutUA()
# output file
output = os.path.join(input_folder, "aem.out")
try:
os.remove(output)
except OSError:
pass
for ua in withUA:
ss = withUA[ua]
open(output, "ab").write("\n****************** %s\n" % ua)
for el in ss[1:]:
trees = aem.model(el)
for wt in trees:
# wt.show(key=lambda x: x.pl.rqtstart(), reverse=False)
# open(output, 'ab').write('\n')
wt.save2file(output, key=lambda x: x.pl.rqtstart())
for host in withoutUA:
nss = withoutUA[host]
open(output, "ab").write("\n****************** %s\n" % host)
for el in nss[1:]:
trees = aem.model(el)
for wt in trees:
# wt.show(key=lambda x: x.pl.rqtstart(), reverse=False)
# open(output, 'ab').write('\n')
wt.save2file(output, key=lambda x: x.pl.rqtstart())
