From b9802923627cdb66ff80d0a39e5f5f5ce247e8aa Mon Sep 17 00:00:00 2001
From: Arun Babu Neelicattu <arun.neelicattu@gmail.com>
Date: Mon, 5 Oct 2020 01:23:16 +0200
Subject: [PATCH] init: handle pyproject exceptions for existing file

Resolves: #3073
---
 poetry/console/commands/init.py     |  3 ++-
 tests/console/commands/test_init.py | 36 +++++++++++++++++++++++++++++
 2 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/poetry/console/commands/init.py b/poetry/console/commands/init.py
index d6817bee975..af72318c31a 100644
--- a/poetry/console/commands/init.py
+++ b/poetry/console/commands/init.py
@@ -13,6 +13,7 @@
 from cleo import option
 from tomlkit import inline_table
 
+from poetry.core.pyproject import PyProjectException
 from poetry.core.pyproject.toml import PyProjectTOML
 from poetry.utils._compat import OrderedDict
 from poetry.utils._compat import Path
@@ -378,7 +379,7 @@ def _parse_requirements(
 
         try:
             cwd = self.poetry.file.parent
-        except RuntimeError:
+        except (PyProjectException, RuntimeError):
             cwd = Path.cwd()
 
         for requirement in requirements:
diff --git a/tests/console/commands/test_init.py b/tests/console/commands/test_init.py
index 55c0b41ff11..305aec88bd4 100644
--- a/tests/console/commands/test_init.py
+++ b/tests/console/commands/test_init.py
@@ -623,6 +623,42 @@ def test_init_existing_pyproject_simple(
     )
 
 
+def test_init_non_interactive_existing_pyproject_add_dependency(
+    tester, source_dir, init_basic_inputs, repo
+):
+    pyproject_file = source_dir / "pyproject.toml"
+    existing_section = """
+[tool.black]
+line-length = 88
+"""
+    pyproject_file.write_text(decode(existing_section))
+
+    repo.add_package(get_package("foo", "1.19.2"))
+
+    tester.execute(
+        "--author 'Your Name <you@example.com>' "
+        "--name 'my-package' "
+        "--python '^3.6' "
+        "--dependency foo",
+        interactive=False,
+    )
+
+    expected = """\
+[tool.poetry]
+name = "my-package"
+version = "0.1.0"
+description = ""
+authors = ["Your Name <you@example.com>"]
+
+[tool.poetry.dependencies]
+python = "^3.6"
+foo = "^1.19.2"
+
+[tool.poetry.dev-dependencies]
+"""
+    assert "{}\n{}".format(existing_section, expected) in pyproject_file.read_text()
+
+
 def test_init_existing_pyproject_with_build_system_fails(
     tester, source_dir, init_basic_inputs
 ):