aboutsummaryrefslogtreecommitdiffstats
path: root/mediagoblin/tools/processing.py
blob: aa086a308166f0e8179f5f6a022a37b66845910f (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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
# GNU MediaGoblin -- federated, autonomous media hosting
# Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging
import json
import traceback

from urllib import request, parse

_log = logging.getLogger(__name__)

TESTS_CALLBACKS = {}


def create_post_request(url, data, **kw):
    '''
    Issue a HTTP POST request.

    Args:
        url: The URL to which the POST request should be issued
        data: The data to be send in the body of the request
        **kw:
        data_parser: The parser function that is used to parse the `data`
            argument
    '''
    data_parser = kw.get('data_parser', parse.urlencode)
    headers = kw.get('headers', {})

    return request.Request(url, data_parser(data), headers=headers)


def json_processing_callback(entry):
    '''
    Send an HTTP post to the registered callback url, if any.
    '''
    if not entry.processing_metadata:
        _log.debug('No processing callback URL for {}'.format(entry))
        return

    url = entry.processing_metadata[0].callback_url

    _log.debug('Sending processing callback for {} to {}'.format(
        entry,
        url))

    headers = {
            'Content-Type': 'application/json'}

    data = {
            'id': entry.id,
            'state': entry.state}

    # Trigger testing mode, no callback will be sent
    if url.endswith('secrettestmediagoblinparam'):
        TESTS_CALLBACKS.update({url: data})
        return True

    request = create_post_request(
            url,
            data,
            headers=headers,
            data_parser=json.dumps)

    try:
        request.urlopen(request)
        _log.debug('Processing callback for {} sent'.format(entry))

        return True
    except request.HTTPError:
        _log.error('Failed to send callback: {}'.format(
            traceback.format_exc()))

        return False