Class DNSManager

python3_cron_scripts/libs3/DNSManager.py:24–369

Marinus collects DNS information from multiple sources. In the original version of Marinus, there was a separate table for each source. Maintaining separate tables makes it difficult to search. Therefore, the tables have been merged into "all_dns". This class acts as the interfa

class DNSManager(object):
    """
    Marinus collects DNS information from multiple sources.
    In the original version of Marinus, there was a separate table for each source.
    Maintaining separate tables makes it difficult to search.
    Therefore, the tables have been merged into "all_dns".
    This class acts as the interface for translating a DNS record into the all_dns format.
    When submitting a DNS record, a "source" must be provided.
    The approved sources are "virustotal", "common_crawl", "sonar_dns", and "ssl".
    These will eventually be saved as constants.
    """

    all_dns_collection = None
    _mongo_connector = None
    _ip_manager = None
    _logger = None

    def _log(self):
        """
        Get the log
        """
        return logging.getLogger(__name__)

    def __init__(self, mongo_connector, alternative_collection=None):
        """
        Initialize the object with the necessary database configurations
        """
        self._logger = self._log()
        self._mongo_connector = mongo_connector
        if alternative_collection is not None:
            try:
                self.all_dns_collection = getattr(
                    mongo_connector, alternative_collection
                )()
            except:
                self._logger.error(
                    "FATAL: Could not fetch dynamic collection in DNS Manager"
                )
                exit(1)
        else:
            self.all_dns_collection = mongo_connector.get_all_dns_connection()

    @staticmethod
    def monthdelta(date, delta):
        """
        Return the date from the given delta

        :param date: The original date
        :param delta: The change from the original date that is to be calculated.
        """
        m, y = (date.month + delta) % 12, date.year + ((date.month) + delta - 1) // 12
        if not m:
            m = 12
        d = min(
            date.day,
            [
                31,
                29 if y % 4 == 0 and not y % 400 == 0 else 28,