mirror of
https://github.com/balkian/gists.git
synced 2024-11-21 17:22:29 +00:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import mechanize, cookielib
|
|
from BeautifulSoup import BeautifulSoup
|
|
import re, os
|
|
from zipfile import ZipFile
|
|
|
|
def get_image(tag, folder):
|
|
if not os.path.isdir(folder):
|
|
os.makedirs(folder)
|
|
print image
|
|
filename = os.path.join(folder+'/'+url.split('/')[-1])
|
|
data = br.open(url).read()
|
|
br.back()
|
|
save = open(filename, 'wb')
|
|
save.write(data)
|
|
save.close()
|
|
return filename
|
|
|
|
|
|
br=mechanize.Browser()
|
|
cj = cookielib.LWPCookieJar()
|
|
br.set_cookiejar(cj)
|
|
br.set_handle_equiv(True)
|
|
br.set_handle_gzip(True)
|
|
br.set_handle_redirect(True)
|
|
br.set_handle_referer(True)
|
|
br.set_handle_robots(False)
|
|
br.addheaders = [('User-agent', 'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.1) Gecko/2008071615 Fedora/3.0.1-1.fc9 Firefox/3.0.1')]
|
|
|
|
tozip = [];
|
|
|
|
#Individuales
|
|
r = br.open('http://www.etsit.upm.es/fotospromo85.html')
|
|
br.select_form(nr=0)
|
|
br.form['user']='entregadiplomas'
|
|
br.form['pass']='entrega2013'
|
|
r = br.submit()
|
|
r_tags = BeautifulSoup(r)
|
|
link = r_tags(text=re.compile(r'individual'))[0].parent['href']
|
|
galery = BeautifulSoup(br.open(link))
|
|
images_tags = galery(href=re.compile(r'pics/.{10}.jpg'))
|
|
folder = 'individuales'
|
|
for image in images_tags:
|
|
url = image['href']
|
|
tozip.append(get_image(url, folder))
|
|
|
|
# Ahora las de grupo
|
|
link=r_tags(text=re.compile(r'grupo'))[0].parent['href']
|
|
galery = BeautifulSoup(br.open(link))
|
|
images_tags = galery(href=re.compile(r'pics/.{10}.jpg'))
|
|
folder = 'grupo'
|
|
for image in images_tags:
|
|
url = image['href']
|
|
tozip.append(get_image(url, folder))
|
|
|
|
with ZipFile('fotos-85.zip', 'w') as myzip:
|
|
for i in tozip:
|
|
myzip.write(i) |