def connect_db(*a, **k):
if 'config' in k:
if not os.path.exists(k['config']):
raise Exception('Config file missing.')
config = Config(k['config'])
plist = [
('engine','engine','engine'),
('host','host','dbhost'),
('name','db','dbname'),
('user','user','dbuser'),
('passwd','password','dbpass')
]
db_params = {}
for p,s,t in plist:
v = k.get(p, config.get('database', t, db_defaults.get(p)))
if v:
if p == 'name':
s = 'db'
else:
s = p
db_params[s] = v
try:
db = DB(**db_params)
except:
logger.error('Error connecting to database with config %s' % repr((a,k)))
raise
logger.debug('Connected to db with config %s' % repr((a,k)))
else:
try:
db = DB(*a, **k)
except:
logger.error('Error connecting to database with config %s' % repr((a,k)))
raise
logger.debug('Connected to db with %s' % repr((a,k)))
return db
class Instance(object):
"""represents an installed DataZoomer instance
Use for performing methods on an entire instance.
"""
def __init__(self, name, path='.'):
self.name = name
config_path = locate_config('dz.conf', path)
if not config_path:
raise Exception('dz.conf missing')
self.config = Config(config_path)
@property
def sites_path(self):
"""the path to the sites of the instance"""
return self.config.get('sites', 'path')
@property
def path(self):
"""the path of the instance"""
path, _ = os.path.split(self.sites_path)
return path
@property
def sites(self):
"""a list of sites for the instance"""
listdir = os.listdir
isdir = os.path.isdir
join = os.path.join
path = self.sites_path
return [Site(name) for name in listdir(path) if isdir(join(path, name))]
def run(self, *jobs):
"""run jobs on an entire instance"""
logger = logging.getLogger(self.name)
for site in self.sites:
try:
zoom.system.setup(self.path, site.name)
except Exception, e:
logger.warning(str(e))
logger.warning('unable to setup {}'.format(site.name))
continue
if zoom.system.background:
logger.debug('initialized site {}'.format(site.name))
for job in jobs:
try:
job_name = job.name
except:
job_name = job.__name__
logger.debug('running {}.{} for {}'.format(
self.name,
job_name,
site.name)
)
job()
class Instance(object):
"""represents an installed DataZoomer instance
Use for performing methods on an entire instance.
"""
def __init__(self, name, path='.'):
self.name = name
self.config = Config(locate_config('dz.conf', path))
@property
def sites_path(self):
"""the path to the sites of the instance"""
return self.config.get('sites', 'path')
@property
def path(self):
"""the path of the instance"""
path, _ = os.path.split(self.sites_path)
return path
@property
def sites(self):
"""a list of sites for the instance"""
listdir = os.listdir
isdir = os.path.isdir
join = os.path.join
path = self.sites_path
return [
Site(name) for name in listdir(path) if isdir(join(path, name))
]
def run(self, *jobs):
"""run jobs on an entire instance"""
logger = logging.getLogger(self.name)
for site in self.sites:
try:
zoom.system.setup(self.path, site.name)
except Exception, e:
logger.warning(str(e))
logger.warning('unable to setup {}'.format(site.name))
continue
if zoom.system.background:
logger.debug('initialized site {}'.format(site.name))
for job in jobs:
try:
job_name = job.name
except:
job_name = job.__name__
logger.debug('running {}.{} for {}'.format(
self.name, job_name, site.name))
job()