Implement pep 503 data-requires for simple repository.

Carreau · mpacer · commit 3e1ce8fb2acd · 2016-10-05T16:03:41.000-07:00
end
This exposes the data-requires-python field when the packages registers it.
diff --git a/store.py b/store.py
@@ -718,15 +718,18 @@ def get_package_urls(self, name, relative=None):
         file_urls = []
 
         # uploaded files
-        safe_execute(cursor, '''select filename, python_version, md5_digest
-            from release_files where name=%s''', (name,))
-        for fname, pyversion, md5 in cursor.fetchall():
+        safe_execute(cursor,
+        '''
+        SELECT filename, requires_python, md5_digest
+        FROM release_files
+        WHERE release_files.name=%s
+        ''', (name,))
+        for fname, requires_python, md5 in cursor.fetchall():
             # Put files first, to have setuptools consider
             # them before going to other sites
-            url = self.gen_file_url(pyversion, name, fname, relative) + \
+            url = self.gen_file_url('<not used arg>', name, fname, relative) + \
                 "#md5=" + md5
-            file_urls.append((url, "internal", fname))
-
+            file_urls.append((url, "internal", fname, requires_python))
         return sorted(file_urls)
 
     def get_uploaded_file_urls(self, name):
diff --git a/webui.py b/webui.py
@@ -1042,7 +1042,7 @@ def simple_body(self, path):
         html.append("""<html><head><title>Links for %s</title><meta name="api-version" value="2" /></head>"""
                     % cgi.escape(path))
         html.append("<body><h1>Links for %s</h1>" % cgi.escape(path))
-        for href, rel, text in urls:
+        for href, rel, text, requires_python in urls:
             if href.startswith('http://cheeseshop.python.org/pypi') or \
                     href.startswith('http://pypi.python.org/pypi') or \
                     href.startswith('http://www.python.org/pypi'):
@@ -1054,7 +1054,10 @@ def simple_body(self, path):
                 rel = ''
             href = cgi.escape(href, quote=True)
             text = cgi.escape(text)
-            html.append("""<a href="%s"%s>%s</a><br/>\n""" % (href, rel, text))
+            data_attr = ''
+            if requires_python:
+                data_attr = ' data-requires-python="{}"'.format(cgi.escape(requires_python))
+            html.append('<a%s href="%s"%s>%s</a><br/>\n' % (data_attr, href, rel, text))
         html.append("</body></html>")
         html = ''.join(html)
         return html
