diff --git a/src/main.py b/src/main.py index f5b8e2c..7d8bbca 100644 --- a/src/main.py +++ b/src/main.py @@ -42,6 +42,7 @@ import base64 import json import re +from urllib import parse # +++++++++++++++++++++++++++++++++++++++++++++++++++++ # IMPORTS (3rd Party) @@ -52,7 +53,6 @@ import requests from requests.structures import CaseInsensitiveDict - # +++++++++++++++++++++++++++++++++++++++++++++++++++++ # Local Imports # +++++++++++++++++++++++++++++++++++++++++++++++++++++ @@ -104,6 +104,7 @@ def __init__(self, config_=None): self.redirect_page = Path( self.output_folder, self.config["pages"]["redirect"] ) + self.all_redirects = self.config["pages"]["all-redirects"] self.robots_txt_page = Path( self.output_folder, self.config["pages"]["robots"] ) @@ -412,6 +413,18 @@ def create_redirect_toml_file(self): else: helpers.log_to_console("WARNING", "No Redirect File found") + for redirect in self.all_redirects: + rules.append( + [ + f"[[redirects]]\n", + f'from = "{redirect["from"]}"\n', + f'to = "{redirect["to"]}"\n', + f'status = {redirect["status"]}\n', + "force = true\n", + "\n", + ] + ) + netlify_toml_file = Path(self.output_folder, "netlify.toml") with open(netlify_toml_file, "w", encoding="utf-8") as f: @@ -460,6 +473,28 @@ def create_redirect_toml_file(self): except: pass + # redirects + all_redirects = [] + try: + redirects_source = src_url + "/wp-json/redirection/v1/redirect" + red_response = requests.get( + redirects_source, headers={"Authorization": "Basic " + wp_token} + ) + + redirects_dict = json.loads(red_response.content) + + all_redirects = [ + { + "from": red["url"], + "to": red["action_data"]["url"], + "status": red["action_code"], + "force": True, + } + for red in redirects_dict["items"] + ] + except: + pass + if wordpress_simply_static_zip_url: configurations = { "root": "", @@ -473,6 +508,7 @@ def create_redirect_toml_file(self): "redirect": page_redirects, "robots": page_robots, "search": page_search, + "all-redirects": all_redirects, }, "search": { "title": "true",