Make generated files executable when they have shebang (kyuridenamida#138)

kmyk · kyuridenamida · commit a574ed349db8 · 2019-04-11T00:02:06.000+09:00
diff --git a/atcodertools/fileutils/create_contest_file.py b/atcodertools/fileutils/create_contest_file.py
@@ -1,4 +1,5 @@
 import os
+import stat
 from typing import List
 
 from atcodertools.client.models.sample import Sample
@@ -13,6 +14,9 @@ def _make_text_file(file_path, text):
 def create_code(source_code: str,
                 file_path: str):
     _make_text_file(file_path, source_code)
+    if source_code.startswith('#!'):
+        st = os.stat(file_path)
+        os.chmod(file_path, st.st_mode | stat.S_IEXEC)
 
 
 def create_example(example: Sample, in_example_name: str, out_example_name: str):
