def test_location_interpolation(self):
config = Config(self.file_one)
# file_one is a StringIO, so it has no location.
self.assertEquals(config.get('one', 'location'), '${HERE}')
# file_two is a real file, so it has a location.
file_two_loc = os.path.dirname(self.file_two)
self.assertEquals(config.get('three', 'location'), file_two_loc)
def main(args=None):
# 1. read the command-line options
args = _read_args(args)
if args.version:
print(__version__)
sys.exit(0)
# 2. grab the config
from konfig import Config
config = Config(args.config).get_map('udun')
# 3. grab things in the redis list
events = _get_redis_events(config.get('redis_host', 'localhost'),
config.get('redis_port', 6379),
config.get('redis_listname', 'udun'))
# 4. combine per-collection
collection_ids = _get_impacted_collections(events)
# 5. send it to Balrog
_poke_balrog(collection_ids)
def main(args=None):
# 1. read the command-line options
args = _read_args(args)
if args.version:
print(__version__)
sys.exit(0)
# 2. grab the config
from konfig import Config
config = Config(args.config).get_map('udun')
# 3. grab things in the redis list
events = _get_redis_events(config.get('redis_host', 'localhost'),
config.get('redis_port', 6379),
config.get('redis_listname', 'udun'))
# 4. combine per-collection
collection_ids = _get_impacted_collections(events)
# 5. send it to Balrog
_poke_balrog(collection_ids)
def test_reader(self):
config = Config(self.file_one)
# values conversion
self.assertMultiLineEqual(config.get('one', 'foo'), 'bar')
self.assertTrue(config.get('one', 'num') == -12)
self.assertMultiLineEqual(config.get('one', 'not_a_num'), "12abc")
self.assertMultiLineEqual(config.get('one', 'st'), 'o=k')
self.assertListEqual(config.get('one', 'lines'), [1, 'two', 3])
self.assertMultiLineEqual(config.get('one', 'env'), 'some stuff')
# getting a map
_map = config.get_map()
self.assertMultiLineEqual(_map['one.foo'], 'bar')
_map = config.get_map('one')
self.assertMultiLineEqual(_map['foo'], 'bar')
del os.environ['__STUFF__']
self.assertMultiLineEqual(config.get('one', 'env'), 'some stuff')
# extends
self.assertMultiLineEqual(config.get('three', 'more'), 'stuff')
self.assertMultiLineEqual(config.get('one', 'two'), 'a')
def main():
configure_logger()
if len(sys.argv) > 1:
ini = sys.argv[1]
else:
ini = "kompost.ini"
os.environ["CURDIR"] = os.getcwd()
config = Config(codecs.open(ini, "r", "utf8"))
config = dict(config.items("kompost"))
target = config["target"]
src = config["src"]
socket.setdefaulttimeout(int(config.get("timeout", 10)))
config["media"] = os.path.abspath(os.path.join(target, "media"))
config["generic"] = os.path.join(src, "generic.mako")
config["cats"] = os.path.join(src, "category.mako")
config["icons"] = ("pen.png", "info.png", "thumbsup.png", "right.png", "flash.png")
config["metadata"] = os.path.join(target, "metadata.json")
generate(config)
pdf(config)
def main(argv):
parser = argparse.ArgumentParser(prog=argv[0], description="Location Importer")
parser.add_argument("--dry-run", action="store_true")
# TODO rely on ICHNAEA_CFG / ichnaea.config, as the worker is relying
# on it anyways
parser.add_argument("config", help="config file")
parser.add_argument("source", help="source file")
args = parser.parse_args(argv[1:])
settings = Config(args.config).get_map("ichnaea")
db = Database(settings["db_master"], socket=settings.get("db_master_socket"), create=False)
session = db.session()
added = load_file(session, args.source)
print("Added %s records." % added)
if args.dry_run:
session.rollback()
else: # pragma: no cover
session.commit()
return added
def main():
configure_logger()
if len(sys.argv) > 1:
ini = sys.argv[1]
else:
ini = "kompost.ini"
os.environ['CURDIR'] = os.getcwd()
config = Config(codecs.open(ini, "r", "utf8"))
config = dict(config.items('kompost'))
target = config['target']
src = config['src']
socket.setdefaulttimeout(int(config.get('timeout', 10)))
config['media'] = os.path.abspath(os.path.join(target, 'media'))
config['generic'] = os.path.join(src, 'generic.mako')
config['cats'] = os.path.join(src, 'category.mako')
config['icons'] = ('pen.png', 'info.png', 'thumbsup.png', 'right.png',
'flash.png')
config['metadata'] = os.path.join(target, 'metadata.json')
generate(config)
pdf(config)
def test_utf8(self):
utf8 = os.path.join(os.path.dirname(__file__), 'utf8.ini')
config = Config(utf8)
self.assertEqual(config.get('ok', 'yeah'), u'é')
def test_override_mode(self):
config = Config(self.file_override)
self.assertEquals(config.get('one', 'foo'), 'baz')
self.assertEquals(config.get('three', 'more'), 'stuff')
