Add support for stats --mode flag

e2e/test.py

@@ -73,7 +73,7 @@ def emit(self, record):
     logger.fatal('Expected to restored file to contain %s (got %s)',
                  RESTORED_DATA_EXPECTED, RESTORED_DATA_ACTUAL)
 
-stats = restic.stats()
+stats = restic.stats(mode='blobs-per-file')
 logger.info('repo stats: %s', stats)
 if stats['total_size'] != len(RESTORED_DATA_EXPECTED):
     logger.fatal('Expected to total size of %d (got %d)',

restic/__init__.py

@@ -43,8 +43,8 @@ def self_update():
     return internal_self_update.run(_make_base_command())
 
 
-def stats():
-    return internal_stats.run(_make_base_command())
+def stats(*args, **kwargs):
+    return internal_stats.run(_make_base_command(), *args, **kwargs)
 
 
 def unlock():

restic/internal/stats.py

@@ -3,6 +3,10 @@
 from restic.internal import command_executor
 
 
-def run(restic_base_command):
+def run(restic_base_command, mode=None):
     cmd = restic_base_command + ['stats']
+
+    if mode:
+        cmd.extend(['--mode', mode])
+
     return json.loads(command_executor.execute(cmd))

restic/internal/stats_test.py

@@ -17,5 +17,19 @@ def test_stats(self, mock_execute):
             "total_file_count": 3
             }
             """
-        restic.stats()
+
+        self.assertEqual({
+            'total_size': 20,
+            'total_file_count': 3
+        }, restic.stats())
+
         mock_execute.assert_called_with(['restic', '--json', 'stats'])
+
+    @mock.patch.object(stats.command_executor, 'execute')
+    def test_stats_with_mode(self, mock_execute):
+        mock_execute.return_value = '{}'
+
+        restic.stats(mode='blobs-per-file')
+
+        mock_execute.assert_called_with(
+            ['restic', '--json', 'stats', '--mode', 'blobs-per-file'])