From 03378ebbbeea3e540807c5b2cb46efe1caf4813e Mon Sep 17 00:00:00 2001
From: Sven Slootweg <jamsoftgamedev@gmail.com>
Date: Fri, 21 Sep 2012 06:30:29 +0200
Subject: [PATCH] Add .ee syntax

---
 pythonwhois/__init__.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pythonwhois/__init__.py b/pythonwhois/__init__.py
index d41310f..d3a5998 100644
--- a/pythonwhois/__init__.py
+++ b/pythonwhois/__init__.py
@@ -92,10 +92,19 @@ grammar = {
 		'name_servers':		'nserver:\s*(?P<val>.+)',
 		'status':		'state:\s*(?P<val>.+)',
 		'updated_date':		'Last updated on (?P<val>.+) [a-z]{3}'
+	},
+	".*\.ee$": {
+		'domain_name':		'domain:\s*(?P<val>.+)',
+		'registrar':		'registrar:\s*(?P<val>.+)',
+		'creation_date':	'registered:\s*(?P<val>.+)',
+		'expiration_date':	'expire:\s*(?P<val>.+)',
+		'name_servers':		'nserver:\s*(?P<val>.+)',
+		'status':		'state:\s*(?P<val>.+)'
 	}
 }
 
 def whois(domain):
+	global grammar
 	ruleset = None
 	
 	for regex, rules in grammar.iteritems():
@@ -160,6 +169,7 @@ def whois(domain):
 	return data
 
 def parse_dates(dates):
+	global grammar
 	parsed_dates = []
 	
 	for date in dates: