summaryrefslogtreecommitdiffstats
path: root/deluge/plugins/Blocklist/deluge_blocklist/decompressers.py
blob: cd2ee8cad970b46da0a47e5c07f3dc84e01cb719 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#
# Copyright (C) 2009-2010 John Garland <johnnybg+deluge@gmail.com>
#
# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with
# the additional special exception to link portions of this program with the OpenSSL library.
# See LICENSE for more details.
#
# pylint: disable=redefined-builtin

import bz2
import gzip
import zipfile


def Zipped(reader):  # NOQA: N802
    """Blocklist reader for zipped blocklists"""

    def _open(self):
        z = zipfile.ZipFile(self.file)
        f = z.open(z.namelist()[0])
        return f

    reader.open = _open
    return reader


def GZipped(reader):  # NOQA: N802
    """Blocklist reader for gzipped blocklists"""

    def _open(self):
        return gzip.open(self.file)

    reader.open = _open
    return reader


def BZipped2(reader):  # NOQA: N802
    """Blocklist reader for bzipped2 blocklists"""

    def _open(self):
        return bz2.BZ2File(self.file)

    reader.open = _open
    return reader