def __init__(self) -> None:
super().__init__()
self._initialize_directories()
self._initialize_database()
self.logger = log_manager.get_logger('file_manager')
self.scheduler_running = False
def __init__(self) -> None:
super().__init__()
self.file_manager = FileManager()
self.scheduled_deliveries_hour = None
self.handler_running = False
self.scheduler_running = False
self.logger = log_manager.get_logger('session_manager')
def __init__(self) -> None:
super().__init__()
# Configure logging
log_manager.setup_logging()
self.logger = log_manager.get_logger('session_manager')
# Initialize session variables
self.logger.info(utils.whos_calling('Fetching Firefox session...'))
self.previous_session = self._fetch_previous_session()
self.active_browser_connection = False
self.active_phone_connection = False
self.cookies = {}
self.driver = None
# Try reuse previous persisted session
if 'session' in self.previous_session:
self.session = self.previous_session['session']
self.session_id = self.session['session_id']
self.executor_url = self.session['executor_url']
self.logger.info('Fetching driver session with ID: %s' %
self.session_id)
self.driver = SessionManager.get_existing_driver_session()
else:
self._create_new_driver_session()
self.logger.debug('Using session with ID: %s' % self.driver.session_id)
"""
Try load webpage - if this fails, then session is invalid
* If previous session exists, reuse session
* If previous session does not exist, create a new session
"""
self.logger.info('Starting WhatsApp web')
try:
self.driver.get('https://web.whatsapp.com/')
self.monitor_connection()
# self._load_cookies()
# self.driver.get('https://web.whatsapp.com/')
except (WebDriverException, SessionNotCreatedException,
ConnectionRefusedError, URLError, NewConnectionError,
MaxRetryError):
self.logger.exception('Connection refused', exc_info=False)
self._create_new_driver_session()
self.monitor_connection()
finally:
uptime_manager.process_up(self)
# Start API & service monitoring
self.start_api()
}
apm = ElasticAPM(app)
handler = LoggingHandler(client=apm.client)
handler.setLevel(logging.WARN)
app.logger.addHandler(handler)
# Example usage:
# app.logger.error('Failed to send: Invalid number',
# exc_info=True,
# extra={
# 'uid': uid
# })
logger = log_manager.get_logger('api_manager')
@app.route("/ping")
def ping():
return 'healthy', 200
@app.route("/health")
def health_check():
if not check_auth():
return 'unauthorized', 400
else:
pass
logger.info('Handling /health request')
from celery import group
from celery.exceptions import SoftTimeLimitExceeded
from celery.exceptions import MaxRetriesExceededError
from elasticapm import Client
from task_queue.celery_config import *
client = Client({
'SERVICE_NAME': os.environ['ELASTIC_APM_SERVICE_NAME'],
'SERVER_URL': os.environ['ELASTIC_APM_SERVER_URL']
})
file_manager = FileManager()
alert_manager = AlertManager()
logger = log_manager.get_logger('session_manager')
def purge_tasks():
"""
Clear all tasks from queues
:returns number of tasks purged
"""
return app.control.purge()
def task_in_queue(check_id, queue_id):
"""
Inspect queue to check for ID: reserved, active, or scheduled
:returns true if task in queue
"""