Source code for PyOpenWorm.data_trans.http_ds

from contextlib import contextmanager
from six.moves.urllib.request import urlopen
from ..datasource import Informational
from .file_ds import FileDataSource


class HTTPFileDataSource(FileDataSource):
    url = Informational(display_name='URL')

    @contextmanager
    def file_contents(self):
        return urlopen(self.url.one())


__yarom_mapped_classes__ = (HTTPFileDataSource,)