From 7359a4799d3505bd072fbbabef8241e399558ff7 Mon Sep 17 00:00:00 2001
From: OSAMATAMER <59906580+OSAMATAMER@users.noreply.github.com>
Date: Fri, 1 Sep 2023 13:33:46 +0300
Subject: [PATCH 1/7] Added if statement modification

---
 README.md                 | 15 ++++++++++++--
 html2django/djangohtml.py | 43 +++++++++++++++++++++++++--------------
 2 files changed, 41 insertions(+), 17 deletions(-)
diff --git a/README.md b/README.md
index 23b372f..a1d4bc8 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,7 @@
 
 ## Description
 
-Modifies the HTML content of a file to replace all <link> tags' href attributes, all <script> tags' src attributes, and all <img> src attributes by default, also provides support for custom modification of tags and attribute with Django template tags, and adds {% load static %} to the beginning of the file.
+Modifies the HTML content of a file to replace all <link> tags' href attributes, all <script> tags' src attributes, and all <img> src attributes by default, also provides support for custom modification of tags and attribute with Django template tags, and adds {% load static %} to the beginning of the file, Also add if.
 
 
 ## Installation
@@ -52,6 +52,10 @@ Suppose you have an HTML file named index.html that looks like this:
 <body>
     <h1>Hello, World!</h1>
     <img src="image.png">
+    <div if="error_message">
+       {{ error_message }}
+       <button type="button"></button>
+    </div>
 </body>
 </html>
 ```
@@ -78,12 +82,19 @@ will modify the file to look like this:
     <body>
         <h1>Hello, World!</h1>
         <img src="{% static 'image.png' %}">
+    {% if error_message %}
+          <div>
+           {{ error_message }}
+           <button type="button">
+           </button>
+          </div>
+    {% endif %}
     </body>
     </html>
 ```
 
 Here, the <link>, <script>, and <img> tags' href and src attributes have been replaced with Django template tags, and {% load static %} has been added to the beginning of the file.
-
+Also every element with if attribute will be wrap with if tag, with the value of the attribute is the condition.
 # Parameters
 
     - file (str): The path to the input HTML file.
diff --git a/html2django/djangohtml.py b/html2django/djangohtml.py
index 5338d07..1f969f7 100644
--- a/html2django/djangohtml.py
+++ b/html2django/djangohtml.py
@@ -1,11 +1,13 @@
 from bs4 import BeautifulSoup
 
 def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: str = None,custom_value: str = None) -> None:
-    
+
     """
         Modifies the HTML content of a file to replace all <link> tags' href attributes, all <script> tags' src attributes and
         all <img> src attributes by default,also provides support for  custom  modification of  tags and attribute
-        with Django template tags, and adds "{% load static %}" to the beginning of the file. 
+        with Django template tags, and adds "{% load static %}" to the beginning of the file, Add django if statements to each element
+        that have if attribute with the val of the attribute is the condition.
+
 
         Parameters:
             - file (str): The path to the input HTML file.
@@ -20,16 +22,16 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
         Raises:
             FileNotFoundError: If the input file cannot be found or accessed.
     """
-    
+
     try:
         with open(file, 'r') as f:
             readhtml = f.readlines()
-            
+
             if readhtml[0] == "{% load static %}\n":
                 original_html = "".join(readhtml)
             else:
                 original_html = "{% load static %}" + "".join(readhtml)
-            
+
     except FileNotFoundError:
         print(f"\nThe path or name of {file} is not correct.\n")
         return 1
@@ -40,21 +42,22 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
     script = soup.find_all('script')
     img = soup.find_all('img')
     custom = soup.find_all(custom_tag)
-    
+    elements_with_if = soup.find_all(attrs={"if": True})
+
     for tag in script:
         if tag.has_attr('src') and not any(x in tag['src'] for x in ['static', 'https','http']):
             src = tag['src']
             static_src = "{% static " + f"'{src}'" + " %}"
             tag['src'] = static_src
-            
-    
+
+
     for link in links:
         if not any(x in link['href'] for x in ['static', 'https','http']):
             href = link['href']
             static_href = "{% static " + f"'{href}'" + " %}"
             link['href'] = static_href
-            
-            
+
+
     if img:
         for image in img:
             if not any(x in image['src'] for x in ['static', 'https','http']):
@@ -62,22 +65,32 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
 
                 static_img = "{% static " + f"'{img_src}'" + " %}"
                 image["src"] = static_img
