cd103777ab
Sometimes script cannot delete vhost and return a 500 status code Change-Id: I8df789ee94bbc529fb0b15c036f11ceeab3af931
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
from pyrabbit.api import Client
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description="Script for creating rabbitmq"
|
|
"users and vhost for jenkins's"
|
|
"jobs")
|
|
parser.add_argument("-rabbitmq_url", dest='rabbitmq_url', type=str,
|
|
help="URL of using RabbitMQ", default='localhost:55672')
|
|
parser.add_argument("-rabbitmq_username", dest='rabbitmq_username', type=str,
|
|
help="Username for RabbitMQ auth", default='guest')
|
|
parser.add_argument("-rabbitmq_password", dest='rabbitmq_password', type=str,
|
|
help="Password for RabbitMQ auth", default='guest')
|
|
parser.add_argument("-username", dest='username', type=str,
|
|
help="Username", default='test')
|
|
parser.add_argument("-password", dest='password', type=str,
|
|
help="Password", default='swordfish')
|
|
parser.add_argument("-vhostname", dest='vhostname', type=str,
|
|
help="Vhost name", default='test')
|
|
args = parser.parse_args()
|
|
rabbitmq_url = args.rabbitmq_url
|
|
rabbitmq_user = args.rabbitmq_username
|
|
rabbitmq_password = args.rabbitmq_password
|
|
user = args.username
|
|
password = args.password
|
|
vhost = args.vhostname
|
|
cl = Client(rabbitmq_url, rabbitmq_user, rabbitmq_password)
|
|
assert cl.is_alive()
|
|
|
|
for queue in cl.get_queues():
|
|
if queue['vhost'] == vhost:
|
|
cl.purge_queue(vhost, queue['name'])
|
|
cl.delete_queue(vhost, queue['name'])
|
|
|
|
for vhost_ in cl.get_all_vhosts():
|
|
if vhost_['name'] == vhost:
|
|
while True:
|
|
try:
|
|
cl.delete_vhost(vhost_['name'])
|
|
break
|
|
except Exception:
|
|
pass
|
|
|
|
for user_ in cl.get_users():
|
|
if user_['name'] == user:
|
|
cl.delete_user(user_['name'])
|
|
|
|
cl.create_vhost(vhost)
|
|
cl.create_user(user, password, tags='administrator')
|
|
cl.set_vhost_permissions(vhost, user, '.*', '.*', '.*')
|