This repository has been archived on 2020-03-24. You can view files and clone it, but cannot push or open issues or pull requests.
quay/storage/fakestorage.py

101 lines
2.9 KiB
Python
Raw Normal View History

import cStringIO as StringIO
import hashlib
from collections import defaultdict
from uuid import uuid4
from storage.basestorage import BaseStorageV2
2018-10-31 18:13:25 +00:00
_GLOBAL_FAKE_STORAGE_MAP = defaultdict(StringIO.StringIO)
class FakeStorage(BaseStorageV2):
def __init__(self, context):
2016-01-15 16:15:40 +00:00
super(FakeStorage, self).__init__()
2018-10-31 18:13:25 +00:00
self._fake_storage_map = (defaultdict(StringIO.StringIO)
if context == 'local' else _GLOBAL_FAKE_STORAGE_MAP)
2016-01-15 16:15:40 +00:00
def _init_path(self, path=None, create=False):
return path
def get_direct_download_url(self, path, request_ip=None, expires_in=60, requires_cors=False, head=False):
2016-01-19 20:52:34 +00:00
try:
if self.get_content('supports_direct_download') == 'true':
return 'http://somefakeurl?goes=here'
2016-01-19 20:52:34 +00:00
except:
pass
return None
def get_content(self, path):
2018-10-31 18:13:25 +00:00
if not path in self._fake_storage_map:
raise IOError('Fake file %s not found. Exist: %s' % (path, self._fake_storage_map.keys()))
2015-05-29 22:08:17 +00:00
2018-10-31 18:13:25 +00:00
self._fake_storage_map.get(path).seek(0)
return self._fake_storage_map.get(path).read()
def put_content(self, path, content):
2018-10-31 18:13:25 +00:00
self._fake_storage_map.pop(path, None)
self._fake_storage_map[path].write(content)
def stream_read(self, path):
2018-10-31 18:13:25 +00:00
io_obj = self._fake_storage_map[path]
io_obj.seek(0)
while True:
buf = io_obj.read(self.buffer_size)
if not buf:
break
yield buf
def stream_read_file(self, path):
return StringIO.StringIO(self.get_content(path))
def stream_write(self, path, fp, content_type=None, content_encoding=None):
2018-10-31 18:13:25 +00:00
out_fp = self._fake_storage_map[path]
out_fp.seek(0)
self.stream_write_to_fp(fp, out_fp)
def remove(self, path):
2018-10-31 18:13:25 +00:00
self._fake_storage_map.pop(path, None)
def exists(self, path):
2018-10-31 18:13:25 +00:00
if self._fake_storage_map.get('all_files_exist', None):
2016-05-05 17:55:24 +00:00
return True
2018-10-31 18:13:25 +00:00
return path in self._fake_storage_map
def get_checksum(self, path):
2018-10-31 18:13:25 +00:00
return hashlib.sha256(self._fake_storage_map[path].read()).hexdigest()[:7]
def initiate_chunked_upload(self):
new_uuid = str(uuid4())
2018-10-31 18:13:25 +00:00
self._fake_storage_map[new_uuid].seek(0)
return new_uuid, {}
2015-11-30 20:45:45 +00:00
def stream_upload_chunk(self, uuid, offset, length, in_fp, _, content_type=None):
2018-10-31 18:13:25 +00:00
upload_storage = self._fake_storage_map[uuid]
upload_storage.seek(offset)
2018-10-31 18:13:25 +00:00
2015-12-10 04:16:33 +00:00
try:
return self.stream_write_to_fp(in_fp, upload_storage, length), {}, None
except IOError as ex:
return 0, {}, ex
def complete_chunked_upload(self, uuid, final_path, _):
2018-10-31 18:13:25 +00:00
self._fake_storage_map[final_path] = self._fake_storage_map[uuid]
self._fake_storage_map.pop(uuid, None)
def cancel_chunked_upload(self, uuid, _):
2018-10-31 18:13:25 +00:00
self._fake_storage_map.pop(uuid, None)
def copy_to(self, destination, path):
2018-10-31 18:13:25 +00:00
if self.exists('break_copying'):
raise IOError('Broken!')
if self.exists('fake_copying'):
return
if self.exists('except_copying'):
raise Exception("I'm an exception!")
content = self.get_content(path)
destination.put_content(path, content)