def test_create_source(self):
source = {'slug': 'foo', 'title': 'Foo source', 'url': 'http://foo'}
Source.ensure(source)
session.commit()
cnt = session.query(Source).count()
assert cnt == 1, cnt
Source.ensure(source)
session.commit()
cnt = session.query(Source).count()
assert cnt == 1, cnt
def load_ba_fixtures(config):
# This is messy. Would be cool to do it more cleanly, but how?
if not len(BA_FIXTURES['entities']):
with open(os.path.join(FIXTURES, 'ba.mapping.yaml'), 'rb') as fh:
mapping = yaml.load(fh)
mapper = Mapper(mapping, config.resolver, scope=config.base_uri)
with open(os.path.join(FIXTURES, 'ba.csv'), 'rb') as csvfh:
reader = unicodecsv.DictReader(csvfh)
for row in reader:
_, data = mapper.apply(row)
BA_FIXTURES['entities'].append(data)
source = Source.ensure({
'slug': BA_SOURCE,
'title': 'BiH Parliament',
'url': 'http://foo.ba/'
})
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = False
permission.resource_id = source.id
permission.resource_type = Permission.SOURCE
session.add(permission)
session.commit()
for entity in BA_FIXTURES['entities']:
config.entities.save(entity['$schema'], entity, source_id=source.id)
get_loom_indexer().index(source=BA_SOURCE)
def load_ba_fixtures(config):
# This is messy. Would be cool to do it more cleanly, but how?
if not len(BA_FIXTURES['entities']):
with open(os.path.join(FIXTURES, 'ba.mapping.yaml'), 'rb') as fh:
mapping = yaml.load(fh)
mapper = Mapper(mapping, config.resolver, scope=config.base_uri)
with open(os.path.join(FIXTURES, 'ba.csv'), 'rb') as csvfh:
reader = unicodecsv.DictReader(csvfh)
for row in reader:
_, data = mapper.apply(row)
BA_FIXTURES['entities'].append(data)
source = Source.ensure({
'slug': BA_SOURCE,
'title': 'BiH Parliament',
'url': 'http://foo.ba/'
})
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = False
permission.resource_id = source.id
permission.resource_type = Permission.SOURCE
session.add(permission)
session.commit()
for entity in BA_FIXTURES['entities']:
config.entities.save(entity['$schema'], entity, source_id=source.id)
get_loom_indexer().index(source=BA_SOURCE)
def source(self):
if not hasattr(self, '_source_id'):
source = self.get('source')
log.info("Registering source: %r", source['slug'])
self._source_id = Source.ensure(source).id
return self._source_id
