Running module should back up existing files only

Also reduces NULL padding of new files
rdipardo · Jan 8, 2022 · 41a8116 · 41a8116
1 parent 7af3547
commit 41a8116
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/rplugin/pythonx/cpywrite/__main__.py b/rplugin/pythonx/cpywrite/__main__.py
@@ -57,15 +57,15 @@ def main():
                 if license_text:
                     if not path.exists(generator.out_file):
                         with open(generator.out_file, 'w') as src:
-                            src.truncate(1024)
+                            src.truncate(8)
 
                         print("Created new %s file: %s"
                               % (generator.lang, generator.out_file))
-
-                    with open(generator.out_file + '.bak', 'w') as bak:
-                        with open(generator.out_file, 'rt') as source:
-                            source_file = source.read()
-                            bak.write(source_file)
+                    else:
+                        with open(generator.out_file + '.bak', 'w') as bak:
+                            with open(generator.out_file, 'rt') as source:
+                                source_file = source.read()
+                                bak.write(source_file)
 
                     _, tmp_source = tempfile.mkstemp(text=True)