def setUp(self):
if not os.path.exists('config/'):
os.makedirs('config/')
db = DBAggregator()
for db_name, db_path in dbs.iteritems():
db.add(db_name, sqlite.format(db_path))
create_tables(db.db_conf.engine)
self.mm = MinerManager(db_conf=db.db_conf, db_miner=db.db_miner)
self.db = db
def test___init__(self):
mm = MinerManager(db_ini='templates/miner-db.ini')
self.assertTrue(hasattr(mm, 'db_conf'))
self.assertTrue(isinstance(mm.db_conf, DB))
self.assertTrue(hasattr(mm, 'db_miner'))
self.assertTrue(isinstance(mm.db_miner, DB))
self.assertTrue(hasattr(mm, 'connections'))
self.assertTrue(isinstance(mm.connections, dict))
self.assertTrue(hasattr(mm, 'miners'))
self.assertTrue(isinstance(mm.miners, dict))
self.assertTrue(hasattr(mm, 'mc'))
self.assertTrue(isinstance(mm.mc, MinerCore))
self.assertEqual(self.db.db_conf, self.mm.db_conf)
self.assertEqual(self.db.db_miner, self.mm.db_miner)
# coding: utf8
import json
import bottle
import sqlalchemy as sa
from bson import json_util
from simpleminer.manager import MinerManager
from simpleminer import models as tbl
mm = MinerManager()
db = mm.dbaggregator
def dict_to_list(d, keys=None):
for k, v in d.iteritems():
if isinstance(v, dict):
if keys:
if k in keys:
d[k] = v.values()
else:
d[k] = v.values()
dict_to_list(v, keys)
def category_dict(category):
return {
'id': category.id,
'name': category.name,
'slug': category.slug,
'miners': [],
'categories': {},
}
