From 7afb735a68d47d2791c33ea1110bdd80a840baac Mon Sep 17 00:00:00 2001
From: Thomas Reifenberger <tom-mi@users.noreply.github.com>
Date: Sat, 14 Nov 2020 13:26:50 +0100
Subject: [PATCH] Add num_files field
---
mkp/__init__.py | 1 +
test/integration/test_mkp.py | 6 ++++++
2 files changed, 7 insertions(+)
diff --git a/mkp/__init__.py b/mkp/__init__.py
index 4758688..9a152b3 100644
--- a/mkp/__init__.py
+++ b/mkp/__init__.py
@@ -28,6 +28,7 @@ def dist(info, path=None):
path = os.path.dirname(os.path.realpath(main.__file__))
info['files'] = find_files(path)
+ info['num_files'] = sum(len(file_list) for file_list in info['files'].values())
dist_dir = os.path.join(path, _DIST_DIR)
filename = '{}-{}.mkp'.format(info['name'], info['version'])
diff --git a/test/integration/test_mkp.py b/test/integration/test_mkp.py
index 7aee282..90c3e77 100644
--- a/test/integration/test_mkp.py
+++ b/test/integration/test_mkp.py
@@ -137,6 +137,8 @@ def test_dist(tmpdir):
'author': 'John Doe',
'name': 'foo',
'version': '42',
+ 'version.min_required': '1.2.6p5',
+ 'version.usable_until': None,
}
mkp.dist(info, str(tmpdir))
@@ -147,8 +149,11 @@ def test_dist(tmpdir):
assert package.info['name'] == 'foo'
assert package.info['files']['agents'] == ['special/agent_test']
assert package.info['files']['checks'] == ['foo']
+ assert package.info['num_files'] == 2
assert package.info['version'] == '42'
assert package.info['version.packaged'] == 'python-mkp'
+ assert package.info['version.min_required'] == '1.2.6p5'
+ assert package.info['version.usable_until'] is None
def test_dist_uses_script_path_when_no_path_is_given(tmpdir):
@@ -179,3 +184,4 @@ dist({
assert package.info['files']['checks'] == ['foo']
assert package.info['version'] == '42'
assert package.info['version.packaged'] == 'python-mkp'
+ assert package.info['num_files'] == 2
\ No newline at end of file
--
GitLab