0
|
1 """Utility methods for handling XML, reading HTTP, etc"""
|
|
2
|
|
3 import sys
|
|
4 import urllib
|
|
5 import urllib2
|
|
6 import logging
|
|
7
|
|
8
|
1
|
9 srvTxtUtilsVersion = "1.1"
|
0
|
10
|
|
11 def getInt(number, default=0):
|
|
12 """returns always an int (0 in case of problems)"""
|
|
13 try:
|
|
14 return int(number)
|
|
15 except:
|
|
16 return int(default)
|
|
17
|
1
|
18 def getAt(array, idx, default=None):
|
|
19 """returns element idx from array or default (in case of problems)"""
|
|
20 try:
|
|
21 return array[idx]
|
|
22 except:
|
|
23 return default
|
|
24
|
0
|
25 def getText(node):
|
|
26 """returns all text content of a node and its subnodes"""
|
|
27 if node is None:
|
|
28 return ""
|
|
29 # ElementTree:
|
|
30 text = node.text or ""
|
|
31 for e in node:
|
|
32 text += gettext(e)
|
|
33 if e.tail:
|
|
34 text += e.tail
|
|
35
|
|
36 # 4Suite:
|
|
37 #nodelist=node.childNodes
|
|
38 #text = ""
|
|
39 #for n in nodelist:
|
|
40 # if n.nodeType == node.TEXT_NODE:
|
|
41 # text = text + n.data
|
|
42
|
|
43 return text
|
|
44
|
|
45
|
|
46
|
|
47 def getHttpData(url, data=None, num_tries=3, timeout=10):
|
|
48 """returns result from url+data HTTP request"""
|
|
49 # we do GET (by appending data to url)
|
|
50 if isinstance(data, str) or isinstance(data, unicode):
|
|
51 # if data is string then append
|
|
52 url = "%s?%s"%(url,data)
|
|
53 elif isinstance(data, dict) or isinstance(data, list) or isinstance(data, tuple):
|
|
54 # urlencode
|
|
55 url = "%s?%s"%(url,urllib.urlencode(data))
|
|
56
|
|
57 response = None
|
|
58 errmsg = None
|
|
59 for cnt in range(num_tries):
|
|
60 try:
|
|
61 logging.debug("getHttpData(#%s %ss) url=%s"%(cnt+1,timeout,url))
|
|
62 if sys.version_info < (2, 6):
|
|
63 # set timeout on socket -- ugly :-(
|
|
64 import socket
|
|
65 socket.setdefaulttimeout(float(timeout))
|
|
66 response = urllib2.urlopen(url)
|
|
67 else:
|
|
68 # timeout as parameter
|
|
69 response = urllib2.urlopen(url,timeout=float(timeout))
|
|
70 # check result?
|
|
71 break
|
|
72 except urllib2.HTTPError, e:
|
|
73 logging.error("getHttpData: HTTP error(%s): %s"%(e.code,e))
|
|
74 errmsg = str(e)
|
|
75 # stop trying
|
|
76 break
|
|
77 except urllib2.URLError, e:
|
|
78 logging.error("getHttpData: URLLIB error(%s): %s"%(e.reason,e))
|
|
79 errmsg = str(e)
|
|
80 # stop trying
|
|
81 #break
|
|
82
|
|
83 if response is not None:
|
|
84 data = response.read()
|
|
85 response.close()
|
|
86 return data
|
|
87
|
|
88 raise IOError("ERROR fetching HTTP data from %s: %s"%(url,errmsg))
|
|
89 #return None
|
|
90
|