Fix create project error

nietsuu · Aug 1, 2023 · 5f28910 · 5f28910
1 parent a9954ae
commit 5f28910
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/create_pipenv_project/__init__.py b/create_pipenv_project/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.2"
+__version__ = "0.3"
 
 import os
 import shutil
@@ -76,7 +76,11 @@ def _copy_user_files(self, project_name: str) -> None:
         user_files_dirpath = os.path.join(cpp_dirpath, "user_files")
 
         for filename in os.listdir(user_files_dirpath):
-            paste_dir = mapping[filename]
+            try:
+                paste_dir = mapping[filename]
+            except KeyError:
+                continue
+
             shutil.copy(
                 os.path.join(user_files_dirpath, filename),
                 paste_dir,