Fix getting license in post_gen hook

* The solution? Put it in some random txt file in the license folder and read it in. You can't use """""" or "" or '' without some kind of error. By reading in the text instead, we completely avoid this.

Signed-off-by: Andrew-Chen-Wang <[email protected]>

hooks/post_gen_project.py

@@ -319,18 +319,23 @@ def handle_licenses():
         "GNU Lesser General Public License v3.0": "COPYING.LESSER",
         "The Unlicense": "UNLICENSE",
     }
+    with open(os.path.join("licenses", "-temporary-placeholder.txt")) as f:
+        selected_title = f.readline()
     for filename in os.listdir("licenses"):
+        if filename == "-temporary-placeholder.txt":
+            continue
         # You'll always see: '---\n' marking beginning + end of Jekyll format
         with open(os.path.join("licenses", filename)) as f:
             contents = f.readlines()
-        title = contents[1].replace("title: ", "").replace("\n", "")
-        if title != "{{ cookiecutter.open_source_license }}":
+        title = contents[1].replace("title: ", "").rstrip()
+        if title != selected_title:
             continue
         with open(special_license_files.get(title, "LICENSE"), "w") as f:
             # +2 to get rid of the --- and and an extra new line
             f.writelines(contents[contents.index("---\n", 1) + 2 :])
         break
-    if "{{ cookiecutter.open_source_license }}" == "Not open source":
+
+    if selected_title == "Not open source":
         os.remove("CONTRIBUTORS.txt")
     shutil.rmtree("licenses")
 

tests/test_cookiecutter_generation.py

@@ -29,6 +29,8 @@ def generate_license_file_titles():
     directory = os.path.join("{{cookiecutter.project_slug}}", "licenses")
     titles = []
     for file in os.listdir(directory):
+        if file == "-temporary-placeholder.txt":
+            continue
         with open(os.path.join(directory, file)) as f:
             titles.append(f.readlines()[1].replace("title: ", "").replace("\n", ""))
     return titles

{{cookiecutter.project_slug}}/licenses/-temporary-placeholder.txt

@@ -0,0 +1 @@
+{{ cookiecutter.open_source_license }}