source4/scripting/python/samba/external.py

   1 #!/usr/bin/python
   2
   3 # Unix SMB/CIFS implementation.
   4 # Copyright (C) Andrew Tridgell 2010
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19
  20 import sys, os
  21
  22 def samba_find_external(directory):
  23     '''insert into out module search path the path to an
  24        external library'''
  25     for p in sys.path:
  26         dir = os.path.join(p, directory)
  27         if os.path.isdir(dir):
  28             sys.path.insert(0, dir)
  29             return
  30
  31     # finally try in the local directory, to handle in-tree testing
  32     dir = os.path.join("scripting/python", directory)
  33     if os.path.isdir(dir):
  34         sys.path.insert(0, dir)
  35         return
  36
  37     print "Failed to find external python library %s" % directory
  38     raise
  39
  40
  41 def samba_external_dns_resolver():
  42     '''try and import the dns.resolver library, and if it fails
  43     then use a local copy from the external directory'''
  44
  45     try:
  46         import dns.resolver as dns
  47     except:
  48         samba_find_external("samba_external/dnspython")
  49         import dns.resolver as dns
  50     return dns