Skip to content
Snippets Groups Projects
Commit 7eea5b36 authored by Ravi Tripathi's avatar Ravi Tripathi
Browse files

Changes in this commit:

* Add a function to percent encode username
* Add parameter 'queuename' to add_user function.
* Add parameter 'queuename' to consume function.
parent 0eceb2af
No related branches found
No related tags found
No related merge requests found
...@@ -2,21 +2,25 @@ import logging ...@@ -2,21 +2,25 @@ import logging
import argparse import argparse
from rc_rmq import RCRMQ from rc_rmq import RCRMQ
import json import json
from urllib.parse import quote
rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'}) rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})
tasks = {'create_account': None, 'git_commit': None, 'dir_verify': None, 'subscribe_mail_list': None, 'notify_user': None} tasks = {'create_account': None, 'git_commit': None, 'dir_verify': None, 'subscribe_mail_list': None, 'notify_user': None}
logger_fmt = '%(asctime)s [%(module)s] - %(message)s' logger_fmt = '%(asctime)s [%(module)s] - %(message)s'
def add_account(username, email, full='', reason=''): def add_account(username, queuename, email, full="", reason=""):
rc_rmq.publish_msg({ rc_rmq.publish_msg(
'routing_key': 'request.' + username, {
'msg': { "routing_key": "request." + queuename,
"msg": {
"username": username, "username": username,
"email": email, "email": email,
"fullname": full, "fullname": full,
"reason": reason "reason": reason,
"queuename": queuename,
},
} }
}) )
rc_rmq.disconnect() rc_rmq.disconnect()
def worker(ch, method, properties, body): def worker(ch, method, properties, body):
...@@ -34,18 +38,20 @@ def worker(ch, method, properties, body): ...@@ -34,18 +38,20 @@ def worker(ch, method, properties, body):
rc_rmq.stop_consume() rc_rmq.stop_consume()
rc_rmq.delete_queue() rc_rmq.delete_queue()
def consume(username, routing_key='', callback=worker, debug=False): def consume(queuename, routing_key="", callback=worker, debug=False):
if routing_key == '': if routing_key == "":
routing_key = 'complete.' + username routing_key = "complete." + queuename
if debug: if debug:
sleep(5) sleep(5)
else: else:
rc_rmq.start_consume({ rc_rmq.start_consume(
'queue': username, {
'routing_key': routing_key, "queue": queuename,
'cb': callback "routing_key": routing_key,
}) "cb": callback,
}
)
rc_rmq.disconnect() rc_rmq.disconnect()
return { 'success' : True } return { 'success' : True }
...@@ -72,3 +78,9 @@ def get_logger(args=None): ...@@ -72,3 +78,9 @@ def get_logger(args=None):
logging.basicConfig(format=logger_fmt, level=logger_lvl) logging.basicConfig(format=logger_fmt, level=logger_lvl)
return logging.getLogger(__name__) return logging.getLogger(__name__)
def encode_name(uname):
uname_quote = quote(uname)
if "." in uname_quote:
uname_quote = uname_quote.replace(".", "%2E")
return uname_quote
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment