mirror of https://github.com/MISP/PyMISP
29 lines
835 B
Python
29 lines
835 B
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from .abstractgenerator import AbstractMISPObjectGenerator
|
||
|
|
import logging
|
||
|
|
from pyfaup.faup import Faup
|
||
|
|
from urllib.parse import unquote_plus
|
||
|
|
|
||
|
|
logger = logging.getLogger('pymisp')
|
||
|
|
|
||
|
|
faup = Faup()
|
||
|
|
|
||
|
|
|
||
|
|
class URLObject(AbstractMISPObjectGenerator):
|
||
|
|
|
||
|
|
def __init__(self, url, standalone=True, **kwargs):
|
||
|
|
# PY3 way:
|
||
|
|
# super().__init__('file')
|
||
|
|
super(URLObject, self).__init__('url', standalone=standalone, **kwargs)
|
||
|
|
faup.decode(unquote_plus(url))
|
||
|
|
self.generate_attributes()
|
||
|
|
|
||
|
|
def generate_attributes(self):
|
||
|
|
self.add_attribute('url', value=faup.url.decode())
|
||
|
|
if faup.get_host():
|
||
|
|
self.add_attribute('host', value=faup.get_host())
|
||
|
|
if faup.get_domain():
|
||
|
|
self.add_attribute('domain', value=faup.get_domain())
|