-            
+
     if bool(custom_tag) and bool(custom_attr):
         for element in custom:
             if element.has_attr(custom_attr) and not any(x in element[custom_attr] for x in ['static', 'https','http']):
                 attr_val = element[custom_attr]
                 jinja_value = "{% static " + f"'{attr_val}'" + " %}"
                 element[custom_attr] = jinja_value
-            
-   
+
+
+    # edit the html to add if statements to the elements with if attribute
+    if elements_with_if:
+        for element in elements_with_if:
+            val = element['if']
+            if val:
+                if_val = "{% if " + f"{val}" + " %}"
+                element.insert_before(if_val)
+                element.insert_after("{% endif %}")
+                del element['if']
+
     # Prettifying the output with beautiful soup prettify() method
     modified_html = soup.prettify(formatter='html')
-    
+
     with open(file, 'w') as f:
         f.write(modified_html)
 
-            
+
 
 if __name__ =="__main__":
     djangoify('index.html')
\ No newline at end of file

From f77caf394be7d5d6572076b203df684dc2a611eb Mon Sep 17 00:00:00 2001
From: OSAMATAMER <59906580+OSAMATAMER@users.noreply.github.com>
Date: Sat, 2 Sep 2023 16:49:20 +0300
Subject: [PATCH 2/7] Added For loops and modification to if else system

---
 html2django/djangohtml.py | 39 +++++++++++++++++++++++++++++++--------
 1 file changed, 31 insertions(+), 8 deletions(-)

diff --git a/html2django/djangohtml.py b/html2django/djangohtml.py
index 1f969f7..c54075b 100644
--- a/html2django/djangohtml.py
+++ b/html2django/djangohtml.py
@@ -43,6 +43,7 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
     img = soup.find_all('img')
     custom = soup.find_all(custom_tag)
     elements_with_if = soup.find_all(attrs={"if": True})
+    elements_with_for = soup.find_all(attrs={"for": True})
 
     for tag in script:
         if tag.has_attr('src') and not any(x in tag['src'] for x in ['static', 'https','http']):
@@ -75,14 +76,36 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
 
 
     # edit the html to add if statements to the elements with if attribute
-    if elements_with_if:
-        for element in elements_with_if:
-            val = element['if']
-            if val:
-                if_val = "{% if " + f"{val}" + " %}"
-                element.insert_before(if_val)
-                element.insert_after("{% endif %}")
-                del element['if']
+    for element in elements_with_if:
+        val = element['if']
+        if val:
+            possible_else = element.find_next_sibling()
+            if_val = "{% if " + f"{val}" + " %}"
+            element.insert_before(if_val)
+
+            while possible_else.has_attr('elif') or possible_else.has_attr('else'):
+                if possible_else.has_attr('elif') and possible_else['elif'] != '':
+                    possible_else.insert_before("{% elif " + f"{possible_else['elif']}" + " %}")
+                    del possible_else['elif']
+                elif possible_else.has_attr('else'):
+                    possible_else.insert_before("{% else %}")
+                    del possible_else['else']
+
+                possible_else = possible_else.find_next_sibling()
+
+            possible_else.insert_before("{% endif %}")
+            del element['if']
+
+
+    for element in elements_with_for:
+        val = element['for']
+        if val != '':
+            if_val = "{% for " + f"{val}" + " %}"
+            element.insert_before(if_val)
+
+            element.insert_after("{% endfor %}")
+            del element['for']
+
 
     # Prettifying the output with beautiful soup prettify() method
     modified_html = soup.prettify(formatter='html')

From 11f9e63a18c0b94973902c71318d760af00218b0 Mon Sep 17 00:00:00 2001
From: OSAMATAMER <59906580+OSAMATAMER@users.noreply.github.com>
Date: Mon, 4 Sep 2023 16:20:38 +0300
Subject: [PATCH 3/7] Added forms modification and a_links to dynamic call the
 urls.

---
 html2django/djangohtml.py | 42 ++++++++++++++++++++++++++++++---------
 1 file changed, 33 insertions(+), 9 deletions(-)

diff --git a/html2django/djangohtml.py b/html2django/djangohtml.py
index c54075b..eafa842 100644
--- a/html2django/djangohtml.py
+++ b/html2django/djangohtml.py
@@ -1,6 +1,6 @@
 from bs4 import BeautifulSoup
 
