# Path: pushkin/context.py
# PERSIST_LOGGER_PREFFIX = 'persist_'
# def setup_loggers():
# def setup_configuration(configuration_filename):
# def start_processors():
# 
# Path: pushkin/config.py
# DATABASE_CONFIG_SECTION = 'Database'
# SERVER_SPECIFIC_CONFIG_SECTION = 'ServerSpecific'
# EVENT_CONFIG_SECTION = 'Event'
# LOG_CONFIG_SECTION = 'Log'
# SERVER_CONFIG_SECTION = 'Server'
# MESSENGER_CONFIG_SECTION = 'Messenger'
# REQUEST_PROCESSOR_CONFIG_SECTION = 'RequestProcessor'
# SENDER_CONFIG_SECTION = 'Sender'
# REQUEST_HANDLER_SECTION = 'RequestHandler'
# def init(configuration_file):
# 
# Path: pushkin/database/database.py
# ENGINE = None
# SESSION = None
# ALEMBIC_CONFIG = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'alembic.ini')
#         ENGINE = create_engine(config.sqlalchemy_url, poolclass=pool.NullPool)
#         SESSION = sessionmaker(bind=ENGINE)
# def init_db():
# def create_database():
# def upgrade_database():
# def get_head_revision():
# def get_current_revision():
# def execute_query(query):
# def execute_query_with_results(query):
# def session_scope():
# def get_device_tokens(login_id):
# def get_raw_messages(login_id, title, content, screen, game, world_id, dry_run, message_id=0, event_ts_bigint=None,
#                      expiry_millis=None, priority=GCM2.PRIORITY_NORMAL, filter_platform_id=None, filter_device_token=None):
# def update_canonicals(canonicals):
# def update_unregistered_devices(unregistered):
# def process_user_login(login_id, language_id, platform_id, device_token, application_version):
# def upsert_login(login_id, language_id):
# def upsert_device(login_id, platform_id, device_token, application_version, unregistered_ts=None):
# def get_all_logins():
# def get_all_message_blacklist():
# def upsert_message_blacklist(login_id, blacklist):
# def get_login(login_id):
# def get_devices(login):
# def delete_login(login):
# def delete_device(device):
# def get_localized_message(login_id, message_id):
# def upsert_message(message_name, cooldown_ts, trigger_event_id, screen, expiry_millis, priority):
# def upsert_message_localization(message_name, language_id, message_title, message_text):
# def add_message(message_name, language_id, message_title, message_text, trigger_event_id=None, cooldown_ts=None,
#                 screen='', expiry_millis=None, priority=GCM2.PRIORITY_NORMAL):
# def get_all_messages():
# def get_message(message_name):
# def get_message_localizations(message):
# def delete_message(message):
# def delete_message_localization(message_localization):
# def get_event_to_message_mapping():
# def get_and_update_messages_to_send(user_message_set):


from pushkin import context
from pushkin import config
from pushkin.database import database
class EventRequestBatch():
    """Encapsulates a batch of events ready for processing."""

    def __init__(self, events):
        self.events = events

    def process(self):
        messages = self.build_messages()
        for message in self.filter_messages(messages):
            context.request_processor.sender_manager.submit(message)

    def build_messages(self):
        """Creates a list of messages if there are registered event handlers"""
        messages = []
        for event in self.events:
            try:
                event_params = event.pairs
                if self.validate_single(event, event_params):
                    new_messages = self.process_single(event, event_params)
                    messages.extend(new_messages)
                else:
                    context.main_logger.error("Event proto is not valid: {}".format(event))
            except Exception as e:
                context.main_logger.exception("Error while processing event proto: {}".format(event))
        return messages

    def filter_messages(self, messages):
        """Filters out messages that shouldn't be send according to cooldown"""
        if len(messages) > 0:
            pairs = {(message['login_id'], message['message_id']) for message in messages}

