add new function

Author: itzCozi

PDL/package/pdlparse/__init__.py

@@ -1,3 +1,3 @@
 #https://www.freecodecamp.org/news/build-your-first-python-package/
 from parse import scrap_library, get_variable, filter_file, get_strings, get_class, get_type, get_value, get_comments
-from library import get_libs, get_lib_info, get_main_lib, read_lib, format_lib
+from library import get_libs, get_lib_info, get_main_lib, read_lib, format_lib, remove_comments

PDL/package/pdlparse/library.py

@@ -84,34 +84,66 @@ def format_lib(library):
         with open(library, 'w') as edit:
           replacement = file_content.replace('}', '};')
           edit.write(replacement)
+          continue
     if '{' in line:
       index = line.find('{')
       name = line[0:index-1]
       if not 'class' in line:
         with open(library, 'w') as edit:
           replacement = file_content.replace(f'{name}', f'class {name}')
           edit.write(replacement)
+          continue
     if 'int' in line:
       if "'" in line:
         index_var = line.find("'")
         variable = line[index_var+1:-1]
         with open(library, 'w') as edit:
           replacement = file_content.replace(f"'{variable}'", f"{variable}")
           edit.write(replacement)
+          continue
     if 'str' in line:
       if not "'" in line:
         index_var = line.find('=')
         variable = line[index_var+2:]
         with open(library, 'w') as edit:
           replacement = file_content.replace(f"{variable}", f"'{variable}'")
           edit.write(replacement)
+          continue
     if 'flt' in line:
       if "'" in line:
         index_var = line.find("'")
         variable = line[index_var+1:-1]
         with open(library, 'w') as edit:
           replacement = file_content.replace(f"'{variable}'", f"{variable}")
           edit.write(replacement)
+          continue
+    if 'raw' in line:
+      if not "'" or '/' in line:
+        index_var = line.find('=')
+        variable = line[index_var+2:]
+        with open(library, 'w') as edit:
+          replacement = file_content.replace(f"{variable}", f"/'{variable}'/")
+          edit.write(replacement)
+          continue
     else:
       pass
 
+
+def remove_comments(library):
+  if not os.path.exists(library):
+    raise FileNotFoundError(f'ERROR: Package pdlparse cannot find file {library}.')
+
+  with open(library, 'r+') as file:
+    retlist = []
+    file_content = file.read()
+  for line in file_content.split('\n'):
+    if "//" in line:
+      index = line.find("//")
+      string = line[index:].replace("// ", "")
+      retlist.append(string)
+    for item in retlist:
+      with open(library, 'w') as edit:
+        replacement = file_content.replace(string, '')
+        edit.write(replacement)
+  return retlist
+