diff options
Diffstat (limited to 'src/test/resources/http-cache/sdc_proxy.py')
| -rw-r--r-- | src/test/resources/http-cache/sdc_proxy.py | 124 |
1 files changed, 124 insertions, 0 deletions
diff --git a/src/test/resources/http-cache/sdc_proxy.py b/src/test/resources/http-cache/sdc_proxy.py new file mode 100644 index 000000000..301e7af9d --- /dev/null +++ b/src/test/resources/http-cache/sdc_proxy.py @@ -0,0 +1,124 @@ +#!/usr/bin/env python2 +### +# ============LICENSE_START======================================================= +# ONAP CLAMP +# ================================================================================ +# Copyright (C) 2018 AT&T Intellectual Property. All rights +# reserved. +# ================================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============LICENSE_END============================================ +# =================================================================== +# ECOMP is a trademark and service mark of AT&T Intellectual Property. +### + +import json +import requests +import os +import sys +import SimpleHTTPServer +import SocketServer +import argparse +import tempfile +import signal + +parser = argparse.ArgumentParser(description="SDC Cache & Replay") +parser.add_argument("--username", "-u", type=str, help="Set the username for contacting SDC") +parser.add_argument("--password", "-p", type=str, help="Set the password for contacting SDC") +parser.add_argument("--root", "-r", default=tempfile.mkdtemp, type=str, help="Root folder for the proxy cache") +parser.add_argument("--proxy" , type=str, help="Url of the Act as a proxy. If not set, this script only uses the cache and will return a 404 if files aren't found") +parser.add_argument("--port", "-P", type=int, default="8081", help="Port on which the proxy should listen to") +parser.add_argument("--verbose", "-v", type=bool, help="Print more information in case of error") +options = parser.parse_args() + + +PORT = options.port +SDC_HOST = options.proxy +SDC_AUTH = (options.username, options.password) +SDC_HEADERS = {'X-ECOMP-InstanceID':'CLAMP'} +CACHE_ROOT = options.root + +def signal_handler(signal_sent, frame): + global httpd + if signal_sent == signal.SIGINT: + print('Got Ctrl-C (SIGINT)') + httpd.socket.close() + httpd.shutdown() + httpd.server_close() + +class Proxy(SimpleHTTPServer.SimpleHTTPRequestHandler): + + def print_headers(self): + for header,value in self.headers.items(): + print("%s : %s" % (header, value)) + + def check_credentials(self): + pass + + def _send_content(self, header_file, content_file): + self.send_response(200) + with open(header_file, 'rb') as f: + headers = json.load(f) + for key,value in headers.items(): + if key in ('Transfer-Encoding',): + continue + self.send_header(key, value) + self.end_headers() + with open(content_file,'rb') as f: + fc = f.read() + self.wfile.write(fc) + + def _write_cache(self,cached_file, header_file, content_file, response): + os.makedirs(cached_file, True) + with open(content_file, 'w') as f: + f.write(response.raw.read()) + with open(header_file, 'w') as f: + json.dump(dict(response.raw.headers), f) + # Entry point of the code + def do_GET(self): + + self.print_headers() + self.check_credentials() + + cached_file = '%s/%s' % (CACHE_ROOT, self.path,) + cached_file_content = "%s/.file" % (cached_file,) + cached_file_header = "%s/.header" % (cached_file,) + + _file_available = os.path.exists(cached_file_content) + if not _file_available and not SDC_HOST: + self.send_response(404) + return "404 Not found" + + if not _file_available: + print("Request for not cached file : %s" % (cached_file,)) + url = '%s%s' % (SDC_HOST, self.path) + response = requests.get(url, auth=SDC_AUTH, headers=SDC_HEADERS, stream=True) + + if response.status_code == 200: + self._write_cache(cached_file, cached_file_header, cached_file_content, response) + else: + print('Error when requesting file :') + print('Requested url : %s' % (url,)) + print('Status code : %s' % (response.status_code,)) + print('Content : %s' % (response.content,)) + self.send_response(response.status_code) + return response.content + + self._send_content(cached_file_header, cached_file_content) + +# Main code that start the HTTP server +httpd = SocketServer.ForkingTCPServer(('', PORT), Proxy) +httpd.allow_reuse_address = True +print "Listening on port "+ str(PORT) + " and caching in " + CACHE_ROOT + "(Press Ctrl+C to stop HTTPD Caching script)" +signal.signal(signal.SIGINT, signal_handler) +httpd.serve_forever()
\ No newline at end of file |