Add method to update multiple key/values to .env

Following things were changed.
* `update_dict_to_dotenv` was added to `main.py`
* `make_env_line` was extracted from `set_key`
* `test_update_dict_to_dotenv` was added to `test_main.py`

Author: mrjng

src/dotenv/__init__.py

@@ -1,5 +1,5 @@
 from .compat import IS_TYPE_CHECKING
-from .main import load_dotenv, get_key, set_key, unset_key, find_dotenv, dotenv_values
+from .main import load_dotenv, get_key, set_key, unset_key, find_dotenv, dotenv_values, update_dict_to_dotenv
 
 if IS_TYPE_CHECKING:
     from typing import Any, Optional
@@ -43,4 +43,5 @@ def get_cli_string(path=None, action=None, key=None, value=None, quote=None):
            'set_key',
            'unset_key',
            'find_dotenv',
-           'load_ipython_extension']
+           'load_ipython_extension',
+           'update_dict_to_dotenv']

src/dotenv/main.py

@@ -143,30 +143,40 @@ def rewrite(path):
         shutil.move(dest.name, path)
 
 
-def set_key(dotenv_path, key_to_set, value_to_set, quote_mode="always", export=False):
-    # type: (_PathLike, Text, Text, Text, bool) -> Tuple[Optional[bool], Text, Text]
+def make_env_line(key, value, quote_mode="always", export=False):
+    # type: (Text, Text, Text, bool) -> Text
     """
-    Adds or Updates a key/value to the given .env
-
-    If the .env path given doesn't exist, fails instead of risking creating
-    an orphan .env somewhere in the filesystem
+    Make a line which format fits to .env
     """
     if quote_mode not in ("always", "auto", "never"):
         raise ValueError("Unknown quote_mode: {}".format(quote_mode))
 
     quote = (
         quote_mode == "always"
-        or (quote_mode == "auto" and not value_to_set.isalnum())
+        or (quote_mode == "auto" and not value.isalnum())
     )
 
     if quote:
-        value_out = "'{}'".format(value_to_set.replace("'", "\\'"))
+        value_out = "'{}'".format(value.replace("'", "\\'"))
     else:
-        value_out = value_to_set
+        value_out = value
     if export:
-        line_out = 'export {}={}\n'.format(key_to_set, value_out)
+        line_out = 'export {}={}\n'.format(key, value_out)
     else:
-        line_out = "{}={}\n".format(key_to_set, value_out)
+        line_out = "{}={}\n".format(key, value_out)
+
+    return line_out
+
+
+def set_key(dotenv_path, key_to_set, value_to_set, quote_mode="always", export=False):
+    # type: (_PathLike, Text, Text, Text, bool) -> Tuple[Optional[bool], Text, Text]
+    """
+    Adds or Updates a key/value to the given .env
+
+    If the .env path given doesn't exist, fails instead of risking creating
+    an orphan .env somewhere in the filesystem
+    """
+    line_out = make_env_line(key_to_set, value_to_set, quote_mode, export)
 
     with rewrite(dotenv_path) as (source, dest):
         replaced = False
@@ -356,3 +366,29 @@ def dotenv_values(
         override=True,
         encoding=encoding,
     ).dict()
+
+
+def update_dict_to_dotenv(dotenv_path, env_dict, quote_mode="always", export=False):
+    # type: (_PathLike, Dict[Text, Optional[Text]], Text, bool) -> None
+    """
+    Adds or Updates key/value pairs in the given dictionary to the given .env
+
+    If the .env path given doesn't exist, fails instead of risking creating
+    an orphan .env somewhere in the filesystem
+    """
+    key_to_line = {}
+
+    for key_to_set, value_to_set in env_dict.items():
+        env_line = make_env_line(key_to_set, value_to_set, quote_mode, export)
+        key_to_line[key_to_set] = env_line
+
+    with rewrite(dotenv_path) as (source, dest):
+        for mapping in with_warn_for_invalid_lines(parse_stream(source)):
+            if mapping.key in key_to_line:
+                line_out = key_to_line.pop(mapping.key)
+                dest.write(line_out)
+            else:
+                dest.write(mapping.original.string)
+
+        for _, line_out in key_to_line.items():
+            dest.write(line_out)

tests/test_main.py

@@ -367,3 +367,25 @@ def test_dotenv_values_stream(env, string, interpolate, expected):
         result = dotenv.dotenv_values(stream=stream, interpolate=interpolate)
 
         assert result == expected
+
+
+@pytest.mark.parametrize(
+    "before,env_dict,after",
+    [
+        ("", {"a1": "", "a2": "b", "a3": "'b'", "a4": "\"b\""},
+         "a1=''\na2='b'\na3='\\'b\\''\na4='\"b\"'\n"),
+        ("", {"a1": "b'c", "a2": "b\"c"}, "a1='b\\'c'\na2='b\"c'\n"),
+        ("a=b\nb=c\n", {"b": "cc", "c": "d", "d": "e"},
+         "a=b\nb='cc'\nc='d'\nd='e'\n")
+    ],
+)
+def test_update_dict_to_dotenv(dotenv_file, before, env_dict, after):
+    logger = logging.getLogger("dotenv.main")
+    with open(dotenv_file, "w") as f:
+        f.write(before)
+
+    with mock.patch.object(logger, "warning") as mock_warning:
+        dotenv.update_dict_to_dotenv(dotenv_file, env_dict)
+
+    assert open(dotenv_file, "r").read() == after
+    mock_warning.assert_not_called()