def store(self, message):
'''
Store messages flagging a business
format of message:
{
'business_id': ----
'drug': ---
}
Message should be a dictionary object
'''
flag = Flag(business_id=message['business_id'],
drug_name=message.get('drug_name', 'Unspecified',
date=datetime.utcnow()))
flag.save()
def __init__(self):
# handle zmq setup
zmq_port = os.environ['PERSIST_RESPONSE_PORT']
context = zmq.Context()
socket = context.socket(zmq.REP)
self.socket = socket
# now, setup couchdb
db_port = os.environ.get('COUCH_PORT', '5984')
db_server = os.environ.get('COUCH_HOST', '127.0.0.1')
views_path = os.environ['COUCH_VIEWS_LOCATION']
couch = couchdbkit.Server(uri='http://%s:%s' % (db_server, db_port))
db = couch.get_or_create_db('flags')
push(views_path, db)
Flag.set_db(db)
def retreive(self, business):
'''
Given a particular business ID, retrieve all flags
if any associated with the business.
'''
flag_results = Flag.view('flag/business', key=business)
flags = []
for flag in flag_results:
flags.append(flag)
return flags
