From f0772d299a4e93c179a6ac1671c42c18aeb13276 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Wed, 28 Mar 2018 10:27:02 +0200 Subject: [PATCH] new: Add helper for ASN object --- pymisp/tools/__init__.py | 1 + pymisp/tools/asnobject.py | 22 ++++++++++++++++++++++ 2 files changed, 23 insertions(+) create mode 100644 pymisp/tools/asnobject.py diff --git a/pymisp/tools/__init__.py b/pymisp/tools/__init__.py index 705b2d1..811a857 100644 --- a/pymisp/tools/__init__.py +++ b/pymisp/tools/__init__.py @@ -13,6 +13,7 @@ from .openioc import load_openioc, load_openioc_file # noqa from .sbsignatureobject import SBSignatureObject # noqa from .fail2banobject import Fail2BanObject # noqa from .domainipobject import DomainIPObject # noqa +from .asnobject import ASNObject # noqa if sys.version_info >= (3, 6): from .emailobject import EMailObject # noqa diff --git a/pymisp/tools/asnobject.py b/pymisp/tools/asnobject.py new file mode 100644 index 0000000..ce80029 --- /dev/null +++ b/pymisp/tools/asnobject.py @@ -0,0 +1,22 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from .abstractgenerator import AbstractMISPObjectGenerator +import logging + +logger = logging.getLogger('pymisp') + + +class ASNObject(AbstractMISPObjectGenerator): + + def __init__(self, parameters, strict=True, standalone=True, **kwargs): + super(ASNObject, self).__init__('asn', strict=strict, standalone=standalone, **kwargs) + self._parameters = parameters + self.generate_attributes() + + def generate_attributes(self): + first = self._sanitize_timestamp(self._parameters.pop('first-seen', None)) + self._parameters['first-seen'] = first + last = self._sanitize_timestamp(self._parameters.pop('last-seen', None)) + self._parameters['last-seen'] = last + return super(ASNObject, self).generate_attributes()