Add studio host

richardpenman · May 16, 2024 · 5cc9b77 · 5cc9b77
1 parent 370d26a
commit 5cc9b77
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/whois/parser.py b/whois/parser.py
@@ -1362,12 +1362,14 @@ def __init__(self, domain, text):
         WhoisRu.__init__(self, domain, text)
 
 
-class WhoisStudio(WhoisRu):
+class WhoisStudio(WhoisBz):
     """Whois parser for .studio domains"""
 
     def __init__(self, domain, text):
-        WhoisRu.__init__(self, domain, text)
-
+        if "Domain not found." in text:
+            raise PywhoisError(text)
+        else:
+            WhoisEntry.__init__(self, domain, text, self.regex)
 
 class WhoisStyle(WhoisRu):
     """Whois parser for .style domains"""

diff --git a/whois/whois.py b/whois/whois.py
@@ -99,6 +99,7 @@ class NICClient(object):
     SHOP_HOST = "whois.nic.shop"
     SG_HOST = "whois.sgnic.sg"
     STORE_HOST = "whois.centralnic.com"
+    STUDIO_HOST = "whois.nic.studio"
     DETI_HOST = "whois.nic.xn--d1acj3b"
     MOSKVA_HOST = "whois.registry.nic.xn--80adxhks"
     RF_HOST = "whois.registry.tcinet.ru"
@@ -343,7 +344,7 @@ def choose_server(self, domain):
         elif tld == "design":
             return NICClient.DESIGN_HOST
         elif tld == "studio":
-            return NICClient.RU_HOST
+            return NICClient.STUDIO_HOST
         elif tld == "style":
             return NICClient.RU_HOST
         elif tld == "su":