Revision 8417c3df

b/snf-cyclades-app/synnefo/logic/management/commands/queue_inspect.py
1
# Copyright 2011-2012 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or without
4
# modification, are permitted provided that the following conditions
5
# are met:
6
#
7
#   1. Redistributions of source code must retain the above copyright
8
#      notice, this list of conditions and the following disclaimer.
9
#
10
#  2. Redistributions in binary form must reproduce the above copyright
11
#     notice, this list of conditions and the following disclaimer in the
12
#     documentation and/or other materials provided with the distribution.
13
#
14
# THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
15
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17
# ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
18
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24
# SUCH DAMAGE.
25
#
26
# The views and conclusions contained in the software and documentation are
27
# those of the authors and should not be interpreted as representing official
28
# policies, either expressed or implied, of GRNET S.A.
29
#
30

  
31
import pprint
32

  
33
from optparse import make_option
34
from django.core.management.base import BaseCommand, CommandError
35

  
36
from synnefo.lib.amqp import AMQPClient
37

  
38

  
39
class Command(BaseCommand):
40
    args = "<queue name>"
41
    help = "Inspect the messages of a queue. Close all other clients in "\
42
           "order to be able to inspect unacknowledged messages."
43

  
44
    option_list = BaseCommand.option_list + (
45
        make_option('--no-requeue', action='store_false', dest='requeue',
46
                    default=True, help="Do not requeue the messages"),
47
        make_option('-i', '--interactive', action='store_true', default=False,
48
                    dest='interactive', help="Interactive mode")
49
        )
50

  
51
    def handle(self, *args, **options):
52
        if len(args) != 1:
53
            raise CommandError("Please provide a queue")
54

  
55
        queue = args[0]
56
        interactive = options['interactive']
57
        requeue = options['requeue']
58

  
59
        client = AMQPClient()
60
        client.connect()
61

  
62
        pp = pprint.PrettyPrinter(indent=4, width=4)
63

  
64
        more_msgs = True
65
        counter = 0
66
        sep = '-' * 80
67
        while more_msgs:
68
            msg = client.basic_get(queue=queue)
69
            if msg:
70
                counter += 1
71
                print sep
72
                print 'Message %d:' % counter
73
                print sep
74
                pp.pprint(msg)
75
                if not requeue or interactive:
76
                    if interactive and not get_user_confirmation():
77
                        continue
78
                    # Acknowledging the message will remove it from the queue
79
                    client.basic_ack(msg)
80
            else:
81
                more_msgs = False
82

  
83

  
84
def get_user_confirmation():
85
    ans = raw_input("Do you want to delete this message (N/y):")
86

  
87
    if not ans:
88
        return False
89
    if ans not in ['Y', 'y']:
90
        return False
91
    return True

Also available in: Unified diff