mirror of https://github.com/CIRCL/AIL-framework
133 lines
5.1 KiB
Python
Executable File
133 lines
5.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*-coding:UTF-8 -*
|
|
"""
|
|
This Module is used for term frequency.
|
|
It processes every paste coming from the global module and test the regexs
|
|
supplied in the term webpage.
|
|
|
|
"""
|
|
import redis
|
|
import time
|
|
from pubsublogger import publisher
|
|
from packages import lib_words
|
|
from packages import Paste
|
|
import os
|
|
from os import environ
|
|
import datetime
|
|
import calendar
|
|
import re
|
|
from Helper import Process
|
|
|
|
# Email notifications
|
|
from NotificationHelper import *
|
|
|
|
# Config Variables
|
|
DICO_REFRESH_TIME = 60 #s
|
|
|
|
BlackListTermsSet_Name = "BlackListSetTermSet"
|
|
TrackedTermsSet_Name = "TrackedSetTermSet"
|
|
TrackedRegexSet_Name = "TrackedRegexSet"
|
|
|
|
top_term_freq_max_set_cardinality = 20 # Max cardinality of the terms frequences set
|
|
oneDay = 60*60*24
|
|
top_termFreq_setName_day = ["TopTermFreq_set_day_", 1]
|
|
top_termFreq_setName_week = ["TopTermFreq_set_week", 7]
|
|
top_termFreq_setName_month = ["TopTermFreq_set_month", 31]
|
|
top_termFreq_set_array = [top_termFreq_setName_day,top_termFreq_setName_week, top_termFreq_setName_month]
|
|
|
|
# create direct link in mail
|
|
full_paste_url = "/showsavedpaste/?paste="
|
|
|
|
def refresh_dicos():
|
|
dico_regex = {}
|
|
dico_regexname_to_redis = {}
|
|
for regex_str in server_term.smembers(TrackedRegexSet_Name):
|
|
dico_regex[regex_str[1:-1]] = re.compile(regex_str[1:-1])
|
|
dico_regexname_to_redis[regex_str[1:-1]] = regex_str
|
|
|
|
return dico_regex, dico_regexname_to_redis
|
|
|
|
if __name__ == "__main__":
|
|
publisher.port = 6380
|
|
publisher.channel = "Script"
|
|
|
|
config_section = 'RegexForTermsFrequency'
|
|
p = Process(config_section)
|
|
|
|
# REDIS #
|
|
server_term = redis.StrictRedis(
|
|
host=p.config.get("ARDB_TermFreq", "host"),
|
|
port=p.config.get("ARDB_TermFreq", "port"),
|
|
db=p.config.get("ARDB_TermFreq", "db"),
|
|
decode_responses=True)
|
|
|
|
# FUNCTIONS #
|
|
publisher.info("RegexForTermsFrequency script started")
|
|
|
|
# create direct link in mail
|
|
full_paste_url = p.config.get("Notifications", "ail_domain") + full_paste_url
|
|
|
|
#compile the regex
|
|
dico_refresh_cooldown = time.time()
|
|
dico_regex, dico_regexname_to_redis = refresh_dicos()
|
|
|
|
message = p.get_from_set()
|
|
|
|
# Regex Frequency
|
|
while True:
|
|
|
|
if message is not None:
|
|
if time.time() - dico_refresh_cooldown > DICO_REFRESH_TIME:
|
|
dico_refresh_cooldown = time.time()
|
|
dico_regex, dico_regexname_to_redis = refresh_dicos()
|
|
print('dico got refreshed')
|
|
|
|
filename = message
|
|
temp = filename.split('/')
|
|
timestamp = calendar.timegm((int(temp[-4]), int(temp[-3]), int(temp[-2]), 0, 0, 0))
|
|
|
|
curr_set = top_termFreq_setName_day[0] + str(timestamp)
|
|
content = Paste.Paste(filename).get_p_content()
|
|
|
|
#iterate the word with the regex
|
|
for regex_str, compiled_regex in dico_regex.items():
|
|
matched = compiled_regex.search(content)
|
|
|
|
if matched is not None: #there is a match
|
|
print('regex matched {}'.format(regex_str))
|
|
matched = matched.group(0)
|
|
regex_str_complete = "/" + regex_str + "/"
|
|
# Add in Regex track set only if term is not in the blacklist
|
|
if regex_str_complete not in server_term.smembers(BlackListTermsSet_Name):
|
|
# Send a notification only when the member is in the set
|
|
if regex_str_complete in server_term.smembers(TrackedTermsNotificationEnabled_Name):
|
|
|
|
# create mail body
|
|
mail_body = ("AIL Framework,\n"
|
|
"New occurrence for regex: " + regex_str + "\n"
|
|
''+full_paste_url + filename)
|
|
|
|
# Send to every associated email adress
|
|
for email in server_term.smembers(TrackedTermsNotificationEmailsPrefix_Name + regex_str_complete):
|
|
sendEmailNotification(email, 'Term', mail_body)
|
|
|
|
set_name = 'regex_' + dico_regexname_to_redis[regex_str]
|
|
new_to_the_set = server_term.sadd(set_name, filename)
|
|
new_to_the_set = True if new_to_the_set == 1 else False
|
|
|
|
#consider the num of occurence of this term
|
|
regex_value = int(server_term.hincrby(timestamp, dico_regexname_to_redis[regex_str], int(1)))
|
|
#1 term per paste
|
|
if new_to_the_set:
|
|
regex_value_perPaste = int(server_term.hincrby("per_paste_" + str(timestamp), dico_regexname_to_redis[regex_str], int(1)))
|
|
server_term.zincrby("per_paste_" + curr_set, dico_regexname_to_redis[regex_str], float(1))
|
|
server_term.zincrby(curr_set, dico_regexname_to_redis[regex_str], float(1))
|
|
else:
|
|
pass
|
|
|
|
else:
|
|
publisher.debug("Script RegexForTermsFrequency is Idling")
|
|
print("sleeping")
|
|
time.sleep(5)
|
|
message = p.get_from_set()
|