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

Add user_info to Jobs & Repro (#327)

Browse Source 
This adds information about the user that created a job or repro VM to the respective resources.

This expands on the addition made to tasks in #303.
This commit is contained in:
bmc-msft
2020-11-20 10:46:52 -05:00
committed by GitHub
parent d96209c659
commit 9e2a61fe66
4 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/api-service/__app__/onefuzzlib/repro.py
View File

@ -11,7 +11,7 @@ from azure.mgmt.compute.models import VirtualMachine
from onefuzztypes.enums import OS, ContainerType, ErrorCode, VmState
from onefuzztypes.models import Error
from onefuzztypes.models import Repro as BASE_REPRO
from onefuzztypes.models import ReproConfig, TaskVm
from onefuzztypes.models import ReproConfig, TaskVm, UserInfo
from .azure.auth import build_auth
from .azure.containers import StorageType, save_blob
@ -219,7 +219,9 @@ class Repro(BASE_REPRO, ORMMixin):
return cls.search(query=query)
@classmethod
def create(cls, config: ReproConfig) -> Union[Error, "Repro"]:
def create(
cls, config: ReproConfig, user_info: Optional[UserInfo]
) -> Union[Error, "Repro"]:
report = get_report(config.container, config.path)
if not report:
return Error(
@ -233,6 +235,8 @@ class Repro(BASE_REPRO, ORMMixin):
vm = cls(config=config, task_id=task.task_id, os=task.os, auth=build_auth())
if vm.end_time is None:
vm.end_time = datetime.utcnow() + timedelta(hours=config.duration)
vm.user_info = user_info
vm.save()
return vm