def __init__(self, verbose=False):
# TODO: add uniq id
self.verbose = verbose
uniq = Uniq()
self.uniqid = uniq.getid(uniq.WORKER)
# zookeeper connection
zb = ZooBorgFactory(Configuration.getworker()['zookeeper']['mockmode'],
Configuration.getworker()['zookeeper']['ip'],
Configuration.getworker()['zookeeper']['port'])
addconf = zb.getConf(ZooConst.WORKER)
con_str = addconf['broker']['connectionstr']
zb.register(ZooConst.WORKER, self.uniqid, self.zoo_conf_changed)
self.worker = MajorDomoWorker(con_str, "echo", self.verbose)
# init mongodb pool
self.mp = MongoPool(Configuration.getworker()['mongo']['host'],
Configuration.getworker()['mongo']['port'],
Configuration.getworker()['mongo']['db'],
Configuration.getworker()['mongo']['maxcon'])
def FoodDAOFactory(id=None, name_fr=None, pro=None, lip=None, glu=None, cal=None, qty=None, unit=None):
factory = {}
factory['REAL'] = FoodDAOMongo
factory['MOCK'] = FoodDAOMock
workerdaomockmode = Configuration.getworker()['dbdaoimpl']
return factory[workerdaomockmode](id, name_fr, pro, lip, glu, cal, qty, unit)
list.append(FoodDAOMock('3', 'random food 3' + pattern))
list.append(FoodDAOMock('4', 'random food 4' + pattern))
return list
def save(self):
pass
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Test objects')
parser.add_argument('-c', '--conf', help='conf to load', type=str)
args = parser.parse_args()
# init conf
conf = Configuration(args.conf)
# init mongo pool
mp = MongoPool(Configuration.getworker()['mongo']['host'],
Configuration.getworker()['mongo']['port'],
Configuration.getworker()['mongo']['db'])
impl = FoodDAOFactory('youpi', 'label')
print impl
print type(impl)
print FoodDAOFactory().get('myid')
fdao = FoodDAOFactory()
try:
res = fdao.searchFoodByPattern('an')
except:
print 'we got an error'
finally:
print res
# encoding: utf-8
import argparse
import traceback
import codecs
import re
from distark.majordaemon.worker.db.mongopool import MongoPool
from distarkcli.utils.MyConfiguration import Configuration
# TODO: pass the configuration by command line
MAX_CONN = 1
host = Configuration.getworker()["mongo"]["host"]
port = Configuration.getworker()["mongo"]["port"]
db = Configuration.getworker()["mongo"]["db"]
mp = MongoPool(host, port, db, MAX_CONN)
refloat = re.compile("^\d+(\.\d+)*$")
def is_valid_qte(line):
res = False
qte = [u"100 g", u"1 part", u"1 pot", u"1 unité", u"1 tranche", u"1 portion", u"100 ml", u"1 canette", u"c.c."]
if line in qte:
res = True
return res
def store_data(datadic):
# store in db
