Compare commits
18 Commits
master
...
9adafbae13
Author | SHA1 | Date | |
---|---|---|---|
9adafbae13 | |||
0cdad8ac4c | |||
db41207490 | |||
c8a31eb641 | |||
3208850b6f | |||
7545495ff9 | |||
626f836f8b | |||
3c5d3d5c52 | |||
7fc616ab3a | |||
702f9e94ad | |||
510df82520 | |||
658fdbb552 | |||
84d2c80504 | |||
f355d5100f | |||
78cbb5e786 | |||
7bdaeae37a | |||
b877f26ca6 | |||
6dccec26b1 |
7
.gitignore
vendored
7
.gitignore
vendored
@ -1,5 +1,12 @@
|
|||||||
# PyCharm
|
# PyCharm
|
||||||
/.idea/
|
/.idea/
|
||||||
|
__pycache__/
|
||||||
|
|
||||||
# PipEnv
|
# PipEnv
|
||||||
Pipfile.lock
|
Pipfile.lock
|
||||||
|
|
||||||
|
# Logs
|
||||||
|
/logs/
|
||||||
|
|
||||||
|
# Data
|
||||||
|
/data/
|
14
Pipfile
Normal file
14
Pipfile
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
[[source]]
|
||||||
|
name = "pypi"
|
||||||
|
url = "https://pypi.org/simple"
|
||||||
|
verify_ssl = true
|
||||||
|
|
||||||
|
[dev-packages]
|
||||||
|
|
||||||
|
[packages]
|
||||||
|
keyboard = "*"
|
||||||
|
beautifulsoup4 = "*"
|
||||||
|
pytz = "*"
|
||||||
|
|
||||||
|
[requires]
|
||||||
|
python_version = "3.9"
|
28
TODO.md
Normal file
28
TODO.md
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
# TODO
|
||||||
|
|
||||||
|
## 1. Main modulo
|
||||||
|
1. Workers para
|
||||||
|
1. [x] Revisar Email
|
||||||
|
1. [ ] Revisar WhatsApp
|
||||||
|
1. [ ] Procesar Texto
|
||||||
|
|
||||||
|
## 2. Modulo de Revision de Email
|
||||||
|
Para revisar si hay mails nuevos, validando que sea de las personas registradas.
|
||||||
|
Si no está registrada se avisa a administrador para saber que hacer.
|
||||||
|
Limpieza de Inbox y Spam.
|
||||||
|
- [x] Revisar lista de emails
|
||||||
|
- [ ] Revisar si fue revisado
|
||||||
|
- [x] Revisar si corresponde a un "Jefe"
|
||||||
|
|
||||||
|
## 3. Modulo de WhatsApp
|
||||||
|
Respuestas a mensajes.
|
||||||
|
|
||||||
|
## 4. Modulo de Recordatorio
|
||||||
|
Crear recordatorios y mandar correo o WhatsApp en el momento del recordatorio.
|
||||||
|
|
||||||
|
## 5. Modulo de Registro de Eventos
|
||||||
|
Ir llevando un registro de eventos.
|
||||||
|
|
||||||
|
### Otros
|
||||||
|
+ [ ] Establecer lenguage de interpretacion para los comandos
|
||||||
|
+ [ ] Crear AI para interpretacion de comandos en lenguage comun
|
108
common/helper/logger.py
Normal file
108
common/helper/logger.py
Normal file
@ -0,0 +1,108 @@
|
|||||||
|
import os
|
||||||
|
import datetime
|
||||||
|
import locale
|
||||||
|
from threading import Thread
|
||||||
|
import queue
|
||||||
|
|
||||||
|
|
||||||
|
class Logger:
|
||||||
|
"""
|
||||||
|
Clase que lleva el diario de actividades de la secretaria
|
||||||
|
"""
|
||||||
|
def __init__(self, log_folder, timezone):
|
||||||
|
self.folder = log_folder
|
||||||
|
self.tz = timezone
|
||||||
|
self.messages = []
|
||||||
|
|
||||||
|
def load_last(self):
|
||||||
|
files = [f for f in os.listdir(self.folder) if os.path.isfile(os.path.join(self.folder, f))]
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
if len(files) == 0:
|
||||||
|
self.messages = []
|
||||||
|
return
|
||||||
|
last = files[-1]
|
||||||
|
self.messages = []
|
||||||
|
with open(os.path.join(self.folder, last), 'r') as f:
|
||||||
|
self.messages.append(f.readline())
|
||||||
|
|
||||||
|
def get_filename(self):
|
||||||
|
files = [f for f in os.listdir(self.folder) if os.path.isfile(os.path.join(self.folder, f)) and 'dairy' in f]
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
if len(files) == 0:
|
||||||
|
return os.path.join(self.folder, 'diary-{0}.log'.format(today.strftime('%Y-%m-%d')))
|
||||||
|
last = files[-1]
|
||||||
|
return os.path.join(self.folder, last)
|
||||||
|
|
||||||
|
def start_new(self):
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
filename = os.path.join(self.folder, 'diary-{0}.log'.format(today.strftime('%Y-%m-%d')))
|
||||||
|
with open(filename, 'w') as f:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def start_log(self):
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
locale.setlocale(locale.LC_TIME, 'es_ES')
|
||||||
|
msg = 'Inicio un nuevo dia siendo las {0} del {1}'.format(today.strftime('%H:%M:%S'),
|
||||||
|
today.strftime('%d de %B de %Y'))
|
||||||
|
self.log(msg)
|
||||||
|
|
||||||
|
def stop_log(self):
|
||||||
|
now = datetime.datetime.now(tz=self.tz)
|
||||||
|
locale.setlocale(locale.LC_TIME, 'es_ES')
|
||||||
|
msg = 'Siendo las {0}, termino mis registros por hoy'.format(now.strftime('%H:%M:%S'))
|
||||||
|
self.log(msg)
|
||||||
|
msg = '--------'
|
||||||
|
self.log(msg)
|
||||||
|
|
||||||
|
def log_action(self, action):
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
locale.setlocale(locale.LC_TIME, 'es_ES')
|
||||||
|
msg = 'A las {0} del {1}, he realizado {2}'.format(today.strftime('%H:%M:%S'),
|
||||||
|
today.strftime('%d de %B de %Y'), action)
|
||||||
|
self.log(msg)
|
||||||
|
|
||||||
|
def log_not_action(self, action):
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
locale.setlocale(locale.LC_TIME, 'es_ES')
|
||||||
|
msg = 'A las {0} del {1}, no he podido realizar {2}'.format(today.strftime('%H:%M:%S'),
|
||||||
|
today.strftime('%d de %B de %Y'), action)
|
||||||
|
self.log(msg)
|
||||||
|
|
||||||
|
def log(self, message):
|
||||||
|
line = message.rstrip('.') + '.'
|
||||||
|
self.messages.append(line)
|
||||||
|
if len(self.messages) > 1000:
|
||||||
|
self.start_new()
|
||||||
|
self.load_last()
|
||||||
|
with open(self.get_filename(), 'a') as f:
|
||||||
|
f.write(line + "\n")
|
||||||
|
|
||||||
|
|
||||||
|
class Worker(Thread):
|
||||||
|
def __init__(self, params, configs):
|
||||||
|
super().__init__()
|
||||||
|
self.event = params['events']['stop']
|
||||||
|
self.queue = params['queues']['log']
|
||||||
|
self.wait = configs.get('supervisor.wait')
|
||||||
|
self.logger = Logger(params['folders']['log'], configs.get('timezone'))
|
||||||
|
self.logging = params['logging']
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logging.log('Starting', caller=type(self))
|
||||||
|
while not self.event.is_set():
|
||||||
|
self.logging.log('Looping status {0}'.format(not self.event.is_set()), caller=type(self))
|
||||||
|
try:
|
||||||
|
message = self.queue.get(timeout=self.wait)
|
||||||
|
self.logging.log('Logger received message', caller=type(self))
|
||||||
|
if 'is_start' in message and message['is_start']:
|
||||||
|
self.logger.start_log()
|
||||||
|
continue
|
||||||
|
if 'not' in message and message['not']:
|
||||||
|
self.logger.log_not_action(message['action'])
|
||||||
|
continue
|
||||||
|
self.logger.log_action(message['action'])
|
||||||
|
except queue.Empty:
|
||||||
|
pass
|
||||||
|
self.logger.stop_log()
|
||||||
|
self.logging.log('Exiting', caller=type(self))
|
||||||
|
return
|
43
common/helper/logging.py
Normal file
43
common/helper/logging.py
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
import os
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
|
||||||
|
class Logging:
|
||||||
|
"""
|
||||||
|
Clase que registra los eventos del flujo del programa en archivos
|
||||||
|
"""
|
||||||
|
def __init__(self, timezone, folder=None):
|
||||||
|
self.tz = timezone
|
||||||
|
self.folder = folder
|
||||||
|
self.filename = self.get_filename()
|
||||||
|
self.log('------')
|
||||||
|
|
||||||
|
def get_filename(self):
|
||||||
|
files = [f for f in os.listdir(self.folder) if os.path.isfile(os.path.join(self.folder, f)) and 'logging' in f]
|
||||||
|
if len(files) > 0:
|
||||||
|
filename = files[-1]
|
||||||
|
if os.path.getsize(os.path.join(self.folder, filename)) < 1024 * 1024:
|
||||||
|
return filename
|
||||||
|
today = datetime.datetime.now(tz=self.tz)
|
||||||
|
filename = 'logging-{0}.log'.format(today.strftime('%Y-%m-%d'))
|
||||||
|
base_name = filename
|
||||||
|
n = 1
|
||||||
|
while os.path.isfile(os.path.join(self.folder, filename)):
|
||||||
|
filename = '-'.join([
|
||||||
|
base_name,
|
||||||
|
str(n)
|
||||||
|
])
|
||||||
|
n += 1
|
||||||
|
return filename
|
||||||
|
|
||||||
|
def log(self, message, caller=None):
|
||||||
|
now = datetime.datetime.now(tz=self.tz)
|
||||||
|
if caller is None:
|
||||||
|
msg = '[{0}] {1}.'.format(now.strftime('%Y-%m-%d %H:%M:%S'), message.strip('.'))
|
||||||
|
else:
|
||||||
|
msg = '[{0}] ({2}) {1}.'.format(now.strftime('%Y-%m-%d %H:%M:%S'), message.strip('.'), caller)
|
||||||
|
print(msg)
|
||||||
|
if self.folder is not None:
|
||||||
|
filename = os.path.join(self.folder, self.filename)
|
||||||
|
with open(filename, 'a') as f:
|
||||||
|
f.write(msg + "\n")
|
7
config/email.json
Normal file
7
config/email.json
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
{
|
||||||
|
"imap_server": "imap.yandex.com",
|
||||||
|
"port": 993,
|
||||||
|
"username": "secretary@incoviba.cl",
|
||||||
|
"password": "quzshqzyfcnydevp",
|
||||||
|
"ssl": true
|
||||||
|
}
|
3
config/supervisor.json
Normal file
3
config/supervisor.json
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
{
|
||||||
|
"wait": 15
|
||||||
|
}
|
72
entry/email/inbox.py
Normal file
72
entry/email/inbox.py
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
import argparse
|
||||||
|
import imaplib
|
||||||
|
import email
|
||||||
|
|
||||||
|
|
||||||
|
# Email class that fetches emails from server by uid and can parse body for secretary, can also delete email by uid
|
||||||
|
class Email:
|
||||||
|
def __init__(self, uid):
|
||||||
|
self.uid = uid
|
||||||
|
self.message = ''
|
||||||
|
|
||||||
|
def get(self, imap):
|
||||||
|
status, raw_data = imap.uid('fetch', self.uid, '(RFC822)')
|
||||||
|
if status != 'OK':
|
||||||
|
raise Exception('Could not recover message {0}'.format(self.uid))
|
||||||
|
|
||||||
|
self.message = email.message_from_bytes(raw_data[0][1])
|
||||||
|
|
||||||
|
def delete(self, imap):
|
||||||
|
status, result = imap.uid('STORE', self.uid, '+FLAGS', '(\\Deleted)')
|
||||||
|
if status != 'OK':
|
||||||
|
raise Exception('Could not flag message {0}'.format(self.uid))
|
||||||
|
|
||||||
|
|
||||||
|
def connect(imap_url, port, username, password, ssl=False):
|
||||||
|
if ssl:
|
||||||
|
server = imaplib.IMAP4_SSL(imap_url, port=port)
|
||||||
|
else:
|
||||||
|
server = imaplib.IMAP4(imap_url, port=port)
|
||||||
|
server.login(username, password)
|
||||||
|
return server
|
||||||
|
|
||||||
|
|
||||||
|
def check_inbox(server):
|
||||||
|
status, msg = server.select('INBOX')
|
||||||
|
if status != 'OK':
|
||||||
|
return None
|
||||||
|
|
||||||
|
status, ids = server.uid('search', None, 'All')
|
||||||
|
if status != 'OK':
|
||||||
|
return None
|
||||||
|
|
||||||
|
ids = ids[0].decode().split()
|
||||||
|
emails = []
|
||||||
|
for mid in ids:
|
||||||
|
em = Email(mid)
|
||||||
|
em.get(server)
|
||||||
|
emails.append(em)
|
||||||
|
|
||||||
|
return emails
|
||||||
|
|
||||||
|
|
||||||
|
def main(args):
|
||||||
|
print('Connecting')
|
||||||
|
imap_server = connect(imap_url=args.url, port=args.port, username=args.username, password=args.password,
|
||||||
|
ssl=args.ssl)
|
||||||
|
print('Checking emails')
|
||||||
|
emails = check_inbox(imap_server)
|
||||||
|
print(len(emails))
|
||||||
|
[print(e) for e in emails]
|
||||||
|
imap_server.close()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument('-u', '--url')
|
||||||
|
parser.add_argument('-p', '--port')
|
||||||
|
parser.add_argument('-un', '--username')
|
||||||
|
parser.add_argument('-up', '--password')
|
||||||
|
parser.add_argument('--ssl')
|
||||||
|
_args = parser.parse_args()
|
||||||
|
main(_args)
|
51
entry/main.py
Normal file
51
entry/main.py
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
import argparse
|
||||||
|
import os
|
||||||
|
import pytz
|
||||||
|
from common.helper.logging import Logging
|
||||||
|
from setup.config import load_config
|
||||||
|
from src.supervisor import Supervisor
|
||||||
|
from src.bosses import Bosses
|
||||||
|
|
||||||
|
|
||||||
|
def main(args):
|
||||||
|
configs = load_config(args.config_folder)
|
||||||
|
configs.set('timezone', pytz.timezone('America/Santiago'))
|
||||||
|
params = {
|
||||||
|
'folders': {
|
||||||
|
'config': args.config_folder,
|
||||||
|
'log': args.log_folder,
|
||||||
|
'data': args.data_folder
|
||||||
|
},
|
||||||
|
'bosses': Bosses(args.data_folder),
|
||||||
|
'logging': Logging(configs.get('timezone'), args.log_folder)
|
||||||
|
}
|
||||||
|
|
||||||
|
params['logging'].log('Starting', caller='main')
|
||||||
|
|
||||||
|
supervisor = Supervisor(configs=configs, params=params)
|
||||||
|
|
||||||
|
supervisor.register_worker('common.helper.logger', 'Worker')
|
||||||
|
supervisor.add_queue('log')
|
||||||
|
supervisor.register_worker('src.email', 'RevisorEmailWorker')
|
||||||
|
supervisor.register_worker('src.email', 'EmailInterpreter')
|
||||||
|
supervisor.add_queue('emails')
|
||||||
|
supervisor.add_queue('questions')
|
||||||
|
supervisor.register_worker('src.action.questions', 'QuestionWorker')
|
||||||
|
supervisor.register_worker('src.action.reminder', 'RemindRegisterWorker')
|
||||||
|
supervisor.register_worker('src.action.reminder', 'ReminderWorker')
|
||||||
|
supervisor.add_queue('reminders')
|
||||||
|
supervisor.add_lock('reminders')
|
||||||
|
|
||||||
|
supervisor.start()
|
||||||
|
supervisor.join()
|
||||||
|
|
||||||
|
params['logging'].log('Waiting for Supervisor', caller='main')
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument('-c', '--config_folder', default=os.path.join(os.path.realpath('../'), 'config'))
|
||||||
|
parser.add_argument('-d', '--data_folder', default=os.path.join(os.path.realpath('..'), 'data'))
|
||||||
|
parser.add_argument('-l', '--log_folder', default=os.path.join(os.path.realpath('..'), 'logs'))
|
||||||
|
_args = parser.parse_args()
|
||||||
|
main(_args)
|
70
setup/config.py
Normal file
70
setup/config.py
Normal file
@ -0,0 +1,70 @@
|
|||||||
|
import os
|
||||||
|
import json
|
||||||
|
|
||||||
|
|
||||||
|
class ConfigFile:
|
||||||
|
def __init__(self, name):
|
||||||
|
self.name = name
|
||||||
|
self._data = []
|
||||||
|
|
||||||
|
def load(self, folder):
|
||||||
|
filename = os.path.join(folder, self.name + '.json')
|
||||||
|
with open(filename, 'r') as f:
|
||||||
|
self._data = json.load(f)
|
||||||
|
|
||||||
|
def get(self, name):
|
||||||
|
if '.' not in name and name in self._data:
|
||||||
|
return self._data[name]
|
||||||
|
subs = "']['".join(name.split('.'))
|
||||||
|
substr = "self._data['" + subs + "']"
|
||||||
|
print(substr)
|
||||||
|
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
def __init__(self, folder):
|
||||||
|
self.folder = folder
|
||||||
|
self._configs = {}
|
||||||
|
|
||||||
|
def load(self):
|
||||||
|
files = [f for f in os.listdir(self.folder) if os.path.isfile(os.path.join(self.folder, f)) and '.json' in f]
|
||||||
|
configs = {}
|
||||||
|
for f in files:
|
||||||
|
name = f.split('.')[0]
|
||||||
|
filename = os.path.join(self.folder, f)
|
||||||
|
with open(filename, 'r') as fh:
|
||||||
|
configs[name] = json.load(fh)
|
||||||
|
self._configs = configs
|
||||||
|
|
||||||
|
def get(self, name):
|
||||||
|
if '.' not in name and name in self._configs:
|
||||||
|
return self._configs[name]
|
||||||
|
subs = "']['".join(name.split('.'))
|
||||||
|
substr = "self._configs['" + subs + "']"
|
||||||
|
try:
|
||||||
|
sub = eval(substr)
|
||||||
|
return sub
|
||||||
|
except:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def get_all(self, name_list):
|
||||||
|
output = {}
|
||||||
|
for name in name_list:
|
||||||
|
output[name] = self.get(name)
|
||||||
|
return output
|
||||||
|
|
||||||
|
def set(self, name, value):
|
||||||
|
if '.' not in name:
|
||||||
|
self._configs[name] = value
|
||||||
|
return
|
||||||
|
subs = "']['".join(name.split('.'))
|
||||||
|
substr = "self._configs['" + subs + "'] = {0}".format(value)
|
||||||
|
try:
|
||||||
|
eval(substr)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def load_config(folder):
|
||||||
|
config = Config(folder)
|
||||||
|
config.load()
|
||||||
|
return config
|
44
src/action/questions.py
Normal file
44
src/action/questions.py
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
import os
|
||||||
|
from threading import Thread
|
||||||
|
import queue
|
||||||
|
import json
|
||||||
|
|
||||||
|
|
||||||
|
class QuestionWorker(Thread):
|
||||||
|
def __init__(self, params, configs):
|
||||||
|
super(QuestionWorker, self).__init__()
|
||||||
|
self.queue = params['queues']['questions']
|
||||||
|
self.logging = params['logging']
|
||||||
|
self.logger = params['queues']['log']
|
||||||
|
self.event = params['events']['stop']
|
||||||
|
self.wait = configs.get('supervisor.wait')
|
||||||
|
self.folder = params['folders']['data']
|
||||||
|
|
||||||
|
def question(self, text):
|
||||||
|
filename = os.path.join(self.folder, 'questions.json')
|
||||||
|
with open(filename, 'w+', encoding='utf8') as f:
|
||||||
|
try:
|
||||||
|
data = json.load(f)
|
||||||
|
except json.decoder.JSONDecodeError:
|
||||||
|
data = []
|
||||||
|
if text in data:
|
||||||
|
return
|
||||||
|
data.append(text)
|
||||||
|
json.dump(data, f, ensure_ascii=False, indent=4)
|
||||||
|
self.logging.log(text, type(self))
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logging.log('Starting', type(self))
|
||||||
|
self.logger.put({'action': type(self)})
|
||||||
|
while not self.event.is_set():
|
||||||
|
try:
|
||||||
|
question = self.queue.get(timeout=self.wait)
|
||||||
|
if question.type == 'email':
|
||||||
|
self.question(
|
||||||
|
'¿Que hago con este correo de {0} y el texto dice {1} en la fecha {2}?'.
|
||||||
|
format(question.sender, question.text, question.datetime.strftime('%d-%m-%Y %H:%M:%S'))
|
||||||
|
)
|
||||||
|
except queue.Empty:
|
||||||
|
pass
|
||||||
|
self.logging.log('Exiting', type(self))
|
||||||
|
return
|
112
src/action/reminder.py
Normal file
112
src/action/reminder.py
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
import datetime
|
||||||
|
import queue
|
||||||
|
from threading import Thread
|
||||||
|
import os
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
|
||||||
|
|
||||||
|
class Reminder:
|
||||||
|
def __init__(self, timezone):
|
||||||
|
self.datetime = 0
|
||||||
|
self.message = ''
|
||||||
|
self.recursive = False
|
||||||
|
self.last = 0
|
||||||
|
self.repeats = 0
|
||||||
|
self.tz = timezone
|
||||||
|
|
||||||
|
def should_remind(self):
|
||||||
|
now = datetime.datetime.now(tz=self.tz)
|
||||||
|
event = self.get_remind_time()
|
||||||
|
dif = event - now
|
||||||
|
if dif.days * 24 * 60 * 60 + dif.seconds == 0:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
def get_remind_time(self):
|
||||||
|
if self.recursive:
|
||||||
|
event = self.last + self.datetime
|
||||||
|
return event
|
||||||
|
return self.datetime
|
||||||
|
|
||||||
|
def remind(self):
|
||||||
|
# Do remind
|
||||||
|
if self.recursive:
|
||||||
|
self.last = datetime.datetime.now(tz=self.tz)
|
||||||
|
self.repeats -= 1
|
||||||
|
|
||||||
|
|
||||||
|
class RemindRegisterWorker(Thread):
|
||||||
|
def __init__(self, params, configs):
|
||||||
|
super().__init__()
|
||||||
|
self.event = params['events']['stop']
|
||||||
|
self.queue = params['queues']['reminders']
|
||||||
|
self.lock = params['locks']['reminders']
|
||||||
|
self.folder = params['folders']['data']
|
||||||
|
self.logging = params['logging']
|
||||||
|
self.wait = configs.get('supervisor.wait')
|
||||||
|
|
||||||
|
def add_reminder(self, reminder):
|
||||||
|
self.lock.acquire()
|
||||||
|
filename = os.path.join(self.folder, 'reminders.json')
|
||||||
|
with open(filename, 'w+') as f:
|
||||||
|
reminders = json.load(f)
|
||||||
|
reminders.append(reminder)
|
||||||
|
json.dump(reminders, f)
|
||||||
|
self.lock.release()
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logging.log('Starting', caller=type(self))
|
||||||
|
while not self.event.is_set():
|
||||||
|
self.logging.log('Looping status {0}'.format(not self.event.is_set()),
|
||||||
|
caller=type(self))
|
||||||
|
try:
|
||||||
|
reminder = self.queue.get(timeout=self.wait)
|
||||||
|
self.add_reminder(reminder)
|
||||||
|
except queue.Empty:
|
||||||
|
pass
|
||||||
|
self.logging.log('Exiting', caller=type(self))
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
class ReminderWorker(Thread):
|
||||||
|
def __init__(self, params, configs):
|
||||||
|
super().__init__()
|
||||||
|
self.event = params['events']['stop']
|
||||||
|
self.folder = params['folders']['data']
|
||||||
|
self.lock = params['locks']['reminders']
|
||||||
|
self.logging = params['logging']
|
||||||
|
self.wait = configs.get('supervisor.wait')
|
||||||
|
self.tz = configs.get('timezone')
|
||||||
|
|
||||||
|
def check_reminders(self):
|
||||||
|
self.lock.acquire()
|
||||||
|
filename = os.path.join(self.folder, 'reminders.json')
|
||||||
|
if not os.path.isfile(filename):
|
||||||
|
self.lock.release()
|
||||||
|
return
|
||||||
|
with open(filename, 'w+') as f:
|
||||||
|
reminders = json.load(f)
|
||||||
|
new_reminders = []
|
||||||
|
for r in reminders:
|
||||||
|
reminder = Reminder(self.tz)
|
||||||
|
for k, val in r.items():
|
||||||
|
setattr(reminder, k, val)
|
||||||
|
if reminder.should_remind():
|
||||||
|
reminder.remind()
|
||||||
|
if reminder.recursive and reminder.repeats > 0:
|
||||||
|
new_reminders.append(reminder)
|
||||||
|
else:
|
||||||
|
new_reminders.append(reminder)
|
||||||
|
json.dump(new_reminders, f)
|
||||||
|
self.lock.release()
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logging.log('Starting', caller=type(self))
|
||||||
|
while not self.event.is_set():
|
||||||
|
self.logging.log('Looping status {0}'.format(not self.event.is_set()),
|
||||||
|
caller=type(self))
|
||||||
|
self.check_reminders()
|
||||||
|
time.sleep(self.wait)
|
||||||
|
self.logging.log('Exiting', caller=type(self))
|
||||||
|
return
|
40
src/address_book.py
Normal file
40
src/address_book.py
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
import os
|
||||||
|
import json
|
||||||
|
|
||||||
|
|
||||||
|
class Contact:
|
||||||
|
def __init__(self):
|
||||||
|
self.names = ''
|
||||||
|
self.last_name = ''
|
||||||
|
self.maiden_name = ''
|
||||||
|
self.emails = []
|
||||||
|
self.phones = []
|
||||||
|
|
||||||
|
def full_name(self):
|
||||||
|
return ' '.join([self.names, self.last_name, self.maiden_name])
|
||||||
|
|
||||||
|
|
||||||
|
class AddressBook:
|
||||||
|
def __init__(self, data_folder):
|
||||||
|
filename = os.path.join(data_folder, 'address_book.json')
|
||||||
|
with open(filename, 'r') as f:
|
||||||
|
data = json.load(f)
|
||||||
|
|
||||||
|
self.contacts = []
|
||||||
|
for c in data:
|
||||||
|
contact = Contact()
|
||||||
|
for k, val in c.items():
|
||||||
|
setattr(contact, k, val)
|
||||||
|
self.contacts.append(contact)
|
||||||
|
|
||||||
|
def find(self, name):
|
||||||
|
for c in self.contacts:
|
||||||
|
if c.full_name() == name or c.names == name or c.last_name == name:
|
||||||
|
return c
|
||||||
|
return None
|
||||||
|
|
||||||
|
def find_id(self, name):
|
||||||
|
for (k, c) in enumerate(self.contacts):
|
||||||
|
if c.full_name() == name or c.names == name or c.last_name == name:
|
||||||
|
return k
|
||||||
|
return -1
|
37
src/bosses.py
Normal file
37
src/bosses.py
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
import os
|
||||||
|
import json
|
||||||
|
from src.address_book import AddressBook
|
||||||
|
|
||||||
|
|
||||||
|
class Boss:
|
||||||
|
def __init__(self):
|
||||||
|
self.full_name = ''
|
||||||
|
self.aliases = []
|
||||||
|
self.contact = None
|
||||||
|
|
||||||
|
|
||||||
|
class Bosses:
|
||||||
|
def __init__(self, data_folder):
|
||||||
|
filename = os.path.join(data_folder, 'bosses.json')
|
||||||
|
with open(filename, 'r') as f:
|
||||||
|
data = json.load(f)
|
||||||
|
self.bosses = []
|
||||||
|
addrs = AddressBook(data_folder)
|
||||||
|
for b in data:
|
||||||
|
boss = Boss()
|
||||||
|
for k, val in b.items():
|
||||||
|
setattr(boss, k, val)
|
||||||
|
boss.contact = addrs.find(boss.full_name)
|
||||||
|
self.bosses.append(boss)
|
||||||
|
|
||||||
|
def is_boss(self, name):
|
||||||
|
for boss in self.bosses:
|
||||||
|
if boss.full_name in name:
|
||||||
|
return True
|
||||||
|
for m in boss.contact.emails:
|
||||||
|
if m in name:
|
||||||
|
return True
|
||||||
|
for a in boss.aliases:
|
||||||
|
if a in name:
|
||||||
|
return True
|
||||||
|
return False
|
8
src/communication/message.py
Normal file
8
src/communication/message.py
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
class Message:
|
||||||
|
def __init__(self, mtype, sender, datetime, text, original):
|
||||||
|
self.type = mtype
|
||||||
|
self.sender = sender
|
||||||
|
self.datetime = datetime
|
||||||
|
self.text = text
|
||||||
|
self.checked = False
|
||||||
|
self.original = original
|
4
src/email/__init__.py
Normal file
4
src/email/__init__.py
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
from .email_interpreter import EmailInterpreter
|
||||||
|
from .revisor_worker import RevisorEmailWorker
|
||||||
|
from .obtenedor import Obtenedor
|
||||||
|
from .validador import Validador
|
10
src/email/email_interpreter.py
Normal file
10
src/email/email_interpreter.py
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
from threading import Thread
|
||||||
|
|
||||||
|
|
||||||
|
class EmailInterpreter(Thread):
|
||||||
|
def __init__(self, configs, params):
|
||||||
|
super().__init__()
|
||||||
|
self.logging = params['logging']
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logging.log('Starting', type(self))
|
148
src/email/main.py
Normal file
148
src/email/main.py
Normal file
@ -0,0 +1,148 @@
|
|||||||
|
from multiprocessing import Process
|
||||||
|
from queue import Queue
|
||||||
|
from threading import Thread, Event, Lock
|
||||||
|
import argparse
|
||||||
|
import os
|
||||||
|
from common.helper.logging import Logging
|
||||||
|
from setup.config import load_config
|
||||||
|
import pytz
|
||||||
|
from src.bosses import Bosses
|
||||||
|
import keyboard
|
||||||
|
import time
|
||||||
|
|
||||||
|
|
||||||
|
class Email(Process):
|
||||||
|
def __init__(self, configs, params, setup):
|
||||||
|
super(Email, self).__init__()
|
||||||
|
|
||||||
|
self.configs = configs
|
||||||
|
self.params = params
|
||||||
|
self.registry = {}
|
||||||
|
self.workers = []
|
||||||
|
self.working = []
|
||||||
|
self.worker_status = []
|
||||||
|
|
||||||
|
self.add_event('stop')
|
||||||
|
|
||||||
|
def setup(self, data):
|
||||||
|
for (m, n) in data['workers']:
|
||||||
|
self.register_worker(m, n)
|
||||||
|
for q in data['queues']:
|
||||||
|
self.add_queue(q)
|
||||||
|
for e in data['events']:
|
||||||
|
self.add_event(e)
|
||||||
|
for l in data['locks']:
|
||||||
|
self.add_lock(l)
|
||||||
|
|
||||||
|
def register_worker(self, module, name):
|
||||||
|
if module not in self.registry:
|
||||||
|
self.registry[module] = []
|
||||||
|
self.registry[module].append(name)
|
||||||
|
|
||||||
|
def add_worker(self, worker):
|
||||||
|
self.workers.append(worker)
|
||||||
|
|
||||||
|
def start_worker(self, module, name):
|
||||||
|
worker = getattr(module, name)
|
||||||
|
self.add_worker(worker)
|
||||||
|
self.working.append((module, name))
|
||||||
|
worker.start()
|
||||||
|
self.worker_status.append(True)
|
||||||
|
|
||||||
|
def start_workers(self):
|
||||||
|
for module, workers in self.registry:
|
||||||
|
for name in workers:
|
||||||
|
self.start_worker(module, name)
|
||||||
|
|
||||||
|
def check_workers(self):
|
||||||
|
stopped = 0
|
||||||
|
for (k, w) in enumerate(self.workers):
|
||||||
|
if not self.worker_status[k]:
|
||||||
|
stopped += 1
|
||||||
|
continue
|
||||||
|
if not w.is_alive():
|
||||||
|
self.params['logging'].log('Worker {0} stopped'.format(type(w)))
|
||||||
|
self.params['queues']['log'].put({'not': True, 'action': type(w)})
|
||||||
|
stopped += 1
|
||||||
|
self.worker_status[k] = False
|
||||||
|
(m, n) = self.working[k]
|
||||||
|
# Restart worker
|
||||||
|
self.start_worker(m, n)
|
||||||
|
if stopped == len(self.workers):
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
def join_workers(self):
|
||||||
|
[w.join(self.configs.get('supervisor.wait')) for w in self.workers if w.is_alive()]
|
||||||
|
|
||||||
|
def add_queue(self, name):
|
||||||
|
if 'queues' not in self.params or self.params['queues'] is None:
|
||||||
|
self.params['queues'] = {}
|
||||||
|
self.params['queues'][name] = Queue()
|
||||||
|
|
||||||
|
def add_event(self, name):
|
||||||
|
if 'events' not in self.params or self.params['events'] is None:
|
||||||
|
self.params['events'] = {}
|
||||||
|
self.params['events'][name] = Event()
|
||||||
|
|
||||||
|
def add_lock(self, name):
|
||||||
|
if 'locks' not in self.params or self.params['locks'] is None:
|
||||||
|
self.params['locks'] = {}
|
||||||
|
self.params['locks'][name] = Lock()
|
||||||
|
|
||||||
|
def run(self) -> None:
|
||||||
|
self.start_workers()
|
||||||
|
self.add_worker(Thread(target=exit_thread, args=(self.params['events']['stop'], self.params['logging'])))
|
||||||
|
|
||||||
|
while not self.params['events']['stop'].is_set():
|
||||||
|
if not self.check_workers():
|
||||||
|
break
|
||||||
|
time.sleep(self.configs.get('supervisor.wait'))
|
||||||
|
self.join_workers()
|
||||||
|
|
||||||
|
|
||||||
|
def exit_thread(stop, logger):
|
||||||
|
logger.log('Starting exit thread', caller='exit_thread')
|
||||||
|
keyboard.wait('Esc')
|
||||||
|
logger.log('Escape pressed', caller='exit_thread')
|
||||||
|
stop.set()
|
||||||
|
logger.log('Exit signal sent', caller='exit_thread')
|
||||||
|
|
||||||
|
|
||||||
|
def main(args):
|
||||||
|
configs = load_config(args.config_folder)
|
||||||
|
configs.set('timezone', pytz.timezone('America/Santiago'))
|
||||||
|
params = {
|
||||||
|
'folders': {
|
||||||
|
'config': args.config_folder,
|
||||||
|
'log': args.log_folder,
|
||||||
|
'data': args.data_folder
|
||||||
|
},
|
||||||
|
'bosses': Bosses(args.data_folder),
|
||||||
|
'logging': Logging(configs.get('timezone'), args.log_folder)
|
||||||
|
}
|
||||||
|
|
||||||
|
setup = {
|
||||||
|
'workers': [
|
||||||
|
('common.helper.logger', 'Worker'),
|
||||||
|
('src.email', 'Obtenedor'),
|
||||||
|
('src.email', 'Validador')
|
||||||
|
],
|
||||||
|
'queues': ['log', 'emails', 'valid', 'invalid'],
|
||||||
|
'events': [],
|
||||||
|
'locks': []
|
||||||
|
}
|
||||||
|
|
||||||
|
email = Email(configs, params, setup)
|
||||||
|
|
||||||
|
email.start()
|
||||||
|
email.join()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument('-c', '--config_folder', default=os.path.join(os.path.realpath('../..'), 'config'))
|
||||||
|
parser.add_argument('-d', '--data_folder', default=os.path.join(os.path.realpath('../..'), 'data'))
|
||||||
|
parser.add_argument('-l', '--log_folder', default=os.path.join(os.path.realpath('../..'), 'logs'))
|
||||||
|
_args = parser.parse_args()
|
||||||
|
main(_args)
|
78
src/email/obtenedor.py
Normal file
78
src/email/obtenedor.py
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
from src.worker import Worker
|
||||||
|
from types import SimpleNamespace
|
||||||
|
from entry.email.inbox import connect, check_inbox
|
||||||
|
import re
|
||||||
|
from bs4 import BeautifulSoup
|
||||||
|
import email.utils
|
||||||
|
from src.communication.message import Message
|
||||||
|
import time
|
||||||
|
|
||||||
|
|
||||||
|
class Obtenedor(Worker):
|
||||||
|
"""
|
||||||
|
Trabajador que obtiene la lista de correos
|
||||||
|
"""
|
||||||
|
def __init__(self, configs, params):
|
||||||
|
super(Obtenedor, self).__init__(configs, params)
|
||||||
|
|
||||||
|
self.url = configs.get('email.server')
|
||||||
|
self.port = configs.get('email.port')
|
||||||
|
user = {'user': '', 'password': ''}
|
||||||
|
self.user = SimpleNamespace(**user)
|
||||||
|
self.user.name = configs.get('email.user.name')
|
||||||
|
self.user.password = configs.get('email.user.password')
|
||||||
|
self.ssl = configs.get('email.ssl')
|
||||||
|
|
||||||
|
self.revisados = []
|
||||||
|
|
||||||
|
self.queue = params['queues']['emails']
|
||||||
|
self.frec = configs.get('supervisor.wait')
|
||||||
|
|
||||||
|
def is_revisado(self, uid):
|
||||||
|
return uid in self.revisados
|
||||||
|
|
||||||
|
def add_revisado(self, uid):
|
||||||
|
if self.is_revisado(uid):
|
||||||
|
return
|
||||||
|
self.revisados.append(uid)
|
||||||
|
|
||||||
|
def build_message(self, email_part):
|
||||||
|
output = []
|
||||||
|
if email_part.is_multipart():
|
||||||
|
for part in email_part.get_payload():
|
||||||
|
output.append(self.build_message(part))
|
||||||
|
else:
|
||||||
|
html = email_part.get_payload(decode=True)
|
||||||
|
bs = BeautifulSoup(html, 'html.parser')
|
||||||
|
if bs.body:
|
||||||
|
html = bs.body.get_text()
|
||||||
|
else:
|
||||||
|
html = bs.get_text()
|
||||||
|
html = re.sub(' +', ' ', re.sub("\n+", ' ', html)).strip(' ')
|
||||||
|
output.append(html)
|
||||||
|
return output
|
||||||
|
|
||||||
|
def run(self) -> None:
|
||||||
|
self.logger.log('Starting', type(self))
|
||||||
|
self.diary.put({'action': 'Inicio de jornada de Obtenedor'})
|
||||||
|
while not self.stop.is_set():
|
||||||
|
e = 0
|
||||||
|
with connect(self.url, self.port, self.user.name, self.user.password, self.ssl) as imap:
|
||||||
|
self.logger.log('Getting emails', type(self))
|
||||||
|
emails = check_inbox(imap)
|
||||||
|
if emails is None:
|
||||||
|
continue
|
||||||
|
for em in emails:
|
||||||
|
if self.is_revisado(em.uid):
|
||||||
|
continue
|
||||||
|
sender = em.message['from']
|
||||||
|
text = ' '.join([em.message['subject'] + '.'] + self.build_message(em.message))
|
||||||
|
msg = Message('email', text=text, original=em, sender=sender,
|
||||||
|
datetime=email.utils.parsedate_to_datetime(em.message['Date']))
|
||||||
|
self.queue.put(msg)
|
||||||
|
self.add_revisado(em.uid)
|
||||||
|
e += 1
|
||||||
|
self.diary.put({'action': 'Obtenidos {0} correos nuevos'.format(e)})
|
||||||
|
time.sleep(self.frec)
|
||||||
|
self.logger.log('Exiting', type(self))
|
||||||
|
self.diary.put({'action': 'Terminando el turno de Obtenedor'})
|
90
src/email/revisor_worker.py
Normal file
90
src/email/revisor_worker.py
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
from threading import Thread
|
||||||
|
import re
|
||||||
|
import time
|
||||||
|
import email.utils
|
||||||
|
from bs4 import BeautifulSoup
|
||||||
|
from entry.email.inbox import connect, check_inbox
|
||||||
|
from src.communication.message import Message
|
||||||
|
|
||||||
|
|
||||||
|
class RevisorEmailWorker(Thread):
|
||||||
|
def __init__(self, configs, params):
|
||||||
|
super().__init__()
|
||||||
|
self._url = configs.get('email.imap_server')
|
||||||
|
self._port = configs.get('email.port')
|
||||||
|
self._username = configs.get('email.username')
|
||||||
|
self._password = configs.get('email.password')
|
||||||
|
self._ssl = configs.get('email.ssl')
|
||||||
|
self.queue = params['queues']['emails']
|
||||||
|
self.questions = params['queues']['questions']
|
||||||
|
self.event = params['events']['stop']
|
||||||
|
self._wait = configs.get('supervisor.wait')
|
||||||
|
self._bosses = params['bosses']
|
||||||
|
self._logger = params['queues']['log']
|
||||||
|
self.logging = params['logging']
|
||||||
|
self.revisados = []
|
||||||
|
|
||||||
|
def revisado(self, uid):
|
||||||
|
if not self.check_revisado(uid):
|
||||||
|
self.revisados.append(uid)
|
||||||
|
|
||||||
|
def check_revisado(self, uid):
|
||||||
|
if uid in self.revisados:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logging.log('Starting', type(self))
|
||||||
|
self._logger.put({'action': 'Inicio jornada trabajador Revisor Email'})
|
||||||
|
while not self.event.is_set():
|
||||||
|
self.logging.log('Looping status {0}'.format(not self.event.is_set()), type(self))
|
||||||
|
self.logging.log('Connecting to Email Server', type(self))
|
||||||
|
imap = connect(imap_url=self._url, port=self._port, username=self._username, password=self._password,
|
||||||
|
ssl=self._ssl)
|
||||||
|
self.logging.log('Getting emails', type(self))
|
||||||
|
emails = check_inbox(imap)
|
||||||
|
if emails is not None:
|
||||||
|
c = 0
|
||||||
|
p = 0
|
||||||
|
for em in emails:
|
||||||
|
if self.check_revisado(em.uid):
|
||||||
|
continue
|
||||||
|
sender = em.message['from']
|
||||||
|
text = ' '.join([em.message['subject'] + '.'] + self.build_message(em.message))
|
||||||
|
msg = Message('email', text=text, original=em, sender=sender,
|
||||||
|
datetime=email.utils.parsedate_to_datetime(em.message['Date']))
|
||||||
|
if not self._bosses.is_boss(sender):
|
||||||
|
self.logging.log('Sender {0} is not a boss'.format(sender), type(self))
|
||||||
|
self.revisado(em.uid)
|
||||||
|
self.questions.put(msg)
|
||||||
|
p += 1
|
||||||
|
continue
|
||||||
|
self.queue.put(msg)
|
||||||
|
self.revisado(em.uid)
|
||||||
|
c += 1
|
||||||
|
self.logging.log('{0} emails checked'.format(c), type(self))
|
||||||
|
if c > 0:
|
||||||
|
self._logger.put({'action': 'Revise {0} nuevos correos'.format(c)})
|
||||||
|
self.logging.log('{0} emails pending'.format(p), type(self))
|
||||||
|
if p > 0:
|
||||||
|
self._logger.put({'action': 'Tengo dudas en {0} correos'.format(p)})
|
||||||
|
imap.close()
|
||||||
|
time.sleep(self._wait)
|
||||||
|
self.logging.log('Exiting', type(self))
|
||||||
|
return
|
||||||
|
|
||||||
|
def build_message(self, email_part):
|
||||||
|
output = []
|
||||||
|
if email_part.is_multipart():
|
||||||
|
for part in email_part.get_payload():
|
||||||
|
output.append(self.build_message(part))
|
||||||
|
else:
|
||||||
|
html = email_part.get_payload(decode=True)
|
||||||
|
bs = BeautifulSoup(html, 'html.parser')
|
||||||
|
if bs.body:
|
||||||
|
html = bs.body.get_text()
|
||||||
|
else:
|
||||||
|
html = bs.get_text()
|
||||||
|
html = re.sub(' +', ' ', re.sub("\n+", ' ', html)).strip(' ')
|
||||||
|
output.append(html)
|
||||||
|
return output
|
25
src/email/validador.py
Normal file
25
src/email/validador.py
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
from src.worker import Worker
|
||||||
|
|
||||||
|
|
||||||
|
class Validador(Worker):
|
||||||
|
def __init__(self, configs, params):
|
||||||
|
super(Validador, self).__init__(configs, params)
|
||||||
|
|
||||||
|
self.emails = params['queues']['emails']
|
||||||
|
self.validos = params['queues']['valid']
|
||||||
|
self.invalidos = params['queues']['invalid']
|
||||||
|
self.bosses = params['bosses']
|
||||||
|
|
||||||
|
self.frec = configs.get('supervisor.wait')
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.logger.log('Starting', type(self))
|
||||||
|
self.diary.put({'action': 'Inicio de jornada de Validador'})
|
||||||
|
while not self.stop.is_set():
|
||||||
|
em = self.emails.get(timeout=self.frec)
|
||||||
|
if not self.bosses.is_boss(em.sender):
|
||||||
|
self.invalidos.put(em)
|
||||||
|
continue
|
||||||
|
self.validos.put(em)
|
||||||
|
self.logger.log('Exiting', type(self))
|
||||||
|
self.diary.put({'action': 'Terminando la jornada de Validador'})
|
93
src/supervisor.py
Normal file
93
src/supervisor.py
Normal file
@ -0,0 +1,93 @@
|
|||||||
|
from threading import Thread, Event, Lock
|
||||||
|
from queue import Queue
|
||||||
|
import keyboard
|
||||||
|
import time
|
||||||
|
import importlib
|
||||||
|
|
||||||
|
|
||||||
|
class Supervisor(Thread):
|
||||||
|
def __init__(self, configs, params):
|
||||||
|
super().__init__()
|
||||||
|
self._configs = configs
|
||||||
|
self.params = params
|
||||||
|
self._workers = []
|
||||||
|
self._status = []
|
||||||
|
self._registry = {}
|
||||||
|
self.add_event('stop')
|
||||||
|
|
||||||
|
def add_worker(self, worker):
|
||||||
|
self._workers.append(worker)
|
||||||
|
|
||||||
|
def start_workers(self):
|
||||||
|
[w.start() for w in self._workers]
|
||||||
|
[self._status.append(True) for w in self._workers]
|
||||||
|
|
||||||
|
def check_workers(self):
|
||||||
|
stopped = 0
|
||||||
|
for (k, w) in enumerate(self._workers):
|
||||||
|
if not self._status[k]:
|
||||||
|
stopped += 1
|
||||||
|
continue
|
||||||
|
if not w.is_alive():
|
||||||
|
self.params['logging'].log('Worker {0} stopped'.format(type(w)))
|
||||||
|
self.params['queues']['log'].put({'not': True, 'action': type(w)})
|
||||||
|
stopped += 1
|
||||||
|
self._status[k] = False
|
||||||
|
if stopped == len(self._workers):
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
def join_workers(self):
|
||||||
|
[w.join(self._configs.get('supervisor.wait')) for w in self._workers]
|
||||||
|
|
||||||
|
def add_queue(self, name):
|
||||||
|
if 'queues' not in self.params or self.params['queues'] is None:
|
||||||
|
self.params['queues'] = {}
|
||||||
|
self.params['queues'][name] = Queue()
|
||||||
|
|
||||||
|
def add_event(self, name):
|
||||||
|
if 'events' not in self.params or self.params['events'] is None:
|
||||||
|
self.params['events'] = {}
|
||||||
|
self.params['events'][name] = Event()
|
||||||
|
|
||||||
|
def add_lock(self, name):
|
||||||
|
if 'locks' not in self.params or self.params['locks'] is None:
|
||||||
|
self.params['locks'] = {}
|
||||||
|
self.params['locks'][name] = Lock()
|
||||||
|
|
||||||
|
def register_worker(self, module, name):
|
||||||
|
if module not in self._registry:
|
||||||
|
self._registry[module] = []
|
||||||
|
self._registry[module].append(name)
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
for (module_name, ws) in self._registry.items():
|
||||||
|
module = importlib.import_module(module_name)
|
||||||
|
for w in ws:
|
||||||
|
worker = getattr(module, w)
|
||||||
|
self.add_worker(worker(configs=self._configs, params=self.params))
|
||||||
|
|
||||||
|
self.add_worker(Thread(target=exit_thread, args=(self.params['events']['stop'], self.params['logging'])))
|
||||||
|
|
||||||
|
self.params['queues']['log'].put({'is_start': True})
|
||||||
|
self.start_workers()
|
||||||
|
while not self.params['events']['stop'].is_set():
|
||||||
|
self.params['logging'].log('Looping main status {0}'.format(not self.params['events']['stop'].is_set()),
|
||||||
|
caller=type(self))
|
||||||
|
if not self.check_workers():
|
||||||
|
break
|
||||||
|
self.params['logging'].log('Waiting {0} secs'.format(self._configs.get('supervisor.wait')),
|
||||||
|
caller=type(self))
|
||||||
|
time.sleep(self._configs.get('supervisor.wait'))
|
||||||
|
self.params['logging'].log('Exiting', caller=type(self))
|
||||||
|
self.join_workers()
|
||||||
|
self.params['logging'].log('Waiting for Workers', caller=type(self))
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
def exit_thread(event, logging):
|
||||||
|
logging.log('Starting exit thread', caller='exit_thread')
|
||||||
|
keyboard.wait('Esc')
|
||||||
|
logging.log('Escape pressed', caller='exit_thread')
|
||||||
|
event.set()
|
||||||
|
logging.log('Exit signal sent', caller='exit_thread')
|
9
src/worker.py
Normal file
9
src/worker.py
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
from threading import Thread
|
||||||
|
|
||||||
|
|
||||||
|
class Worker(Thread):
|
||||||
|
def __init__(self, configs, params):
|
||||||
|
super(Worker, self).__init__()
|
||||||
|
self.stop = params['events']['stop']
|
||||||
|
self.diary = params['queues']['log']
|
||||||
|
self.logger = params['logging']
|
Reference in New Issue
Block a user