-def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: str = None,custom_value: str = None) -> None:
+def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: str = None,custom_value: str = None, edit_form:bool = True, edit_a:bool = True) -> None:
 
     """
         Modifies the HTML content of a file to replace all <link> tags' href attributes, all <script> tags' src attributes and
@@ -59,13 +59,13 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
             link['href'] = static_href
 
 
-    if img:
-        for image in img:
-            if not any(x in image['src'] for x in ['static', 'https','http']):
-                img_src = image['src']
 
-                static_img = "{% static " + f"'{img_src}'" + " %}"
-                image["src"] = static_img
+    for image in img:
+        if not any(x in image['src'] for x in ['static', 'https','http']):
+            img_src = image['src']
+
+            static_img = "{% static " + f"'{img_src}'" + " %}"
+            image["src"] = static_img
 
     if bool(custom_tag) and bool(custom_attr):
         for element in custom:
@@ -100,13 +100,37 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
     for element in elements_with_for:
         val = element['for']
         if val != '':
-            if_val = "{% for " + f"{val}" + " %}"
-            element.insert_before(if_val)
+            for_val = "{% for " + f"{val}" + " %}"
+            element.insert_before(for_val)
 
             element.insert_after("{% endfor %}")
             del element['for']
 
 
+    if edit_form:
+        forms = soup.find_all('form')
+        for form in forms:
+            if "{% csrf_token %}" not in form.getText():
+                form.insert(0, "{% csrf_token %}")
+
+    if edit_a:
+        aElms = soup.find_all('a')
+        for a in aElms:
+            if a.has_attr('href') and not any(x in a['href'] for x in ['static', 'https','http']) and not a['href'].startswith('#') \
+                    and not a['href'].startswith('mailto') and not a['href'].startswith('{% url'):
+                href = a['href']
+                if href.startswith('/'):
+                    href = href[1:]
+                if href.endswith('/'):
+                    href = href[:-1]
+                if href.endswith('.html'):
+                    href = href[:-5]
+                if href.endswith('.htm'):
+                    href = href[:-4]
+                static_href = "{% url " + f"'{href}'" + " %}"
+                a['href'] = static_href
+
+
     # Prettifying the output with beautiful soup prettify() method
     modified_html = soup.prettify(formatter='html')
 

From fa386b8f2365a53de5292846d93678dac6b6be31 Mon Sep 17 00:00:00 2001
From: OSAMATAMER <59906580+OSAMATAMER@users.noreply.github.com>
Date: Mon, 11 Sep 2023 16:43:30 +0300
Subject: [PATCH 4/7] reusable code and more consistent code edit handling urls

---
 html2django/djangohtml.py | 210 +++++++++++++++++++++-----------------
 1 file changed, 119 insertions(+), 91 deletions(-)

diff --git a/html2django/djangohtml.py b/html2django/djangohtml.py
index eafa842..06b9ae9 100644
--- a/html2django/djangohtml.py
+++ b/html2django/djangohtml.py
@@ -1,28 +1,114 @@
 from bs4 import BeautifulSoup
 
-def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: str = None,custom_value: str = None, edit_form:bool = True, edit_a:bool = True) -> None:
 
+def edit_if(elements_with_if):
+    for element in elements_with_if:
+        val = element['dj-if']
+        if val != '':
+            possible_else = element.find_next_sibling()
+            if_val = "{% if " + f"{val}" + " %}"
+            element.insert_before(if_val)
+
+            while possible_else.has_attr('elif') or possible_else.has_attr('else'):
+                if possible_else.has_attr('elif') and possible_else['elif'] != '':
+                    possible_else.insert_before("{% elif " + f"{possible_else['elif']}" + " %}")
+                    del possible_else['elif']
+                elif possible_else.has_attr('else'):
+                    possible_else.insert_before("{% else %}")
+                    del possible_else['else']
+
+                possible_else = possible_else.find_next_sibling()
+
+            possible_else.insert_before("{% endif %}")
+            del element['dj-if']
+
+
+def edit_for(elements_with_for):
+    for element in elements_with_for:
+        val = element['dj-for']
+        if val != '':
+            for_val = "{% for " + f"{val}" + " %}"
+            element.insert_before(for_val)
+
+            element.insert_after("{% endfor %}")
+            del element['dj-for']
+
+
+def edit_url(a_elms):
+    for a in a_elms:
+        if a.has_attr('href') and not any(x in a['href'] for x in ['static', 'https', 'http']) and not a[
+            'href'].startswith('#') \
+                and not a['href'].startswith('mailto') and not a['href'].startswith('{% url') and not a['href'] == '':
+            href = a['href']
+            if href.endswith('/'):
+                href = href[:-1]
+            if href.endswith('.html'):
+                href = href[:-5]
+            if href.endswith('.htm'):
+                href = href[:-4]
+
+            last_part = href
+            if '/' in href:
+                url_parts = href.split('/')
+                # Get the last part of the URL
+                last_part = url_parts[-1]
+
+            static_href = "{% url " + f"'{last_part}'" + " %}"
+            a['href'] = static_href
+
+
+def replace_with_static(element, attr):
+    if element.has_attr(attr) and not any(x in element[attr] for x in ['static', 'https', 'http']):
+        attr_val = element[attr]
+        jinja_value = "{% static " + f"'{attr_val}'" + " %}"
+        element[attr] = jinja_value
+
+
+def djangoify(file: str, img: bool = False, custom_tag: str = None, custom_attr: str = None, custom_value: str = None,
+              edit_form: bool = True, edit_a: bool = True) -> None:
     """
-        Modifies the HTML content of a file to replace all <link> tags' href attributes, all <script> tags' src attributes and
-        all <img> src attributes by default,also provides support for  custom  modification of  tags and attribute
-        with Django template tags, and adds "{% load static %}" to the beginning of the file, Add django if statements to each element
-        that have if attribute with the val of the attribute is the condition.
+    This function modifies the HTML content of a file to integrate it with Django template tags.
+    It performs several operations such as replacing local file references with Django static template tags,
+    adding Django CSRF tokens to forms, and modifying URLs in anchor tags to use Django URL template tags.
+    Additionally, it processes custom 'dj-if' and 'dj-for' attributes in the HTML to add Django template if and for tags.
+
+    Parameters:
+    - file (str): The path to the input HTML file that needs to be modified.
+
+    - img (bool): If set to True, the function will modify the 'src' attributes of <img> tags to use Django static template tags.
+                  Defaults to False.
+
+    - custom_tag (str): Specifies a custom HTML tag that you want to modify. This parameter should be used in conjunction
+                        with 'custom_attr' and 'custom_value' parameters to define the modifications. Defaults to None.
+
+    - custom_attr (str): Specifies a custom attribute of the 'custom_tag' that you want to modify. Defaults to None.
 
+    - custom_value (str): Specifies the value that should be set for the 'custom_attr' attribute. Defaults to None.
 
-        Parameters:
-            - file (str): The path to the input HTML file.
-            - img (bool): specify True if you want the img tags to be djangofied
-            - custom_tag (str): Custom html tag to modify.To be used with the custom_attr param.
-            - custom_attr (str): custom html attribute of the custom_tag
-            - custom_value (any): the value of the custon attribute
+    - edit_form (bool): If set to True, the function will add Django CSRF tokens to all <form> tags in the HTML file.
+                        This helps in protecting against cross-site request forgery attacks. Defaults to True.
 
-        Returns:
-            None: The function modifies the input file in place.
+    - edit_a (bool): If set to True, the function will modify the 'href' attributes of <a> tags to use Django URL template tags,
+                     making it easier to manage URLs in Django projects. Defaults to True.
 
