diff options
Diffstat (limited to 'src/test/resources/http-cache/sdc_proxy.py')
| -rwxr-xr-x | src/test/resources/http-cache/sdc_proxy.py | 127 |
1 files changed, 0 insertions, 127 deletions
diff --git a/src/test/resources/http-cache/sdc_proxy.py b/src/test/resources/http-cache/sdc_proxy.py deleted file mode 100755 index 45dac413..00000000 --- a/src/test/resources/http-cache/sdc_proxy.py +++ /dev/null @@ -1,127 +0,0 @@ -#!/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,) - print("Cached file name before escaping : %s" % cached_file) - cached_file = cached_file.replace('<','<').replace('>','>').replace('?','?').replace('*','*').replace('\\','*').replace(':',':').replace('|','|') - print("Cached file name after escaping (used for cache storage) : %s" % cached_file) - 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("SDC Request for data currently not present in cache: %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 |