def create_connection(): try: cnx = MySQLdb.connect(host = "localhost", user = "******", passwd = "Team5", db = "testDB") cur = cnx.cursor() wsLogging.audit_logging("CRUD", "String created") return cur except: wsLogging.error_logging("CRUD","Error during Connection")
def service_login(self,username, password,search): if auth_kinit(username, password) == True: audit_logging("webService", "Login successful") #Send Json token with permissions return return_jwt(username) else: audit_logging("webService", "Login Failed") #Send message with failure return "Failed to connect to service"
def send_to(Message): try: channel.basic_publish(exchange='', routing_key="FirstQ", body=str(Message)) #Logs process audit_logging("RabbitSend","Sent to queue") connection.close() except: # Logs Process error_logging("RabbitSend","Couldn't Send")
def service_login(self, username, password): if auth_kinit(username, password) == True: audit_logging("webService", "Login successful") #Send Json token with permissions else: audit_logging("webService", "Login Failed") #Send message with failure return False
def return_jwt(username): try: # Create connection to DB con = create_connection() #Get user and their services and parse out the dict taht is sent back user_service_array = get_user(con, username) user_service_tuple = user_service_array[0] service_unpacked = user_service_tuple[0] encoded = jwt.encode({'user_service': service_unpacked},'secret', algorithm='HS256') wsLogging.audit_logging("Create_Token", "Toek was created") return encoded except: wsLogging.error_logging("Create_Token","Error when creating token")
import pika from wsLogging import error_logging, audit_logging try: # Creates connection we can specify an ipAddress later connection = pika.BlockingConnection(pika.ConnectionParameters('localhost')) # Sets up connection channel = connection.channel() channel.queue_declare(queue='FirstQ') # Logs audit_logging("Rabbit Send", "Connected to Queue") except: error_logging("RabbitSend","Couldn't Create connection") def send_to(Message): try: channel.basic_publish(exchange='', routing_key="FirstQ", body=str(Message)) #Logs process audit_logging("RabbitSend","Sent to queue") connection.close() except: # Logs Process error_logging("RabbitSend","Couldn't Send")
def test_audit_logger_create(self): log = wsLogging.audit_logging("UnitTest","audit") self.assertFalse(log)
def start_consume(): try: audit_logging("Rabbit Recieve", "started consume") channel.start_consuming() except: error_logging("Rabbit Receive", "Error running CallBack")
import pika from wsLogging import error_logging, audit_logging connection = pika.BlockingConnection(pika.ConnectionParameters( host='localhost')) channel = connection.channel() channel.queue_declare(queue='FirstQ') def callback(ch, method, properties, body): print " [x] Received %r" % (body,) try: channel.basic_consume(callback, queue='FirstQ', no_ack=True) audit_logging("Rabbit Receive", "Basic consume") except: error_logging("Rabbit Receive", "Error running CallBack") def start_consume(): try: audit_logging("Rabbit Recieve", "started consume") channel.start_consuming() except: error_logging("Rabbit Receive", "Error running CallBack")