-        Raises:
-            FileNotFoundError: If the input file cannot be found or accessed.
+    Additional Information:
+    - The script processes elements with 'dj-if' attributes to add Django template if tags. The value of the 'dj-if' attribute
+      is used as the condition in the if tag. Similarly, elements with 'dj-for' attributes are processed to add Django template
+      for tags, with the value of the 'dj-for' attribute used as the loop variable and iterable.
+
+    Returns:
+    None: The function modifies the input file in place, and does not return any value.
+
+    Raises:
+    - FileNotFoundError: If the input file cannot be found or accessed, a FileNotFoundError will be raised with a descriptive message.
+
+    Example Usage:
+    >>> djangoify('index.html', img=True, custom_tag='custom-tag', custom_attr='custom-attr', custom_value='custom-value')
+
+    This will modify 'index.html' to integrate it with Django, including modifying <img> tags and custom tags as specified.
     """
 
+
     try:
         with open(file, 'r') as f:
             readhtml = f.readlines()
@@ -39,73 +125,30 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
     soup = BeautifulSoup(original_html, 'html.parser')
 
     links = soup.find_all('link', href=True)
-    script = soup.find_all('script')
-    img = soup.find_all('img')
+    scripts = soup.find_all('script')
+    img_elements = soup.find_all('img')
     custom = soup.find_all(custom_tag)
-    elements_with_if = soup.find_all(attrs={"if": True})
-    elements_with_for = soup.find_all(attrs={"for": True})
-
-    for tag in script:
-        if tag.has_attr('src') and not any(x in tag['src'] for x in ['static', 'https','http']):
-            src = tag['src']
-            static_src = "{% static " + f"'{src}'" + " %}"
-            tag['src'] = static_src
+    elements_with_if = soup.find_all(attrs={"dj-if": True})
+    elements_with_for = soup.find_all(attrs={"dj-for": True})
 
+    for script in scripts:
+        replace_with_static(script, 'src')
 
     for link in links:
-        if not any(x in link['href'] for x in ['static', 'https','http']):
-            href = link['href']
-            static_href = "{% static " + f"'{href}'" + " %}"
-            link['href'] = static_href
-
-
-
-    for image in img:
-        if not any(x in image['src'] for x in ['static', 'https','http']):
-            img_src = image['src']
+        replace_with_static(link, 'href')
 
-            static_img = "{% static " + f"'{img_src}'" + " %}"
-            image["src"] = static_img
+    if img:
+        for image in img_elements:
+            replace_with_static(image, 'src')
 
     if bool(custom_tag) and bool(custom_attr):
         for element in custom:
-            if element.has_attr(custom_attr) and not any(x in element[custom_attr] for x in ['static', 'https','http']):
-                attr_val = element[custom_attr]
-                jinja_value = "{% static " + f"'{attr_val}'" + " %}"
-                element[custom_attr] = jinja_value
-
+            replace_with_static(element, custom_attr)
 
     # edit the html to add if statements to the elements with if attribute
-    for element in elements_with_if:
-        val = element['if']
-        if val:
-            possible_else = element.find_next_sibling()
-            if_val = "{% if " + f"{val}" + " %}"
-            element.insert_before(if_val)
-
-            while possible_else.has_attr('elif') or possible_else.has_attr('else'):
-                if possible_else.has_attr('elif') and possible_else['elif'] != '':
-                    possible_else.insert_before("{% elif " + f"{possible_else['elif']}" + " %}")
-                    del possible_else['elif']
-                elif possible_else.has_attr('else'):
-                    possible_else.insert_before("{% else %}")
-                    del possible_else['else']
-
-                possible_else = possible_else.find_next_sibling()
-
-            possible_else.insert_before("{% endif %}")
-            del element['if']
-
-
-    for element in elements_with_for:
-        val = element['for']
-        if val != '':
-            for_val = "{% for " + f"{val}" + " %}"
-            element.insert_before(for_val)
-
-            element.insert_after("{% endfor %}")
-            del element['for']
+    edit_if(elements_with_if)
 
+    edit_for(elements_with_for)
 
     if edit_form:
         forms = soup.find_all('form')
@@ -114,30 +157,15 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
                 form.insert(0, "{% csrf_token %}")
 
     if edit_a:
-        aElms = soup.find_all('a')
-        for a in aElms:
-            if a.has_attr('href') and not any(x in a['href'] for x in ['static', 'https','http']) and not a['href'].startswith('#') \
-                    and not a['href'].startswith('mailto') and not a['href'].startswith('{% url'):
-                href = a['href']
-                if href.startswith('/'):
-                    href = href[1:]
-                if href.endswith('/'):
-                    href = href[:-1]
-                if href.endswith('.html'):
-                    href = href[:-5]
-                if href.endswith('.htm'):
-                    href = href[:-4]
-                static_href = "{% url " + f"'{href}'" + " %}"
-                a['href'] = static_href
-
+        a_elms = soup.find_all('a')
+        edit_url(a_elms)
 
     # Prettifying the output with beautiful soup prettify() method
-    modified_html = soup.prettify(formatter='html')
+    modified_html = soup.prettify()
 
     with open(file, 'w') as f:
         f.write(modified_html)
 
 
-
-if __name__ =="__main__":
-    djangoify('index.html')
\ No newline at end of file
+if __name__ == "__main__":
+    djangoify('index.html')

From 489827f61f713d33a6339a96d40c36484ad678e8 Mon Sep 17 00:00:00 2001
From: OSAMATAMER <59906580+OSAMATAMER@users.noreply.github.com>
Date: Mon, 11 Sep 2023 16:43:30 +0300
Subject: [PATCH 5/7] reusable code and more consistent code edit handling urls

---
 html2django/djangohtml.py | 213 ++++++++++++++++++++++----------------
 1 file changed, 122 insertions(+), 91 deletions(-)

diff --git a/html2django/djangohtml.py b/html2django/djangohtml.py
index eafa842..361c65e 100644
--- a/html2django/djangohtml.py
+++ b/html2django/djangohtml.py
@@ -1,28 +1,117 @@
 from bs4 import BeautifulSoup
 
-def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: str = None,custom_value: str = None, edit_form:bool = True, edit_a:bool = True) -> None:
+def edit_if(elements_with_if):
+    for element in elements_with_if:
+        val = element['dj-if']
+        if val != '':
+            possible_else = element.find_next_sibling()
+            if_val = "{% if " + f"{val}" + " %}"
+            element.insert_before(if_val)
+
+            while possible_else.has_attr('elif') or possible_else.has_attr('else'):
+                if possible_else.has_attr('elif') and possible_else['elif'] != '':
+                    possible_else.insert_before("{% elif " + f"{possible_else['elif']}" + " %}")
+                    del possible_else['elif']
+                elif possible_else.has_attr('else'):
+                    possible_else.insert_before("{% else %}")
+                    del possible_else['else']
+
+                possible_else = possible_else.find_next_sibling()
+
+            possible_else.insert_before("{% endif %}")
+            del element['dj-if']
+        else:
+            print("dj-if attribute value cannot be empty.")
+
+
+def edit_for(elements_with_for):
+    for element in elements_with_for:
+        val = element['dj-for']
+        if val != '':
+            for_val = "{% for " + f"{val}" + " %}"
+            element.insert_before(for_val)
 
+            element.insert_after("{% endfor %}")
+            del element['dj-for']
+        else:
+            print("dj-for attribute value cannot be empty.")
+
+
+def edit_url(a_elms):
+    for a in a_elms:
+        if a.has_attr('href') and not any(x in a['href'] for x in ['static', 'https', 'http']) and not a[
+            'href'].startswith('#') \
+                and not a['href'].startswith('mailto') and not a['href'].startswith('{% url') and not a['href'] == ' ':
+            href = a['href']
+            if href.endswith('/'):
+                href = href[:-1]
+            if href.endswith('.html'):
+                href = href[:-5]
+            if href.endswith('.htm'):
+                href = href[:-4]
+
+            last_part = href
+            if '/' in href:
+                url_parts = href.split('/')
+                # Get the last part of the URL
+                last_part = url_parts[-1]
+
+            static_href = "{% url " + f"'{last_part}'" + " %}"
+            a['href'] = static_href
+
+
+def replace_with_static(element, attr):
+    if element.has_attr(attr) and not any(x in element[attr] for x in ['static', 'https', 'http']):
+        attr_val = element[attr]
+        jinja_value = "{% static " + f"'{attr_val}'" + " %}"
+        element[attr] = jinja_value
+
+
+def djangoify(file: str, img: bool = False, custom_tag: str = None, custom_attr: str = None, custom_value: str = None,
+              edit_form: bool = True, edit_a: bool = True) -> None:
     """
