parser now defines undefined style rules in css files!

Author: Jack Adam

README.md

@@ -7,8 +7,8 @@
 2. prints undefined classes and IDs, corresponding files
 3. prints unused classes and IDs, corresponding files and line numbers
 4. re-writes .css files without unused classes and IDs
+5. defines previously undefined style rules in .css files
 
 TODO:
 
-0. thoroughly edgecase re-writing stage
-1. add functionality to write empty style rules for undefined rules?
+0. thoroughly edge-case re-writing stage

definer.py

@@ -0,0 +1,25 @@
+from helper import lookup
+from os import path
+
+def define(un, cf):
+    up = {}
+    for rule, file in un.items():
+        # get name of css file
+        css = lookup(cf, file)
+        # if already cleaned, open new file
+        newF = css[:-4] + '-clean.css'
+        css = css if not path.isfile(newF) else newF
+        # count number of definitions
+        if css not in up:
+            up[css] = 1
+        else:
+            up[css] += 1
+        with open(css) as og:
+            data = og.read()
+        # prepend new rules
+        with open(css, 'w') as new:
+            new.write(rule + ' {\n\t/* Define your style rule here */\n}\n' + data)
+    # print changes
+    for file, count in up.items():
+        r = 'rules' if count > 1 else 'rule'
+        print(f'Defined {count} new {r} in {file}')

demo.png

@@ -68,3 +68,9 @@ def solo(line):
 # removes comma if last character of string
 def comma(w):
     return w[:-1] if w[-1:] == ',' else w
+
+# look up html file in dictionary of html and css files
+def lookup(d, f):
+    for h, c in d.items():
+        if h == f:
+            return c

helper.py

@@ -1,40 +1,46 @@
 from parser import parse_css, parse_html
 from cleaner import clean
+from definer import define
 from sys import exit
 
 def main():
     # parse the files
     c = parse_css()
     h = parse_html()
-    unused, fileNames, css, fileCount = {}, [], [], 0
+    unused, undefined, results, fileNames, css, fileCount = {}, {}, '', [], [], 0
 
     # identify unused classes
     for cla, num in c[0][0].items():
         x = cla.split()
         if ':' not in x[0]:
             css.append(x[0])
-        if x[0] not in h[0][0]:
+        if x[0] not in h[0][0][0]:
             unused[cla] = num
 
     # identify unused IDs
     for ID, num in c[1][0].items():
         y = ID.split()
         if ':' not in y[0]:
             css.append(y[0])
-        if y[0] not in h[1][0]:
+        if y[0] not in h[0][1][0]:
             unused[ID] = num
 
-    print(f'\nIdentified {c[0][1]} unique classes and {c[1][1]} unique IDs.\n')
+    i = f'Identified {c[0][1]} unique classes and {c[1][1]} unique IDs.\n'
+    print('\n' + i)
+    results += i
 
     # identify undefined classes and IDs
-    for d in h:
+    for d in h[0]:
         for dd in d:
-            for rule, num in dd.items():
+            for rule, file in dd.items():
                 if rule not in css:
+                    undefined[rule] = file
                     pre = 'ID:   ' if rule[0] == '#' else 'class:'
-                    print(f'Undefined {pre} {rule} : {num}')
+                    o = f'Undefined {pre} {rule} : {file}'
+                    print(o)
+                    results += '\n' + o
     print()
-
+    results += '\n'
     final = dict(unused)
 
     # identify pseudoclasses
@@ -48,21 +54,38 @@ def main():
             if zz[0] + ' : ' + z[2] not in unused:
                 del final[rule]
                 continue
+        o = ''
         if z[0][0] == '.':
-            print(f'Unused class:    {rule}{num}')
+            o = f'Unused class:    {rule}{num}'
         elif z[0][0] == '#':
-            print(f'Unused ID:       {rule}{num}')
+            o = f'Unused ID:       {rule}{num}'
+        print(o)
+        results += '\n' + o
 
     if not final:
-        print('No unused classes nor IDs!')
+        o = 'No unused classes nor IDs!'
+        print(o)
+        resuls += i
     else:
         q = input('\nMay I remove these unused rules and output new .css files? (yes/no): ')
         if q.lower() in ('yes', 'y'):
             clean(final, fileNames, fileCount)
-        elif q.lower() in ('no', 'n'):
-            exit('Thank you.')
-        else:
-            exit('Invalid response.')
+        if undefined:
+            qq = input('May I add definitions for undefined rules? (yes/no): ')
+            if qq.lower() in ('yes', 'y'):
+                define(undefined, h[1])
+            elif qq.lower() in ('no', 'n'):
+                qqq = input('Would you instead like a .txt file with your results? (yes/no): ')
+                if qqq.lower() in ('yes', 'y'):
+                    with open('results.txt', 'w') as f:
+                        f.write(results)
+                        print('Wrote results.txt')
+                elif qqq.lower() in ('no', 'n'):
+                    exit('Thank you.')
+                else:
+                    exit('Invalid response.')
+            else:
+                exit('Invalid response.')
 
 if __name__ == '__main__':
     main()

main.py

@@ -61,7 +61,7 @@ def parse_css():
 def parse_html():
     # obtain filename/path
     filepath = intro('html')
-    cl, id, fc = [], [], 0
+    cl, id, css, fc = [], [], {}, 0
 
     # read each file
     for file in glob(filepath):
@@ -82,20 +82,28 @@ def parse_html():
                     # found a class
                     elif piece[:5] == 'class':
                         start, found = 7, '.'
+                    # found a link
+                    elif piece[:4] == 'href':
+                        start = 6
                     if start:
                         for char in piece[start:]:
                             if char != "'" and char != '"':
                                 found += char
-                            # at end of rule name
+                            # at end, marked by quotes
                             else:
                                 if start == 4:
                                     id[fc][found] = file
                                 elif start == 7:
                                     cl[fc][found] = file
+                                # if link to .css file
+                                elif start == 6 and found[-4:] == '.css':
+                                    # css.append(file + ' ' + found)
+                                    if file not in css:
+                                        css[file] = found
                                 break
         fc += 1
 
     if not cl and not id:
         exit('No .html files in this directory!')
     else:
-        return (cl, id)
+        return ((cl, id), css)