๐Ÿ“ฆ wasuaje / Pypinterest

๐Ÿ“„ pyterest.py ยท 212 lines
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212#!/usr/bin/env python
# -*- coding: utf8 -*-
import time
import urllib
import urllib2
import re
import cookielib
import cStringIO
import gzip
import sys
import json

class Pinterest(object):

    def __init__(self,cookie=None):
        self.cookieJar           = cookie
        self.csrfmiddlewaretoken = None
        self.http_timeout        = 15
        self.boards              = {}
        
    def getCookies(self):
        return self.cookieJar
        
    def login(self,login,password):
        url = 'https://pinterest.com/login/'
        try:
            res,headers,cookies = self.request(url)
        except Exception as e:
            raise NotLogged(e)
        
        post_data = urllib.urlencode({
            'source_url':'/login/',
            'data':json.dumps({"options":{"username_or_email":login,"password":password},"context":{"app_version":"62fcc23","https_exp":False}}),
            'module_path':'App()>LoginPage()>Login()>Button(class_name=primary, text=Log in, type=submit, size=large)'
        }) 
        res,headers,cookies = self.request('http://www.pinterest.com/resource/UserSessionResource/create/',post_data,referrer='https://pinterest.com/login/',ajax=True)        
        if login in res:
            return True
        else:
            raise NotLogged('Not authorized. Cant find "window.userIsAuthenticated = true" in response')

    def search(self,query):
    
        query = urllib.quote(query)
    
        #"bookmarks": ["b28yNXxmMGVmZDA5YjA5Yjk5NDIxZjgzODZjNDRkZWEzNTRhZmIyZDRjZjE3ZWY4YmRmMWE4NmVhZGQwNDg1NTNmOTAw"]
        #url = 'http://www.pinterest.com/search/pins/?q=%s' % query
        
        #res,headers,cookies = self.request(url, referrer='https://pinterest.com/')
        #bookmark = re.findall(r'"bookmarks": \["([0-9a-z]+)"\]',res, re.I | re.M | re.U | re.S)[0]
        bookmark = ''

        url =  'http://www.pinterest.com/resource/SearchResource/get/?source_url=%2Fsearch%2Fpins%2F%3Fq%3Dart%26rs%3Dac%26len%3D1&data=%7B%22options%22%3A%7B%22show_scope_selector%22%3Anull%2C%22scope%22%3A%22pins%22%2C%22constraint_string%22%3Anull%2C%22bookmarks%22%3A%5B%22'+bookmark+'%22%5D%2C%22query%22%3A%22'+query+'%22%7D%2C%22context%22%3A%7B%22app_version%22%3A%22da919e8%22%2C%22https_exp%22%3Afalse%7D%2C%22module%22%3A%7B%22name%22%3A%22GridItems%22%2C%22options%22%3A%7B%22scrollable%22%3Atrue%2C%22show_grid_footer%22%3Atrue%2C%22centered%22%3Atrue%2C%22reflow_all%22%3Atrue%2C%22virtualize%22%3Atrue%2C%22item_options%22%3A%7B%22show_pinner%22%3Atrue%2C%22show_pinned_from%22%3Afalse%2C%22show_board%22%3Atrue%7D%2C%22layout%22%3A%22variable_height%22%2C%22track_item_impressions%22%3Atrue%7D%7D%2C%22append%22%3Atrue%2C%22error_strategy%22%3A1%7D&module_path=App()%3EHeader()%3Eui.SearchForm()%3Eui.TypeaheadField(enable_recent_queries%3Dtrue%2C+name%3Dq%2C+view_type%3Dsearch%2C+class_name%3DinHeader%2C+prefetch_on_focus%3Dtrue%2C+value%3D%22%22%2C+populate_on_result_highlight%3Dtrue%2C+search_delay%3D0%2C+search_on_focus%3Dtrue%2C+placeholder%3DSearch%2C+tags%3Dautocomplete)&_='+str(int(time.time())*10*10*10)
                                                                                                                                      

        res,headers,cookies = self.request(url, referrer='https://pinterest.com/search/pins/?q=%s' % query, ajax=True)
        
        data = json.loads(res)
        posts = data['module']['tree']['children']
        res = []
        for p in posts:
            desc = ''
            for i in p['children']:
                if i['id'] == 'sendPinButton':
                    desc = i['options']['module']['options']['object_description']
                    break
            res.append({
                'id': p['options']['pin_id']
                ,'img': p['data']['images']['orig']['url']
                ,'link': p['data']['link']
                ,'desc': desc
            })
            
        return res
        
    def getBoards(self):
        url       = 'http://pinterest.com/pin/create/bookmarklet/'
        res,headers,cookies = self.request(url,referrer='https://pinterest.com/')
        
        res = re.findall(r'<li(?:.*?)data-id="([^"]+)"(?:.*?)</div>([^<]+)</li>',res, re.I | re.M | re.U | re.S)
        boards = {}
        for idb,name in res:
            boards[unicode(name,'utf-8').lower().strip()] = idb
        self.boards = boards
        return boards
    
    def createPin(self,board='',title='',media='',posturl='',tags=[]):
        url       = 'http://pinterest.com/pin/create/bookmarklet/?media=%s&url=%s&description=%s' % (media,posturl,title)
        post_data = urllib.urlencode({
            'source_url':'/pin/create/bookmarklet/?media=%s&url=%s&description=%s' % (media,posturl,title),
            'data':'{"options":{"board_id":"%s","description":"%s","link":"%s","share_facebook":false,"image_url":"%s","method":"bookmarklet","is_video":null},"context":{"app_version":"62fcc23","https_exp":false}}' % (board,title,posturl,media),
            'module_path':'App()>PinBookmarklet()>PinCreate()>PinForm()>Button(class_name=repinSmall pinIt, text=Pin it, disabled=false, has_icon=true, show_text=false, type=submit, color=primary)'
        }) 
        try:
            res,header,query = self.request('http://www.pinterest.com/resource/PinResource/create/', post_data,referrer=url,ajax=True)
        except Exception as e:
            raise CantCreatePin(e)
        else:
            if 'PinResource' in res:
                return True
            raise CantCreatePin('Cant create pin. Cant find PinResource in response')
            
    def request(self,url,post_data=None,referrer='http://google.com/',ajax=False):
        """Donwload url with urllib2.
        
           Return downloaded data
        """
        handlers = []
        
        urllib2.HTTPRedirectHandler.max_redirections = 10
        
        if not self.cookieJar:
            self.cookieJar = cookielib.CookieJar()
        
        cookie_handler = urllib2.HTTPCookieProcessor(self.cookieJar)
        handlers.append(cookie_handler)
            
        opener = urllib2.build_opener(*handlers)

        opener.addheaders = [
            ('User-Agent', 'Mozilla/5.0 (X11; Linux i686) AppleWebKit/537.1 \
                      (KHTML, like Gecko) Chrome/21.0.1180.89 Safari/537.1'),
            ('Accept', 'image/png,image/*;q=0.8,*/*;q=0.5'),
            ('Accept-Language', 'en-us,en;q=0.5'),
            ('Accept-Encoding', 'gzip,deflate'),
            ('Accept-Charset', 'ISO-8859-1,utf-8;q=0.7,*;q=0.7'),
            ('Keep-Alive', '3600'),
            ('Host','www.pinterest.com'),
            ('Origin','http://www.pinterest.com'),
            ('Connection', 'keep-alive'),
            ('Referer', referrer),
            ('X-NEW-APP','1')
        ]
        if ajax:
            opener.addheaders.append(('X-Requested-With','XMLHttpRequest'))
        if self.csrfmiddlewaretoken:
            opener.addheaders.append(('X-CSRFToken',self.csrfmiddlewaretoken))
        error_happen = False
        html = ''
        try:
            req = urllib2.Request(url, post_data)
            r = opener.open(req,timeout=self.http_timeout)
            html = r.read() 
        except DownloadTimeoutException,e:
            sys.exc_clear()
            error_happen = e
        except Exception,e:
            sys.exc_clear()
            error_happen = e
            
        if error_happen:
            return error_happen,{},{}

        headers = r.info()
        # If we get gzipped data the unzip it
        if ('Content-Encoding' in headers.keys() and headers['Content-Encoding']=='gzip') or \
           ('content-encoding' in headers.keys() and headers['content-encoding']=='gzip'):
            data = cStringIO.StringIO(html)
            gzipper = gzip.GzipFile(fileobj=data)
            # Some servers may return gzip header, but not zip data.
            try:
                html_unzipped = gzipper.read()
            except:
                sys.exc_clear()    
            else:
                html = html_unzipped

        cookies = {cookie.name:cookie.value for cookie in self.cookieJar}
        self.csrfmiddlewaretoken = cookies['csrftoken']
           
        return html,headers,cookies
         
            
class DownloadTimeoutException(Exception):
    pass
    
class NotLogged(Exception):
    pass
    
class CantCreatePin(Exception):
    pass
    
    
if __name__ == "__main__":
    import traceback
    try:
        p = Pinterest()
        p.login('sdfdsfsd@gmail.com','fefsd')
        boards = p.getBoards()
        for k,v in boards.iteritems():
        
            search = p.search("car boat")[:3]
            for s in search:
                res = p.createPin(board=v,title='30 Art Works',
                    media=s['img'],
                    posturl='http://30artworks.blogspot.com/'
                )

                print 'pin created: %s' % res
    except:
        print traceback.format_exc()
    finally:
        time.sleep(30)