@@ -319,6 +319,12 @@ <h1>Source code for cloudpub.ms_azure.utils</h1><div class="highlight"><pre>
319319 < span class ="k "> return</ span > < span class ="n "> plan_name</ span >
320320 < span class ="k "> return</ span > < span class ="sa "> f</ span > < span class ="s2 "> "</ span > < span class ="si "> {</ span > < span class ="n "> plan_name</ span > < span class ="si "> }</ span > < span class ="s2 "> -</ span > < span class ="si "> {</ span > < span class ="n "> arch</ span > < span class ="o "> .</ span > < span class ="n "> lower</ span > < span class ="p "> ()</ span > < span class ="si "> }</ span > < span class ="s2 "> "</ span >
321321
322+ < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> get_safe_security_type</ span > < span class ="p "> (</ span > < span class ="n "> image_type</ span > < span class ="p "> ):</ span >
323+ < span class ="c1 "> # Arches which aren't x86Gen2 (like ARM64) doesn't work well with security type</ span >
324+ < span class ="k "> if</ span > < span class ="n "> image_type</ span > < span class ="o "> !=</ span > < span class ="s2 "> "x64Gen2"</ span > < span class ="p "> :</ span >
325+ < span class ="k "> return</ span > < span class ="kc "> None</ span >
326+ < span class ="k "> return</ span > < span class ="n "> security_type</ span >
327+
322328 < span class ="n "> sku_mapping</ span > < span class ="p "> :</ span > < span class ="n "> Dict</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="nb "> str</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="p "> {}</ span >
323329 < span class ="c1 "> # Update the SKUs for each image in DiskVersions if needed</ span >
324330 < span class ="k "> for</ span > < span class ="n "> disk_version</ span > < span class ="ow "> in</ span > < span class ="n "> disk_versions</ span > < span class ="p "> :</ span >
@@ -339,7 +345,7 @@ <h1>Source code for cloudpub.ms_azure.utils</h1><div class="highlight"><pre>
339345
340346 < span class ="c1 "> # Return the expected SKUs list</ span >
341347 < span class ="n "> res</ span > < span class ="o "> =</ span > < span class ="p "> [</ span >
342- < span class ="n "> VMISku</ span > < span class ="o "> .</ span > < span class ="n "> from_json</ span > < span class ="p "> ({</ span > < span class ="s2 "> "image_type"</ span > < span class ="p "> :</ span > < span class ="n "> k</ span > < span class ="p "> ,</ span > < span class ="s2 "> "id"</ span > < span class ="p "> :</ span > < span class ="n "> v</ span > < span class ="p "> ,</ span > < span class ="s2 "> "security_type"</ span > < span class ="p "> :</ span > < span class ="n "> security_type </ span > < span class ="p "> })</ span >
348+ < span class ="n "> VMISku</ span > < span class ="o "> .</ span > < span class ="n "> from_json</ span > < span class ="p "> ({</ span > < span class ="s2 "> "image_type"</ span > < span class ="p "> :</ span > < span class ="n "> k</ span > < span class ="p "> ,</ span > < span class ="s2 "> "id"</ span > < span class ="p "> :</ span > < span class ="n "> v</ span > < span class ="p "> ,</ span > < span class ="s2 "> "security_type"</ span > < span class ="p "> :</ span > < span class ="n "> get_safe_security_type </ span > < span class ="p "> ( </ span > < span class =" n " > k </ span > < span class =" p " > ) })</ span >
343349 < span class ="k "> for</ span > < span class ="n "> k</ span > < span class ="p "> ,</ span > < span class ="n "> v</ span > < span class ="ow "> in</ span > < span class ="n "> sku_mapping</ span > < span class ="o "> .</ span > < span class ="n "> items</ span > < span class ="p "> ()</ span >
344350 < span class ="p "> ]</ span >
345351 < span class ="k "> return</ span > < span class ="nb "> sorted</ span > < span class ="p "> (</ span > < span class ="n "> res</ span > < span class ="p "> ,</ span > < span class ="n "> key</ span > < span class ="o "> =</ span > < span class ="n "> attrgetter</ span > < span class ="p "> (</ span > < span class ="s2 "> "id"</ span > < span class ="p "> ))</ span >
0 commit comments