ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-18 04:38:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding cache for sas return (#224)

Browse Source
This commit is contained in:
Anshuman Goel
2020-10-29 07:44:25 -07:00
committed by GitHub
parent f4b874e19e
commit 99b69d3e56
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/cli/onefuzz/api.py
View File

@ -10,11 +10,12 @@ import re
import subprocess # nosec import subprocess # nosec
import uuid import uuid
from shutil import which from shutil import which
from typing import Callable, Dict, List, Optional, Tuple, Type, TypeVar from typing import Callable, Dict, List, Optional, Tuple, Type, TypeVar, cast
from uuid import UUID from uuid import UUID
import pkg_resources import pkg_resources
import semver import semver
from memoization import cached
from onefuzztypes import enums, models, primitives, requests, responses from onefuzztypes import enums, models, primitives, requests, responses
from pydantic import BaseModel from pydantic import BaseModel
from six.moves import input # workaround for static analysis from six.moves import input # workaround for static analysis
@ -33,6 +34,8 @@ DEFAULT = {
# This was generated randomly and should be preserved moving forwards # This was generated randomly and should be preserved moving forwards
ONEFUZZ_GUID_NAMESPACE = uuid.UUID("27f25e3f-6544-4b69-b309-9b096c5a9cbc") ONEFUZZ_GUID_NAMESPACE = uuid.UUID("27f25e3f-6544-4b69-b309-9b096c5a9cbc")
ONE_HOUR_IN_SECONDS = 3600
DEFAULT_LINUX_IMAGE = "Canonical:UbuntuServer:18.04-LTS:latest" DEFAULT_LINUX_IMAGE = "Canonical:UbuntuServer:18.04-LTS:latest"
DEFAULT_WINDOWS_IMAGE = "MicrosoftWindowsDesktop:Windows-10:rs5-pro:latest" DEFAULT_WINDOWS_IMAGE = "MicrosoftWindowsDesktop:Windows-10:rs5-pro:latest"
@ -146,6 +149,7 @@ class Files(Endpoint):
endpoint = "files" endpoint = "files"
@cached(ttl=ONE_HOUR_IN_SECONDS)
def _get_client(self, container: str) -> ContainerWrapper: def _get_client(self, container: str) -> ContainerWrapper:
sas = self.onefuzz.containers.get(container).sas_url sas = self.onefuzz.containers.get(container).sas_url
return ContainerWrapper(sas) return ContainerWrapper(sas)
@ -166,7 +170,8 @@ class Files(Endpoint):
""" get a file from a container """ """ get a file from a container """
self.logger.debug("getting file from container: %s:%s", container, filename) self.logger.debug("getting file from container: %s:%s", container, filename)
client = self._get_client(container) client = self._get_client(container)
return client.download_blob(filename) downloaded = cast(bytes, client.download_blob(filename))
return downloaded
def upload_file( def upload_file(
self, container: str, file_path: str, blob_name: Optional[str] = None self, container: str, file_path: str, blob_name: Optional[str] = None