# Path: pushkin/context.py
# PERSIST_LOGGER_PREFFIX = 'persist_'
# def setup_loggers():
# def setup_configuration(configuration_filename):
# def start_processors():
# 
# Path: pushkin/config.py
# DATABASE_CONFIG_SECTION = 'Database'
# SERVER_SPECIFIC_CONFIG_SECTION = 'ServerSpecific'
# EVENT_CONFIG_SECTION = 'Event'
# LOG_CONFIG_SECTION = 'Log'
# SERVER_CONFIG_SECTION = 'Server'
# MESSENGER_CONFIG_SECTION = 'Messenger'
# REQUEST_PROCESSOR_CONFIG_SECTION = 'RequestProcessor'
# SENDER_CONFIG_SECTION = 'Sender'
# REQUEST_HANDLER_SECTION = 'RequestHandler'
# def init(configuration_file):
# 
# Path: pushkin/database/database.py
# ENGINE = None
# SESSION = None
# ALEMBIC_CONFIG = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'alembic.ini')
#         ENGINE = create_engine(config.sqlalchemy_url, poolclass=pool.NullPool)
#         SESSION = sessionmaker(bind=ENGINE)
# def init_db():
# def create_database():
# def upgrade_database():
# def get_head_revision():
# def get_current_revision():
# def execute_query(query):
# def execute_query_with_results(query):
# def session_scope():
# def get_device_tokens(login_id):
# def get_raw_messages(login_id, title, content, screen, game, world_id, dry_run, message_id=0, event_ts_bigint=None,
#                      expiry_millis=None, priority=GCM2.PRIORITY_NORMAL, filter_platform_id=None, filter_device_token=None):
# def update_canonicals(canonicals):
# def update_unregistered_devices(unregistered):
# def process_user_login(login_id, language_id, platform_id, device_token, application_version):
# def upsert_login(login_id, language_id):
# def upsert_device(login_id, platform_id, device_token, application_version, unregistered_ts=None):
# def get_all_logins():
# def get_all_message_blacklist():
# def upsert_message_blacklist(login_id, blacklist):
# def get_login(login_id):
# def get_devices(login):
# def delete_login(login):
# def delete_device(device):
# def get_localized_message(login_id, message_id):
# def upsert_message(message_name, cooldown_ts, trigger_event_id, screen, expiry_millis, priority):
# def upsert_message_localization(message_name, language_id, message_title, message_text):
# def add_message(message_name, language_id, message_title, message_text, trigger_event_id=None, cooldown_ts=None,
#                 screen='', expiry_millis=None, priority=GCM2.PRIORITY_NORMAL):
# def get_all_messages():
# def get_message(message_name):
# def get_message_localizations(message):
# def delete_message(message):
# def delete_message_localization(message_localization):
# def get_event_to_message_mapping():
# def get_and_update_messages_to_send(user_message_set):


# Path: pushkin/request/requests.py
from pushkin import context
from pushkin import config
from pushkin.database import database

class EventRequestBatch():
    """Encapsulates a batch of events ready for processing."""

    def __init__(self, events):
        self.events = events

    def process(self):
        messages = self.build_messages()
        for message in self.filter_messages(messages):
            context.request_processor.sender_manager.submit(message)

    def build_messages(self):
        """Creates a list of messages if there are registered event handlers"""
        messages = []
        for event in self.events:
            try:
                event_params = event.pairs
                if self.validate_single(event, event_params):
                    new_messages = self.process_single(event, event_params)
                    messages.extend(new_messages)
                else:
                    context.main_logger.error("Event proto is not valid: {}".format(event))
            except Exception as e:
                context.main_logger.exception("Error while processing event proto: {}".format(event))
        return messages

    def filter_messages(self, messages):
        """Filters out messages that shouldn't be send according to cooldown"""
        if len(messages) > 0:
            pairs = {(message['login_id'], message['message_id']) for message in messages}

            pairs_to_send = database.get_and_update_messages_to_send(pairs)