o V h @sjddlZddlZddlZddlmZmZmZddlZGdddZ e dkr3e Z e j ddddd Z dSdS) N)ListOptionalTuplec@seZdZdZddZdeefddZdeefddZe dd ed e de ee e e e ffd d Z   dde dedee d e dee ee e e e ff ddZdS) GCPRegionsa] A class for managing and analyzing Google Cloud Platform (GCP) regions. This class provides functionality to initialize, categorize, and analyze GCP regions based on their geographical location, tier classification, and network latency. Attributes: regions (Dict[str, Tuple[int, str, str]]): A dictionary of GCP regions with their tier, city, and country. Methods: tier1: Returns a list of tier 1 GCP regions. tier2: Returns a list of tier 2 GCP regions. lowest_latency: Determines the GCP region(s) with the lowest network latency. Examples: >>> from ultralytics.hub.google import GCPRegions >>> regions = GCPRegions() >>> lowest_latency_region = regions.lowest_latency(verbose=True, attempts=3) >>> print(f"Lowest latency region: {lowest_latency_region[0][0]}") cCsiddddddddd d d d d ddddddddddddddddddd d!d"id#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdI|_dJS)KzaInitializes the GCPRegions class with predefined Google Cloud Platform regions and their details.z asia-east1)TaiwanChinaz asia-east2)z Hong Kongrzasia-northeast1)rTokyoJapanzasia-northeast2)rOsakar zasia-northeast3)r Seoulz South Koreaz asia-south1)r MumbaiIndiaz asia-south2)r Delhirzasia-southeast1)r z Jurong West Singaporezasia-southeast2)r Jakarta Indonesiazaustralia-southeast1)r Sydney Australiazaustralia-southeast2)r Melbournerzeurope-central2)r WarsawPolandz europe-north1)rHaminaFinlandzeurope-southwest1)rMadridSpainz europe-west1)rz St. GhislainBelgiumz europe-west10)r BerlinGermanyz europe-west12)r TurinItalyz europe-west2)r LondonzUnited Kingdomz europe-west3)r Frankfurtrz europe-west4)r Eemshaven Netherlandsz europe-west6)r Zurich Switzerlandz europe-west8)rMilanr!z europe-west9)rParisFrancez me-central1)r DohaQatarzme-west1)rzTel AvivIsraelznorthamerica-northeast1)r MontrealCanadaznorthamerica-northeast2)r Torontor/zsouthamerica-east1)r u São PauloBrazilzsouthamerica-west1)r SantiagoChilez us-central1)rIowa United Stateszus-east1)rzSouth Carolinar5zus-east4)rzNorthern Virginiar5zus-east5)rColumbusr5z us-south1)rDallasr5)rOregonr5)r z Los Angelesr5)r zSalt Lake Cityr5)r z Las Vegasr5)zus-west1zus-west2zus-west3zus-west4N)regionsselfr<S/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/hub/google/__init__.py__init__!s      !"#zGCPRegions.__init__returncCdd|jDS)zPReturns a list of GCP regions classified as tier 1 based on predefined criteria.cS g|] \}}|ddkr|qS)rrr<.0regioninfor<r<r= N z$GCPRegions.tier1..r9itemsr:r<r<r=tier1LzGCPRegions.tier1cCr@)zPReturns a list of GCP regions classified as tier 2 based on predefined criteria.cSrA)rr r<rBr<r<r=rFRrGz$GCPRegions.tier2..rHr:r<r<r=tier2PrKzGCPRegions.tier2rrDattemptsc Csd|d}g}t|D],}z t}tj|dd}t|d}|tdkr-||Wq tjy8Yq w|sJ|tdtdtdtdfSt|dkrUt |nd}|t ||t |t |fS) a Ping a specified GCP region and measure network latency statistics. Args: region (str): The GCP region identifier to ping (e.g., 'us-central1'). attempts (int): Number of ping attempts to make for calculating statistics. Returns: region (str): The GCP region identifier that was pinged. mean_latency (float): Mean latency in milliseconds, or infinity if all pings failed. min_latency (float): Minimum latency in milliseconds, or infinity if all pings failed. max_latency (float): Maximum latency in milliseconds, or infinity if all pings failed. std_dev (float): Standard deviation of latencies in milliseconds, or infinity if all pings failed. Examples: >>> region, mean, min_lat, max_lat, std = GCPRegions._ping_region("us-central1", attempts=3) >>> print(f"Region {region} has mean latency: {mean:.2f}ms") zhttps://z-docker.pkg.dev)timeoutiinfrr) rangetimerequestsheadfloatappendRequestExceptionlen statisticsstdevmeanminmax)rDrMurl latencies_ start_timelatencystd_devr<r<r= _ping_regionTs"    zGCPRegions._ping_regionFNtopverbosetiercs|rtdddkr dnddr!fddjDntj}tjjd d }t|fd d |}Wd n1sGwYt |dd d}|rtddddddddd|D]M\} } } } } j| \}}|d|}| t dkrt| dd|ddddqht| dd|dddd| dd| dd| dd| ddqhtd|dkrd nd!d"|d |D]0\} } } } } j| \}}|d|}t| d|d| dd| dd#| dd| dd$ q|d |S)%ax Determines the GCP regions with the lowest latency based on ping tests. Args: top (int): Number of top regions to return. verbose (bool): If True, prints detailed latency information for all tested regions. tier (int | None): Filter regions by tier (1 or 2). If None, all regions are tested. attempts (int): Number of ping attempts per region. Returns: (List[Tuple[str, float, float, float, float]]): List of tuples containing region information and latency statistics. Each tuple contains (region, mean_latency, std_dev, min_latency, max_latency). Examples: >>> regions = GCPRegions() >>> results = regions.lowest_latency(top=3, verbose=True, tier=1, attempts=2) >>> print(results[0][0]) # Print the name of the lowest latency region z&Testing GCP regions for latency (with  rretryrMz)...cs g|] \}}|dkr|qS)rr<)rCkv)rgr<r=rFrGz-GCPRegions.lowest_latency..2) max_workerscs |S)N)rd)r)rMr;r<r=s z+GCPRegions.lowest_latency..NcSs|dS)Nrr<)xr<r<r=ros)keyRegionz<25Locationz<35Tierz<5z Latency (ms)z, rPz Timeoutz.0fu ± z (z - )z Lowest latency regions:z ms (z))) printr9rIlistkeys concurrentfuturesThreadPoolExecutormapsortedrU)r;rerfrgrMregions_to_testexecutorresultssorted_resultsrDr[stdmin_max_citycountrylocationr<)rMr;rgr=lowest_latencyys*"*  "B6 zGCPRegions.lowest_latency)r)rFNr)__name__ __module__ __qualname____doc__r>rstrrJrL staticmethodintrrUrdboolrrr<r<r<r=r s,+(&r__main__Tr)rerfrgrM)concurrent.futuresr|rYrRtypingrrrrSrrr9rtop_3_latency_tier1r<r<r<r=s$