From c5fd4036a7ca695669e58f1b7590e2bf7132d981 Mon Sep 17 00:00:00 2001
From: Dhinak G <17605561+dhinakg@users.noreply.github.com>
Date: Fri, 18 Jun 2021 20:34:39 -0400
Subject: [PATCH] Add country code probing

---
 Resources/device_probe.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/Resources/device_probe.py b/Resources/device_probe.py
index ef6d8628e..6ae256faf 100644
--- a/Resources/device_probe.py
+++ b/Resources/device_probe.py
@@ -100,9 +100,12 @@ class GPU(PCIDevice):
 @dataclass
 class WirelessCard(PCIDevice):
     CLASS_CODE: ClassVar[int] = 0x028000  # 00800200 hexswapped
+    country_code: str = field(init=False)
     chipset: enum.Enum = field(init=False)
 
     def __post_init__(self):
+        system_profiler = plistlib.loads(subprocess.run("system_profiler -xml SPAirPortDataType", stdout=subprocess.PIPE).stdout)
+        self.country_code = system_profiler[0]["_items"][0]["spairport_airport_interfaces"][0]["spairport_wireless_country_code"]
         self.detect_chipset()
 
     def detect_chipset(self):