File: pages.py

package info (click to toggle)
weboob 0.c-4.1
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 5,008 kB
  • sloc: python: 28,678; perl: 244; sh: 198; makefile: 111; sql: 17
file content (85 lines) | stat: -rw-r--r-- 3,125 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# -*- coding: utf-8 -*-

# Copyright(C) 2011-2012 Laurent Bachelier
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.


from weboob.tools.browser import BasePage, BrokenPageError

__all__ = ['PastePage', 'PostPage']


class BasePastebinPage(BasePage):
    def is_logged(self):
        header = self.parser.select(self.document.getroot(),
                'id("header_bottom")/ul[@class="top_menu"]', 1, 'xpath')
        for link in header.xpath('//ul/li/a'):
            if link.text == 'logout':
                return True
            if link.text == 'login':
                return False


class LoginPage(BasePastebinPage):
    def login(self, username, password):
        self.browser.select_form(nr=1)
        self.browser['user_name'] = username.encode(self.browser.ENCODING)
        self.browser['user_password'] = password.encode(self.browser.ENCODING)
        self.browser.submit()


class PastePage(BasePastebinPage):
    def fill_paste(self, paste):
        header = self.parser.select(self.document.getroot(),
                'id("content_left")//div[@class="paste_box_info"]', 1, 'xpath')
        paste.title = self.parser.select(header,
                '//div[@class="paste_box_line1"]//h1', 1, 'xpath').text
        paste.contents = self.parser.select(self.document.getroot(),
                '//textarea[@id="paste_code"]', 1, 'xpath').text
        visibility_text = self.parser.select(header,
                '//div[@class="paste_box_line1"]//img', 1, 'xpath').attrib['title']
        if visibility_text.startswith('Public'):
            paste.public = True
        elif visibility_text.startswith('Private'):
            paste.public = False
        else:
            raise BrokenPageError('Unable to get the paste visibility')
        return paste

    def get_id(self):
        """
        Find out the ID from the URL
        """
        return self.group_dict['id']


class PostPage(BasePastebinPage):
    def post(self, paste, expiration=None):
        self.browser.select_form(name='myform')
        self.browser['paste_code'] = paste.contents.encode(self.browser.ENCODING)
        self.browser['paste_name'] = paste.title.encode(self.browser.ENCODING)
        if paste.public is True:
            self.browser['paste_private'] = ['0']
        elif paste.public is False:
            self.browser['paste_private'] = ['1']
        if expiration:
            self.browser['paste_expire_date'] = [expiration]
        self.browser.submit()


class UserPage(BasePastebinPage):
    pass