Check for page load in get_page_height function
Manuel QuiƱones authored 12 years ago
Gonzalo Odiard committed 12 years ago
10 | 10 | Gets height (in pixels) of loaded (X)HTML page. |
11 | 11 | This is done via javascript at the moment |
12 | 12 | ''' |
13 | #TODO: Need to check status of page load | |
14 | 13 | js = 'oldtitle=document.title;' + \ |
14 | 'if (document.body == null) {' + \ | |
15 | 'document.title = 0} else {' + \ | |
15 | 16 | 'document.title=Math.max(document.body.scrollHeight, ' + \ |
16 | 17 | 'document.body.offsetHeight,document.documentElement.clientHeight,' + \ |
17 | 18 | 'document.documentElement.scrollHeight, ' + \ |
18 | 'document.documentElement.offsetHeight);' | |
19 | 'document.documentElement.offsetHeight)};' | |
19 | 20 | self.execute_script(js) |
20 | 21 | ret = self.get_main_frame().get_title() |
21 | 22 | js = 'document.title=oldtitle;' |
22 | 23 | self.execute_script(js) |
23 | if ret is None: | |
24 | try: | |
25 | return int(ret) | |
26 | except ValueError: | |
24 | 27 | return 0 |
25 | return int(ret) | |
26 | 28 | |
27 | 29 | def add_bottom_padding(self, incr): |
28 | 30 | ''' |