Added overwrite option for remove_comments

Author: manjavacas

melkit/toolkit.py

@@ -508,14 +508,15 @@ def available_to_csv(self, obj_list: List[Object], title='./available.csv') -> D
 
         return df
 
-    def remove_comments(self, new_file: str = None) -> None:
+    def remove_comments(self, overwrite=False, new_file: str = None) -> None:
         '''
         Remove comments from input file.
         '''
 
+        src_file = self._filename
         new_file = new_file or self._filename + '_NEW'
 
-        with open(self._filename, 'r') as f1, open(new_file, 'w') as f2:
+        with open(src_file, 'r') as f1, open(new_file, 'w') as f2:
             for line in f1:
                 if line.startswith('*') and '*EOR*' not in line:
                     f2.write('')
@@ -524,6 +525,11 @@ def remove_comments(self, new_file: str = None) -> None:
                 else:
                     f2.write(line)
 
+        if overwrite:
+            remove(src_file)
+            rename(new_file, src_file)
+
+
     def id_search(self, obj_list: List[Object], id: str) -> Object:
         '''
         Searches for an input object (CV, FL...) by its ID in a list of input elements.