More code re-organizing

src/engines/blender/blender_downloader.py

@@ -0,0 +1,101 @@
+import logging
+import platform
+import re
+
+import requests
+
+from ..core.downloader_core import download_and_extract_app
+
+# url = "https://download.blender.org/release/"
+url = "https://ftp.nluug.nl/pub/graphics/blender/release/"  # much faster mirror for testing
+
+logger = logging.getLogger()
+supported_formats = ['.zip', '.tar.xz', '.dmg']
+
+
+class BlenderDownloader:
+
+    @staticmethod
+    def get_major_versions():
+        try:
+            response = requests.get(url)
+            response.raise_for_status()
+
+            # Use regex to find all the <a> tags and extract the href attribute
+            link_pattern = r'<a href="([^"]+)">Blender(\d+[^<]+)</a>'
+            link_matches = re.findall(link_pattern, response.text)
+
+            major_versions = [link[-1].strip('/') for link in link_matches]
+            major_versions.sort(reverse=True)
+            return major_versions
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Error: {e}")
+        return None
+
+    @staticmethod
+    def get_minor_versions(major_version, system_os=None, cpu=None):
+
+        base_url = url + 'Blender' + major_version
+
+        response = requests.get(base_url)
+        response.raise_for_status()
+
+        versions_pattern = r'<a href="(?P<file>[^"]+)">blender-(?P<version>[\d\.]+)-(?P<system_os>\w+)-(?P<cpu>\w+).*</a>'
+        versions_data = [match.groupdict() for match in re.finditer(versions_pattern, response.text)]
+
+        # Filter to just the supported formats
+        versions_data = [item for item in versions_data if any(item["file"].endswith(ext) for ext in supported_formats)]
+
+        if system_os:
+            versions_data = [x for x in versions_data if x['system_os'] == system_os]
+        if cpu:
+            versions_data = [x for x in versions_data if x['cpu'] == cpu]
+
+        for v in versions_data:
+            v['url'] = base_url + '/' + v['file']
+
+        versions_data = sorted(versions_data, key=lambda x: x['version'], reverse=True)
+        return versions_data
+
+    @staticmethod
+    def find_LTS_versions():
+        response = requests.get('https://www.blender.org/download/lts/')
+        response.raise_for_status()
+
+        lts_pattern = r'https://www.blender.org/download/lts/(\d+-\d+)/'
+        lts_matches = re.findall(lts_pattern, response.text)
+        lts_versions = [ver.replace('-', '.') for ver in list(set(lts_matches))]
+        lts_versions.sort(reverse=True)
+
+        return lts_versions
+
+    @classmethod
+    def find_most_recent_version(cls, system_os, cpu, lts_only=False):
+        try:
+            major_version = cls.find_LTS_versions()[0] if lts_only else cls.get_major_versions()[0]
+            most_recent = cls.get_minor_versions(major_version, system_os, cpu)[0]
+            return most_recent
+        except IndexError:
+            logger.error("Cannot find a most recent version")
+
+    @classmethod
+    def download_engine(cls, version, download_location, system_os=None, cpu=None):
+        system_os = system_os or platform.system().lower().replace('darwin', 'macos')
+        cpu = cpu or platform.machine().lower().replace('amd64', 'x64')
+
+        try:
+            logger.info(f"Requesting download of blender-{version}-{system_os}-{cpu}")
+            major_version = '.'.join(version.split('.')[:2])
+            minor_versions = [x for x in cls.get_minor_versions(major_version, system_os, cpu) if x['version'] == version]
+            # we get the URL instead of calculating it ourselves. May change this
+
+            download_and_extract_app(remote_url=minor_versions[0]['url'], download_location=download_location)
+        except IndexError:
+            logger.error("Cannot find requested engine")
+
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
+    print(BlenderDownloader.get_major_versions())
+