summaryrefslogtreecommitdiffstats
path: root/deluge/plugins/Blocklist/deluge_blocklist/detect.py
blob: 5d7858d6877d991c2f3dd491c932dfc5abf79c9c (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
45
46
47
48
49
# -*- coding: utf-8 -*-
#
# 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.
#

from .decompressers import BZipped2, GZipped, Zipped
from .readers import EmuleReader, PeerGuardianReader, SafePeerReader

COMPRESSION_TYPES = {b'PK': 'Zip', b'\x1f\x8b': 'GZip', b'BZ': 'BZip2'}

DECOMPRESSERS = {'Zip': Zipped, 'GZip': GZipped, 'BZip2': BZipped2}

READERS = {
    'Emule': EmuleReader,
    'SafePeer': SafePeerReader,
    'PeerGuardian': PeerGuardianReader,
}


class UnknownFormatError(Exception):
    pass


def detect_compression(filename):
    with open(filename, 'rb') as _file:
        magic_number = _file.read(2)
    return COMPRESSION_TYPES.get(magic_number, '')


def detect_format(filename, compression=''):
    file_format = ''
    for reader in READERS:
        if create_reader(reader, compression)(filename).is_valid():
            file_format = reader
            break
    return file_format


def create_reader(file_format, compression=''):
    reader = READERS.get(file_format)
    if reader and compression:
        decompressor = DECOMPRESSERS.get(compression)
        if decompressor:
            reader = decompressor(reader)
    return reader