-
Notifications
You must be signed in to change notification settings - Fork 0
/
worker-server.py
94 lines (75 loc) · 2.91 KB
/
worker-server.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
from usps import USPSApi
import json
import pickle, jsonpickle
import io
import sys, os
import pika
import redis
#####################################################################################
#
# Citations
#
# USPS API: https://pypi.org/project/usps-api/
#
######################################################################################
# Packages needed for to execute
# pip install usps-api
redisHost = os.getenv("REDIS_HOST") or "localhost"
rabbitMQHost = os.getenv("RABBITMQ_HOST") or "localhost"
#checking connection and establishing channel for communication
print("Connecting to rabbitmq({}) and redis({})".format(rabbitMQHost,redisHost))
connection=pika.BlockingConnection(pika.ConnectionParameters(host=rabbitMQHost))
channel=connection.channel()
channel.queue_declare(queue='worker_queue', durable=True)
#connect to redis database
trackingNumtoInfo = redis.Redis(host=redisHost, db=1)
# USPS Login and Order class structure
uspsLogin='777PLATI6457'
usps = USPSApi(uspsLogin)
class Order:
trackingTime = None
trackingDate = None
trackingEvent = None
def getTrackingInfo(order, number):
# Grab data for an order via the tracking number
# NOTE: A tracking number can only be called 35 times before exhausting the resource
try:
trackingInfo = usps.track(str(number))
# Grab data for most recent update
order.trackingTime = trackingInfo.result['TrackResponse']['TrackInfo']['TrackDetail'][0]['EventTime']
order.trackingDate = trackingInfo.result['TrackResponse']['TrackInfo']['TrackDetail'][0]['EventDate']
order.trackingEvent = trackingInfo.result['TrackResponse']['TrackInfo']['TrackDetail'][0]['Event']
#print("\nOrder Info\nTime: ", order.trackingTime, "\nDate: ", order.trackingDate, "\nEvent: ", order.trackingEvent, "\n")
return
except:
order.trackingTime = 'Error'
order.trackingDate = 'Error'
order.trackingEvent = 'Error'
return
def serv_match(trackingNumber):
newOrder = Order()
# Parse received request
try:
getTrackingInfo(newOrder, trackingNumber)
response = { 'trackingNumber':trackingNumber,
'trackingTime' : newOrder.trackingTime,
'trackingDate' : newOrder.trackingDate,
'trackingEvent' : newOrder.trackingEvent
}
except:
response = { 'trackingNumber': trackingNumber, 'trackingTime' : 'Error', 'trackingDate' : 'Error', 'trackingEvent' : 'Error'}
print("Response is\n", response)
return jsonpickle.encode(response)
#when worker receive data from the server
def callback(ch, method, properties, body):
serverMessage=jsonpickle.decode(body)
print("Received message: ", serverMessage)
tracking_str=str(serverMessage['tracking_num'])
if (tracking_str!="none"):
response=serv_match(tracking_str)
trackingNumtoInfo.set(tracking_str,response)
print("[x] Done")
ch.basic_ack(delivery_tag=method.delivery_tag)
#pub-sub consumption by workers
channel.basic_consume(queue='worker_queue', on_message_callback=callback, auto_ack=False)
channel.start_consuming()