From c8920096e04a91cedadde94d3b6e9138434dc2f1 Mon Sep 17 00:00:00 2001 From: calvin Date: Mon, 7 Mar 2005 00:09:22 +0000 Subject: [PATCH] make threadsafe geoip call git-svn-id: https://linkchecker.svn.sourceforge.net/svnroot/linkchecker/trunk/linkchecker@2354 e7d03fd6-7b0d-0410-9947-9c21f3af8025 --- linkcheck/checker/consumer.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/linkcheck/checker/consumer.py b/linkcheck/checker/consumer.py index eb5b6d96..c762fbaa 100644 --- a/linkcheck/checker/consumer.py +++ b/linkcheck/checker/consumer.py @@ -28,6 +28,7 @@ except ImportError: import linkcheck.threader import linkcheck.log import linkcheck.strformat +import linkcheck.checker.geoip from urlbase import stderr @@ -241,3 +242,15 @@ class Consumer (object): finally: self.lock.release() + def get_country_name (self, host): + """ + Return country code for host if found, else None. + """ + self.lock.acquire() + try: + gi = self.config["geoip"] + if gi: + return linkcheck.checker.geoip.get_country(gi, host) + return None + finally: + self.lock.release()