Added flag to pretty print html

Author: 4d11

bootstrapCsskrt.py

@@ -3,14 +3,14 @@
 
 
 class BootstrapCsskrt(Csskrt):
-    def __init__(self, fileName):
+    def __init__(self, fileName, pretty_print):
         tag_styles = {
             'input': 'form-control',
             'select': 'custom-select',
             'button': 'btn btn-primary',
             'checkbox': 'form-check-input',
         }
-        super().__init__(fileName, tag_styles)
+        super().__init__(fileName, pretty_print, tag_styles)
 
     def version(self):
         return "v4.1"

bulmaCsskrt.py

@@ -3,7 +3,7 @@
 
 
 class BulmaCsskrt(Csskrt):
-    def __init__(self, fileName):
+    def __init__(self, fileName, pretty_print):
         tag_styles = {
             'input': 'input',
             'label': 'label',
@@ -19,7 +19,7 @@ def __init__(self, fileName):
             'h5': 'title is-5',
             'h6': 'title is-6'
         }
-        super().__init__(fileName, tag_styles)
+        super().__init__(fileName, pretty_print, tag_styles)
 
     def version(self):
         return "v0.7.1"

csskrt.py

@@ -5,10 +5,11 @@
 
 
 class Csskrt(ABC):
-    def __init__(self, filename: str, tag_styles: Dict):
+    def __init__(self, filename: str, pretty_print, tag_styles: Dict):
         f = open(filename)  # should be able to handle dirs (for later) todo
         f_data = f.read()
 
+        self.pretty_print = pretty_print
         self.file_path = filename
         self.soup = BeautifulSoup(f_data, 'html.parser')
         self.tag_styles = tag_styles
@@ -173,8 +174,10 @@ def output(self) -> NoReturn:
 
         new_file_name = os.path.join(folder, 'csskrt_' + file_name + ext)
         with open(new_file_name, 'w') as out_file:
-            out_file.write(self.soup.prettify())
-
+            if self.pretty_print:
+                out_file.write(str(self.soup))
+            else:
+                out_file.write(self.soup.prettify())
 
     def freshify(self) -> NoReturn:
         """
@@ -205,3 +208,5 @@ def freshify(self) -> NoReturn:
 
         # Output the modified html file
         self.output()
+
+        return self.soup

main.py

@@ -5,19 +5,22 @@
 
 @click.command()
 @click.argument('filename', type=click.Path(exists=True))
-@click.option('--framework', default='bulma', help='Name of the framework')
-def freshify(filename, framework):
+@click.option('--framework', '-f', default='bulma',
+              type=click.Choice(['bulma', 'bootstrap']),
+              help='Name of the framework. De')
+@click.option('--pretty-print', '-p', is_flag=True)
+def freshify(filename, framework, pretty_print):
     csskrter = None
 
     if framework == 'bootstrap':
-        csskrter = BootstrapCsskrt(filename)
+        csskrter = BootstrapCsskrt(filename, pretty_print)
     elif framework == 'bulma':
-        csskrter = BulmaCsskrt(filename)
-    else:
-        raise Exception("Invalid framework")
+        csskrter = BulmaCsskrt(filename, pretty_print)
+
 
     csskrter.freshify()
-    print("Done!")
+    print()
+    print("~~~ Done! ~~~~")
 
 
 if __name__ == '__main__':