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