-        Modifies the HTML content of a file to replace all <link> tags' href attributes, all <script> tags' src attributes and
-        all <img> src attributes by default,also provides support for  custom  modification of  tags and attribute
-        with Django template tags, and adds "{% load static %}" to the beginning of the file, Add django if statements to each element
-        that have if attribute with the val of the attribute is the condition.
+    This function modifies the HTML content of a file to integrate it with Django template tags.
+    It performs several operations such as replacing local file references with Django static template tags,
+    adding Django CSRF tokens to forms, and modifying URLs in anchor tags to use Django URL template tags.
+    Additionally, it processes custom 'dj-if' and 'dj-for' attributes in the HTML to add Django template if and for tags.
+
+    Parameters:
+    - file (str): The path to the input HTML file that needs to be modified.
+
+    - img (bool): If set to True, the function will modify the 'src' attributes of <img> tags to use Django static template tags.
+                  Defaults to False.
+
+    - custom_tag (str): Specifies a custom HTML tag that you want to modify. This parameter should be used in conjunction
+                        with 'custom_attr' and 'custom_value' parameters to define the modifications. Defaults to None.
+
+    - custom_attr (str): Specifies a custom attribute of the 'custom_tag' that you want to modify. Defaults to None.
+
+    - custom_value (str): Specifies the value that should be set for the 'custom_attr' attribute. Defaults to None.
+
+    - edit_form (bool): If set to True, the function will add Django CSRF tokens to all <form> tags in the HTML file.
+                        This helps in protecting against cross-site request forgery attacks. Defaults to True.
+
+    - edit_a (bool): If set to True, the function will modify the 'href' attributes of <a> tags to use Django URL template tags,
+                     making it easier to manage URLs in Django projects. Defaults to True.
+
+    Additional Information:
+    - The script processes elements with 'dj-if' attributes to add Django template if tags. The value of the 'dj-if' attribute
+      is used as the condition in the if tag. Similarly, elements with 'dj-for' attributes are processed to add Django template
+      for tags, with the value of the 'dj-for' attribute used as the loop variable and iterable.
 
