fix bug that existing template breaks

2017-09-23 17:14:04 +09:00 · 2017-09-23 17:14:04 +09:00 · 0a4c2301c7
commit 0a4c2301c7
parent 316a638d9e
1 changed files with 2 additions and 1 deletions
--- a/scripts/scaffold.py
+++ b/scripts/scaffold.py
@ -319,8 +319,9 @@ def insert_code_to_class(path, base_class, new_code):

    lines = lines[:end_line_no] + func_lines + lines[end_line_no:]

+    body = '\n'.join(lines) + '\n'
    with open(path, 'w') as f:
-        f.write('\n'.join(lines))
+        f.write(body)


 def insert_query_codes(service, operation):