From 6396ffae556fb89186846e940fdda40bec4057c6 Mon Sep 17 00:00:00 2001 From: AutomatedTester Date: Mon, 25 Jan 2016 12:19:09 +0000 Subject: [PATCH] Update Python webserver to handle '/page/*' urls --- py/test/selenium/webdriver/common/webserver.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/py/test/selenium/webdriver/common/webserver.py b/py/test/selenium/webdriver/common/webserver.py index f3db74ad8ee38..94eef6e2376e6 100644 --- a/py/test/selenium/webdriver/common/webserver.py +++ b/py/test/selenium/webdriver/common/webserver.py @@ -55,12 +55,18 @@ def do_GET(self): """GET method handler.""" try: path = self.path[1:].split('?')[0] - html = open(os.path.join(HTML_ROOT, path), 'r', encoding='latin-1') + if path[:4] == "page": + html = """Page{page_number} + Page number {page_number} +

top + """.format(page_number=path[5:]) + else: + with open(os.path.join(HTML_ROOT, path), 'r', encoding='latin-1') as f: + html = f.read().encode('utf-8') self.send_response(200) self.send_header('Content-type', 'text/html') self.end_headers() - self.wfile.write(html.read().encode('utf-8')) - html.close() + self.wfile.write(html) except IOError: self.send_error(404, 'File Not Found: %s' % path) @@ -142,5 +148,3 @@ def main(argv=None): if __name__ == "__main__": main() - -