-
Notifications
You must be signed in to change notification settings - Fork 16
/
run_terminate_sqs.py
executable file
·104 lines (78 loc) · 2.81 KB
/
run_terminate_sqs.py
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/env python3.12
from env import env
from run_common import AWSCli
from run_common import print_message
from run_common import print_session
options, args = dict(), list()
if __name__ == "__main__":
from run_common import parse_args
options, args = parse_args()
def run_terminate_standard_queue(name, settings):
print_message(f'terminate standard sqs queue: {name}')
aws_cli = AWSCli(settings['AWS_REGION'])
cmd = ['sqs', 'get-queue-url']
cmd += ['--queue-name', f'{name}-dead-letter']
result = aws_cli.run(cmd, ignore_error=True)
if result:
queue_url = result['QueueUrl']
cmd = ['sqs', 'delete-queue']
cmd += ['--queue-url', queue_url]
aws_cli.run(cmd, ignore_error=True)
cmd = ['sqs', 'get-queue-url']
cmd += ['--queue-name', name]
result = aws_cli.run(cmd, ignore_error=True)
if result:
queue_url = result['QueueUrl']
cmd = ['sqs', 'delete-queue']
cmd += ['--queue-url', queue_url]
aws_cli.run(cmd, ignore_error=True)
def run_terminate_fifo_queue(name, settings):
print_message(f'terminate fifo sqs queue: {name}')
aws_cli = AWSCli(settings['AWS_REGION'])
queue_name = name.replace('.fifo', '')
cmd = ['sqs', 'get-queue-url']
cmd += ['--queue-name', f'{queue_name}-dead-letter.fifo']
result = aws_cli.run(cmd, ignore_error=True)
if result:
queue_url = result['QueueUrl']
cmd = ['sqs', 'delete-queue']
cmd += ['--queue-url', queue_url]
aws_cli.run(cmd, ignore_error=True)
cmd = ['sqs', 'get-queue-url']
cmd += ['--queue-name', name]
result = aws_cli.run(cmd, ignore_error=True)
if result:
queue_url = result['QueueUrl']
cmd = ['sqs', 'delete-queue']
cmd += ['--queue-url', queue_url]
aws_cli.run(cmd, ignore_error=True)
################################################################################
#
# start
#
################################################################################
print_session('terminate sqs')
target_name = None
region = options.get('region')
is_target_exists = False
for settings in env.get('sqs', list()):
if target_name and settings['NAME'] != target_name:
continue
if region and settings['AWS_REGION'] != region:
continue
is_target_exists = True
if settings['SQS_TYPE'] == 'STANDARD':
run_terminate_standard_queue(settings['NAME'], settings)
elif settings['SQS_TYPE'] == 'FIFO':
run_terminate_fifo_queue(settings['NAME'], settings)
else:
print('The SQS TYPE variable has no value.')
raise Exception()
if is_target_exists is False:
mm = list()
if target_name:
mm.append(target_name)
if region:
mm.append(region)
mm = ' in '.join(mm)
print(f'sqs: {mm} is not found in config.json')