Removed pretty print as an attribute of Csskrt

Author: 4d11

csskrt/bootstrapCsskrt.py

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

csskrt/bulmaCsskrt.py

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

csskrt/csskrt.py

@@ -5,11 +5,10 @@
 
 
 class Csskrt(ABC):
-    def __init__(self, filename: str, pretty_print, tag_styles: Dict):
+    def __init__(self, filename: str, 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
@@ -149,7 +148,6 @@ def add_list_classes(self, list_tags: dict) -> NoReturn:
                     # recursive=False to prevent double modifying for nested lists
                     self.add_class_to_element(li, list_tags['li'])
 
-
     def add_general_classes(self):
         """
         Adds styles to single elements
@@ -163,7 +161,7 @@ def add_general_classes(self):
                 for elem in self.soup.find_all(tag):
                     self.add_class_to_element(elem, self.tag_styles[tag])
 
-    def output(self) -> NoReturn:
+    def output(self, pretty_print: bool) -> NoReturn:
         """
         Outputs a new file.
         :return:
@@ -174,7 +172,7 @@ def output(self) -> NoReturn:
 
         new_file_name = os.path.join(folder, 'output/csskrt_' + file_name + ext)
         with open(new_file_name, 'w') as out_file:
-            if self.pretty_print:
+            if pretty_print:
                 out_file.write(str(self.soup))
             else:
                 out_file.write(self.soup.prettify())
@@ -206,4 +204,4 @@ def freshify(self) -> NoReturn:
         # Add styles for the rest of the elements
         self.add_general_classes()
 
-        return self.soup
+        return self.soup

csskrt/main.py

@@ -14,13 +14,12 @@ def freshify(filename, framework, pretty_print):
     csskrter = None
 
     if framework == 'bootstrap':
-        csskrter = BootstrapCsskrt(filename, pretty_print)
+        csskrter = BootstrapCsskrt(filename)
     elif framework == 'bulma':
-        csskrter = BulmaCsskrt(filename, pretty_print)
-
+        csskrter = BulmaCsskrt(filename)
 
     csskrter.freshify()
-    csskrter.output()
+    csskrter.output(pretty_print)
     print()
     print("~~~ Done! ~~~~")