+    Returns:
+    None: The function modifies the input file in place, and does not return any value.
 
-        Parameters:
-            - file (str): The path to the input HTML file.
-            - img (bool): specify True if you want the img tags to be djangofied
-            - custom_tag (str): Custom html tag to modify.To be used with the custom_attr param.
-            - custom_attr (str): custom html attribute of the custom_tag
-            - custom_value (any): the value of the custon attribute
+    Raises:
+    - FileNotFoundError: If the input file cannot be found or accessed, a FileNotFoundError will be raised with a descriptive message.
 
-        Returns:
-            None: The function modifies the input file in place.
+    Example Usage:
+    >>> djangoify('index.html', img=True, custom_tag='custom-tag', custom_attr='custom-attr', custom_value='custom-value')
 
-        Raises:
-            FileNotFoundError: If the input file cannot be found or accessed.
+    This will modify 'index.html' to integrate it with Django, including modifying <img> tags and custom tags as specified.
     """
 
+
     try:
         with open(file, 'r') as f:
             readhtml = f.readlines()
@@ -39,73 +128,30 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
     soup = BeautifulSoup(original_html, 'html.parser')
 
     links = soup.find_all('link', href=True)
-    script = soup.find_all('script')
-    img = soup.find_all('img')
+    scripts = soup.find_all('script')
+    img_elements = soup.find_all('img')
     custom = soup.find_all(custom_tag)
-    elements_with_if = soup.find_all(attrs={"if": True})
-    elements_with_for = soup.find_all(attrs={"for": True})
-
-    for tag in script:
-        if tag.has_attr('src') and not any(x in tag['src'] for x in ['static', 'https','http']):
-            src = tag['src']
-            static_src = "{% static " + f"'{src}'" + " %}"
-            tag['src'] = static_src
+    elements_with_if = soup.find_all(attrs={"dj-if": True})
+    elements_with_for = soup.find_all(attrs={"dj-for": True})
 
+    for script in scripts:
+        replace_with_static(script, 'src')
 
     for link in links:
-        if not any(x in link['href'] for x in ['static', 'https','http']):
-            href = link['href']
-            static_href = "{% static " + f"'{href}'" + " %}"
-            link['href'] = static_href
-
-
+        replace_with_static(link, 'href')
 
-    for image in img:
-        if not any(x in image['src'] for x in ['static', 'https','http']):
-            img_src = image['src']
-
-            static_img = "{% static " + f"'{img_src}'" + " %}"
-            image["src"] = static_img
+    if img:
+        for image in img_elements:
+            replace_with_static(image, 'src')
 
     if bool(custom_tag) and bool(custom_attr):
         for element in custom:
-            if element.has_attr(custom_attr) and not any(x in element[custom_attr] for x in ['static', 'https','http']):
-                attr_val = element[custom_attr]
-                jinja_value = "{% static " + f"'{attr_val}'" + " %}"
-                element[custom_attr] = jinja_value
-
+            replace_with_static(element, custom_attr)
 
     # edit the html to add if statements to the elements with if attribute
-    for element in elements_with_if:
-        val = element['if']
-        if val:
-            possible_else = element.find_next_sibling()
-            if_val = "{% if " + f"{val}" + " %}"
-            element.insert_before(if_val)
-
-            while possible_else.has_attr('elif') or possible_else.has_attr('else'):
-                if possible_else.has_attr('elif') and possible_else['elif'] != '':
-                    possible_else.insert_before("{% elif " + f"{possible_else['elif']}" + " %}")
-                    del possible_else['elif']
-                elif possible_else.has_attr('else'):
-                    possible_else.insert_before("{% else %}")
-                    del possible_else['else']
-
-                possible_else = possible_else.find_next_sibling()
-
-            possible_else.insert_before("{% endif %}")
-            del element['if']
-
-
-    for element in elements_with_for:
-        val = element['for']
-        if val != '':
-            for_val = "{% for " + f"{val}" + " %}"
-            element.insert_before(for_val)
-
-            element.insert_after("{% endfor %}")
-            del element['for']
+    edit_if(elements_with_if)
 
+    edit_for(elements_with_for)
 
     if edit_form:
         forms = soup.find_all('form')
@@ -114,30 +160,15 @@ def djangoify(file: str,img: bool = False ,custom_tag: str = None,custom_attr: s
                 form.insert(0, "{% csrf_token %}")
 
     if edit_a:
-        aElms = soup.find_all('a')
-        for a in aElms:
-            if a.has_attr('href') and not any(x in a['href'] for x in ['static', 'https','http']) and not a['href'].startswith('#') \
-                    and not a['href'].startswith('mailto') and not a['href'].startswith('{% url'):
-                href = a['href']
-                if href.startswith('/'):
-                    href = href[1:]
-                if href.endswith('/'):
-                    href = href[:-1]
-                if href.endswith('.html'):
-                    href = href[:-5]
-                if href.endswith('.htm'):
-                    href = href[:-4]
-                static_href = "{% url " + f"'{href}'" + " %}"
-                a['href'] = static_href
-
+        a_elms = soup.find_all('a')
+        edit_url(a_elms)
 
     # Prettifying the output with beautiful soup prettify() method
-    modified_html = soup.prettify(formatter='html')
+    modified_html = soup.prettify()
 
     with open(file, 'w') as f:
         f.write(modified_html)
 
 
-
-if __name__ =="__main__":
-    djangoify('index.html')
\ No newline at end of file
+if __name__ == "__main__":
+    djangoify('index.html')

From 094544c737fc78c6508d963ea80954e946e5a792 Mon Sep 17 00:00:00 2001
From: OSAMATAMER <59906580+OSAMATAMER@users.noreply.github.com>
Date: Sun, 4 Feb 2024 17:48:41 +0300
Subject: [PATCH 6/7] testing

---
 html2django/process_file.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 html2django/process_file.py

diff --git a/html2django/process_file.py b/html2django/process_file.py
new file mode 100644
index 0000000..e492450
--- /dev/null
+++ b/html2django/process_file.py
@@ -0,0 +1,18 @@
+from djangohtml import djangoify
+import sys
+
+
+def main():
+    # Check if a file path is provided as a command line argument
+    if len(sys.argv) != 2:
+        print("Usage: python script.py <path_to_file>")
+        sys.exit(1)
+
+    file_path = sys.argv[1]
+
+    # Call the djangoify function from the djangohtml module
+    djangoify(file_path)
+
+if __name__ == "__main__":
+    main()
+

From 69340f6935c3a5ab9eece0d15f6b5e9c1349a91b Mon Sep 17 00:00:00 2001
From: Osama <Osama@gmail.com>
Date: Sun, 4 Feb 2024 18:21:29 +0300
Subject: [PATCH 7/7] export_script

---
 html2django/process_file.py | 37 ++++++++++++++++++++++++++++---------
 1 file changed, 28 insertions(+), 9 deletions(-)
 mode change 100644 => 100755 html2django/process_file.py

diff --git a/html2django/process_file.py b/html2django/process_file.py
old mode 100644
new mode 100755
index e492450..e7963ae
--- a/html2django/process_file.py
+++ b/html2django/process_file.py
@@ -1,18 +1,37 @@
-from djangohtml import djangoify
+#!/usr/bin/python3
+
+import os
 import sys
+from djangohtml import djangoify
 
+def process_html_file(file_path):
+    # Perform your operations on the HTML file here
+    # For demonstration, it just prints the file path
+    djangoify(file_path)
+    print("Processing:", file_path)
 
-def main():
-    # Check if a file path is provided as a command line argument
-    if len(sys.argv) != 2:
-        print("Usage: python script.py <path_to_file>")
+def main(directory):
+    # Check if the directory exists
+    if not os.path.exists(directory):
+        print("Directory not found:", directory)
         sys.exit(1)
 
-    file_path = sys.argv[1]
+    # Check if the path is a directory
+    if not os.path.isdir(directory):
+        print("The given path is not a directory:", directory)
+        sys.exit(1)
 
-    # Call the djangoify function from the djangohtml module
-    djangoify(file_path)
+    # Walk through the directory
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            if file.endswith(".html"):
+                file_path = os.path.join(root, file)
+                process_html_file(file_path)
 
 if __name__ == "__main__":
-    main()
+    if len(sys.argv) != 2:
+        print("Usage: python script.py <directory_path>")
+        sys.exit(1)
+
+    main(sys.argv[1])
 
