def _get_scaffold_dir(self, scaffold_name):
if ':' not in scaffold_name:
entry_points = pkg_resources.iter_entry_points('tangled.scaffolds')
for entry_point in entry_points:
if entry_point.name == scaffold_name:
module_name = entry_point.module_name
path = entry_point.attrs[0]
return asset_path(module_name, path)
else:
return None
return asset_path(scaffold_name)
def _get_scaffold_dir(self, scaffold_name):
if ':' not in scaffold_name:
entry_points = pkg_resources.iter_entry_points('tangled.scaffolds')
for entry_point in entry_points:
if entry_point.name == scaffold_name:
module_name = entry_point.module_name
path = entry_point.attrs[0]
return asset_path(module_name, path)
else:
return None
return asset_path(scaffold_name)
def load_usps_street_suffixes(user,
password,
database,
host='localhost',
port=5432):
"""Load USPS street suffixes into database."""
file_name = '{model.__tablename__}.csv'.format(model=USPSStreetSuffix)
path = asset_path('bycycle.core.model', file_name)
engine = create_engine(user, password, host, port, database)
session_factory = sessionmaker(bind=engine)
session = session_factory()
printer.info('Deleting existing USPS street suffixes...', end=' ')
count = session.query(USPSStreetSuffix).delete()
session.commit()
printer.info(count, 'deleted')
printer.info('Adding USPS street suffixes...', end=' ')
with open(path) as fp:
reader = csv.DictReader(fp)
records = [USPSStreetSuffix(**row) for row in reader]
count = len(records)
session.add_all(records)
session.commit()
printer.info(count, 'added')
session.close()
engine.dispose()
def get_template(self, uri):
"""Allow absolute and asset paths."""
if os.path.isabs(uri):
srcfile = uri
elif ':' in uri:
srcfile = asset_path(uri)
else:
srcfile = None
if srcfile and os.path.isabs(srcfile):
try:
if self.filesystem_checks:
return self._check(uri, self._collection[uri])
else:
return self._collection[uri]
except KeyError:
if os.path.isfile(srcfile):
return self._load(srcfile, uri)
return super().get_template(uri)
def get_template(self, uri):
"""Allow absolute and asset paths."""
if os.path.isabs(uri):
srcfile = uri
elif ':' in uri:
srcfile = asset_path(uri)
else:
srcfile = None
if srcfile and os.path.isabs(srcfile):
try:
if self.filesystem_checks:
return self._check(uri, self._collection[uri])
else:
return self._collection[uri]
except KeyError:
if os.path.isfile(srcfile):
return self._load(srcfile, uri)
return super().get_template(uri)
def test_asset_path_for_module(self):
path = util.asset_path('tangled.util')
self.assertTrue(path.endswith('/tangled/tangled'))
def test_asset_path_with_base_rel_path(self):
path = util.asset_path('tangled.util:dir', 'file')
self.assertTrue(path.endswith('/tangled/tangled/dir/file'))
def test_asset_path_for_module(self):
path = util.asset_path('tangled.util')
self.assertTrue(path.endswith('/tangled/tangled/util'))
def test_asset_path_with_base_rel_path(self):
path = util.asset_path('tangled.util:dir', 'file')
self.assertTrue(path.endswith('/tangled/tangled/util/dir/file'))
def test_asset_path_with_rel_path(self):
path = util.asset_path('tangled.util', 'dir/file')
self.assertTrue(path.endswith('/tangled/tangled/util/dir/file'))
