瀏覽代碼

Add a config entry for BORG_CHECK_I_KNOW_WHAT_I_AM_DOING env var (#724).

Reviewed-on: https://projects.torsion.org/borgmatic-collective/borgmatic/pulls/724
Dan Helfman 1 年之前
父節點
當前提交
d29667ca3c
共有 3 個文件被更改,包括 26 次插入2 次删除
  1. 13 2
      borgmatic/borg/environment.py
  2. 6 0
      borgmatic/config/schema.yaml
  3. 7 0
      tests/unit/borg/test_environment.py

+ 13 - 2
borgmatic/borg/environment.py

@@ -11,11 +11,15 @@ OPTION_TO_ENVIRONMENT_VARIABLE = {
     'temporary_directory': 'TMPDIR',
 }
 
-DEFAULT_BOOL_OPTION_TO_ENVIRONMENT_VARIABLE = {
+DEFAULT_BOOL_OPTION_TO_DOWNCASE_ENVIRONMENT_VARIABLE = {
     'relocated_repo_access_is_ok': 'BORG_RELOCATED_REPO_ACCESS_IS_OK',
     'unknown_unencrypted_repo_access_is_ok': 'BORG_UNKNOWN_UNENCRYPTED_REPO_ACCESS_IS_OK',
 }
 
+DEFAULT_BOOL_OPTION_TO_UPPERCASE_ENVIRONMENT_VARIABLE = {
+    'check_i_know_what_i_am_doing': 'BORG_CHECK_I_KNOW_WHAT_I_AM_DOING',
+}
+
 
 def make_environment(storage_config):
     '''
@@ -33,8 +37,15 @@ def make_environment(storage_config):
     for (
         option_name,
         environment_variable_name,
-    ) in DEFAULT_BOOL_OPTION_TO_ENVIRONMENT_VARIABLE.items():
+    ) in DEFAULT_BOOL_OPTION_TO_DOWNCASE_ENVIRONMENT_VARIABLE.items():
         value = storage_config.get(option_name, False)
         environment[environment_variable_name] = 'yes' if value else 'no'
 
+    for (
+        option_name,
+        environment_variable_name,
+    ) in DEFAULT_BOOL_OPTION_TO_UPPERCASE_ENVIRONMENT_VARIABLE.items():
+        value = storage_config.get(option_name, False)
+        environment[environment_variable_name] = 'YES' if value else 'NO'
+
     return environment

+ 6 - 0
borgmatic/config/schema.yaml

@@ -405,6 +405,12 @@ properties:
                     Bypass Borg error about a previously unknown unencrypted
                     repository. Defaults to false.
                 example: true
+            check_i_know_what_i_am_doing:
+                type: boolean
+                description: |
+                    Bypass Borg confirmation about check with repair option.
+                    Defaults to false.
+                example: true
             extra_borg_options:
                 type: object
                 additionalProperties: false

+ 7 - 0
tests/unit/borg/test_environment.py

@@ -25,6 +25,7 @@ def test_make_environment_without_configuration_should_only_set_default_environm
     assert environment == {
         'BORG_RELOCATED_REPO_ACCESS_IS_OK': 'no',
         'BORG_UNKNOWN_UNENCRYPTED_REPO_ACCESS_IS_OK': 'no',
+        'BORG_CHECK_I_KNOW_WHAT_I_AM_DOING': 'NO',
     }
 
 
@@ -34,6 +35,12 @@ def test_make_environment_with_relocated_repo_access_should_override_default():
     assert environment.get('BORG_RELOCATED_REPO_ACCESS_IS_OK') == 'yes'
 
 
+def test_make_environment_check_i_know_what_i_am_doing_should_override_default():
+    environment = module.make_environment({'check_i_know_what_i_am_doing': True})
+
+    assert environment.get('BORG_CHECK_I_KNOW_WHAT_I_AM_DOING') == 'YES'
+
+
 def test_make_environment_with_integer_variable_value():
     environment = module.make_environment({'borg_files_cache_ttl': 40})
     assert environment.get('BORG_FILES_CACHE_TTL') == '40'