Initial commit

benchmark/foundation/buildbot.py

@@ -0,0 +1,79 @@
+import os
+import re
+import requests
+from html.parser import HTMLParser
+
+BUILDBOT_URL = "https://builder.blender.org/"
+BUILDBOT_DOWNLOAD_URL = BUILDBOT_URL + "download/"
+
+
+class BuildbotHTMLParser(HTMLParser):
+    def __init__(self):
+        HTMLParser.__init__(self)
+        self._is_inside_official_table = False
+        self._is_official_table_finished = False
+        self.official_builds = []
+
+    def handle_starttag(self, tag, attrs):
+        tag_lower = tag.lower()
+        if tag_lower == 'a':
+            if self._is_inside_official_table:
+                for attr in attrs:
+                    if attr[0].lower() == 'href':
+                        href = attr[1]
+                        self.official_builds.append(href)
+        elif tag_lower == 'table':
+            classes = ()
+            for attr in attrs:
+                if attr[0].lower() == 'class':
+                    classes = attr[1].lower().split()
+            if 'table-striped' in classes:
+                if self._is_inside_official_table:
+                    self._is_inside_official_table = False
+                    self._is_official_table_finished = True
+                else:
+                    if not self._is_official_table_finished:
+                        self._is_inside_official_table = True
+
+    def handle_endtag(self, tag):
+        pass
+
+    def handle_data(self, data):
+        pass
+
+
+def _getBuildbotPlatformRegex(platform, bitness):
+    platform_lower = platform.lower()
+    if platform_lower in ("linux", "lin"):
+        if bitness.startswith("64"):
+            return re.compile(".*linux-glibc[0-9]+-x86_64.*")
+        elif bitness.startswith("32"):
+            return re.compile(".*linux-glibc[0-9]+-i686.*")
+    else:
+        # TOGO(sergey): Needs implementation
+        pass
+    return None
+
+
+def buildbotGetLatetsVersion(platform, bitness):
+    """
+    Get latest Blender version URL from buildbot website.
+
+    Returns None if something is wrong.
+    """
+    # Get content of the page.
+    r = requests.get(BUILDBOT_DOWNLOAD_URL)
+    if r.status_code != requests.codes.ok:
+        return None
+    # Parse the page.
+    parser = BuildbotHTMLParser()
+    parser.feed(r.text)
+    official_builds = parser.official_builds
+    # Get build which corresponds to requested platform.
+    regex = _getBuildbotPlatformRegex(platform, bitness)
+    if not regex:
+        return None
+    for build in official_builds:
+        if regex.match(build):
+            return BUILDBOT_DOWNLOAD_URL + build
+    return None