def test_filesystem(tmp_path, database):
"""Test loading and using a queue with the filesystem database.
"""
from rse.main import Encyclopedia
config_dir = os.path.join(str(tmp_path), "software")
os.mkdir(config_dir)
config_file = os.path.join(config_dir, "rse.ini")
# System exit if doesn't exist
with pytest.raises(SystemExit):
enc = Encyclopedia(config_file=config_file)
enc = Encyclopedia(config_file=config_file,
generate=True,
database=database)
assert enc.config.configfile == config_file
assert os.path.exists(config_dir)
assert enc.config_dir == config_dir
assert enc.config.configfile == os.path.join(enc.config_dir, "rse.ini")
assert enc.database == database
assert enc.db.database == database
if database == "filesystem":
assert enc.db.data_base == os.path.join(config_dir, "database")
# Test list, empty without anything
assert not enc.list()
# Add a repo
repo = enc.add("github.com/singularityhub/sregistry")
assert len(enc.list()) == 1
# enc.get should return last repo, given no id
lastrepo = enc.get()
assert lastrepo.uid == repo.uid
# Summary
enc.summary()
enc.summary("github.com/singularityhub/sregistry")
enc.analyze("github.com/singularityhub/sregistry")
enc.analyze_bulk()
# Clean up a specific repo (no prompt)
enc.clear(repo.uid, noprompt=True)
assert len(enc.list()) == 0
enc.clear(noprompt=True)
assert not enc.list()
# Get the taxonomy or criteria
enc.list_taxonomy()
enc.list_criteria()
def summary(args, extra):
"""summary will return general counts for taxonomy, criteria, and users"""
client = Encyclopedia(config_file=args.config_file, database=args.database)
metrics = client.summary(args.repo)
# Export only taxonomy metrics
if args.metric_type == "taxonomy":
updated = {}
for field in ["taxonomy", "taxonomy-count", "repos"]:
updated[field] = metrics[field]
metrics = updated
# Export only criteria metrics
elif args.metric_type == "criteria":
updated = {}
for field in ["criteria", "criteria-count", "repos"]:
updated[field] = metrics[field]
metrics = updated
# Export only user metrics
elif args.metric_type == "users":
updated = {}
for field in ["users-count", "users", "repos"]:
updated[field] = metrics[field]
metrics = updated
# Print metrics as json or table
if metrics:
print(json.dumps(metrics, indent=4))