mirror of
https://github.com/microsoft/onefuzz.git
synced 2025-06-20 13:33:46 +00:00
address initial deploy errors (#167)
This commit is contained in:
@ -16,6 +16,7 @@ import uuid
|
|||||||
import zipfile
|
import zipfile
|
||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
from azure.cli.core import CLIError
|
||||||
from azure.common.client_factory import get_client_from_cli_profile
|
from azure.common.client_factory import get_client_from_cli_profile
|
||||||
from azure.common.credentials import get_cli_profile
|
from azure.common.credentials import get_cli_profile
|
||||||
from azure.core.exceptions import ResourceExistsError
|
from azure.core.exceptions import ResourceExistsError
|
||||||
@ -47,6 +48,7 @@ from azure.mgmt.resource.resources.models import (
|
|||||||
DeploymentMode,
|
DeploymentMode,
|
||||||
DeploymentProperties,
|
DeploymentProperties,
|
||||||
)
|
)
|
||||||
|
import time
|
||||||
from azure.mgmt.storage import StorageManagementClient
|
from azure.mgmt.storage import StorageManagementClient
|
||||||
from azure.storage.blob import (
|
from azure.storage.blob import (
|
||||||
BlobServiceClient,
|
BlobServiceClient,
|
||||||
@ -213,6 +215,20 @@ class Client:
|
|||||||
print("\n".join(["* " + x for x in unsupported]))
|
print("\n".join(["* " + x for x in unsupported]))
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
|
def create_password(self, object_id):
|
||||||
|
# Work-around the race condition where the app is created but passwords cannot
|
||||||
|
# be created yet.
|
||||||
|
count = 0
|
||||||
|
while True:
|
||||||
|
time.sleep(5)
|
||||||
|
count += 1
|
||||||
|
try:
|
||||||
|
return add_application_password(object_id)
|
||||||
|
except CLIError as err:
|
||||||
|
if count > 5:
|
||||||
|
raise err
|
||||||
|
logger.info("creating password failed, trying again")
|
||||||
|
|
||||||
def setup_rbac(self):
|
def setup_rbac(self):
|
||||||
"""
|
"""
|
||||||
Setup the client application for the OneFuzz instance.
|
Setup the client application for the OneFuzz instance.
|
||||||
@ -288,7 +304,8 @@ class Client:
|
|||||||
creds = list(client.applications.list_password_credentials(app.object_id))
|
creds = list(client.applications.list_password_credentials(app.object_id))
|
||||||
client.applications.update_password_credentials(app.object_id, creds)
|
client.applications.update_password_credentials(app.object_id, creds)
|
||||||
|
|
||||||
(password_id, password) = add_application_password(app.object_id)
|
(password_id, password) = self.create_password(app.object_id)
|
||||||
|
|
||||||
onefuzz_cli_app_uuid = uuid.UUID(ONEFUZZ_CLI_APP)
|
onefuzz_cli_app_uuid = uuid.UUID(ONEFUZZ_CLI_APP)
|
||||||
cli_app = get_application(onefuzz_cli_app_uuid)
|
cli_app = get_application(onefuzz_cli_app_uuid)
|
||||||
|
|
||||||
@ -367,7 +384,13 @@ class Client:
|
|||||||
account_url=account_url,
|
account_url=account_url,
|
||||||
credential={"account_name": name, "account_key": key},
|
credential={"account_name": name, "account_key": key},
|
||||||
)
|
)
|
||||||
for queue in ["file-changes", "task-heartbeat", "node-heartbeat", "proxy", "update-queue"]:
|
for queue in [
|
||||||
|
"file-changes",
|
||||||
|
"task-heartbeat",
|
||||||
|
"node-heartbeat",
|
||||||
|
"proxy",
|
||||||
|
"update-queue",
|
||||||
|
]:
|
||||||
try:
|
try:
|
||||||
client.create_queue(queue)
|
client.create_queue(queue)
|
||||||
except ResourceExistsError:
|
except ResourceExistsError:
|
||||||
|
@ -1,13 +1,15 @@
|
|||||||
azure-cosmosdb-table~=1.0.6
|
azure-cosmosdb-table==1.0.6
|
||||||
azure-mgmt-storage~=11.0
|
azure-mgmt-storage~=11.2.0
|
||||||
azure-storage-queue~=12.1
|
azure-storage-queue==12.1.3
|
||||||
azure-storage-blob~=12.3
|
azure-storage-blob==12.5.0
|
||||||
azure-core~=1.6
|
|
||||||
azure-graphrbac~=0.61.1
|
azure-graphrbac~=0.60.0
|
||||||
azure-mgmt-eventgrid~=2.2
|
azure-mgmt-eventgrid==3.0.0rc7
|
||||||
azure-mgmt-web~=0.45
|
azure-mgmt-web~=0.47.0
|
||||||
azure-mgmt-resource~=9.0
|
azure-mgmt-resource==10.2.0
|
||||||
azure-mgmt-servicebus~=0.6.0
|
azure-mgmt-servicebus~=0.6.0
|
||||||
azure-cli-core~=2.7
|
azure-cli-core==2.13.0
|
||||||
azure-cli~=2.10.1
|
azure-cli==2.13.0
|
||||||
pyfunctional~=1.4.1
|
pyfunctional==1.4.2
|
||||||
|
cryptography<3.0.0,>=2.3.1
|
||||||
|
# azure-mgmt-core==1.2.0
|
Reference in New Issue
Block a user