133 lines
5.3 KiB
Python
133 lines
5.3 KiB
Python
#!/usr/bin/python3
|
|
from bs4 import BeautifulSoup
|
|
from urllib.parse import urlparse
|
|
import requests, os, argparse, logging
|
|
|
|
def mkdirPath(path_dir, logger):
|
|
if not os.path.exists(path_dir):
|
|
makedir = []
|
|
pathh = path_dir.split("/")
|
|
for i in pathh:
|
|
makedir.append(i)
|
|
repath = "/".join(makedir)
|
|
if not os.path.exists(repath):
|
|
logger.debug("Dossier crée : {0}".format(repath))
|
|
os.mkdir(repath)
|
|
|
|
|
|
def getScriptCss(url, js, css, logger):
|
|
page = requests.get(url)
|
|
page_url = []
|
|
if page.status_code == 200:
|
|
soup = BeautifulSoup(page.text, 'html.parser')
|
|
if js is True:
|
|
script = soup.find_all("script")
|
|
for anchor in script:
|
|
src = anchor.get("src", "/")
|
|
if src != "/":
|
|
u = urlparse(url)
|
|
o = urlparse(src)
|
|
if o.netloc == "":
|
|
o = o._replace(netloc=u.netloc)
|
|
o = o._replace(scheme=u.scheme)
|
|
page_url.append(o.geturl())
|
|
|
|
return page_url
|
|
|
|
def getUrlPage(url, logger):
|
|
page = requests.get(url)
|
|
page_url = []
|
|
if page.status_code == 200:
|
|
soup = BeautifulSoup(page.text, 'html.parser')
|
|
ul = soup.find_all("ul", id="listsmooth")
|
|
for anchor in ul[0].find_all("a"):
|
|
href = anchor.get('href', '/')
|
|
if href != "#":
|
|
page_url.append(href)
|
|
|
|
webpage = []
|
|
for i in page_url:
|
|
page = requests.get(i)
|
|
if page.status_code == 200:
|
|
logger.info("page : {0}".format(i))
|
|
if i not in webpage:
|
|
webpage.append(i)
|
|
soup = BeautifulSoup(page.text, 'html.parser')
|
|
class_div = pagingfirstline = soup.find_all("div", class_="pagingfirstline")
|
|
if len(class_div) > 0:
|
|
pagingfirstline = class_div[0].find_all("a")
|
|
if len(pagingfirstline) > 1:
|
|
lastpage = pagingfirstline[len(pagingfirstline)-1].get("href", "/")
|
|
element_lastpage = lastpage.split("/")[len(lastpage.split("/"))-1]
|
|
number_page = element_lastpage.split("-")[0].split("p")[1]
|
|
number_lastpage = int(number_page) / 10
|
|
for j in range(1,int(number_lastpage)):
|
|
paging = j * 10
|
|
categorie = urlparse(i).path.split("/")
|
|
url_paging = "{0}/archives/p{1}-10.html".format(url, paging)
|
|
if len(categorie) > 2:
|
|
url_paging = "{0}/archives/{1}/p{2}-10.html".format(url, categorie[2], paging)
|
|
logger.info(url_paging)
|
|
if url_paging not in webpage:
|
|
webpage.append(url_paging)
|
|
page = requests.get(url_paging)
|
|
if page.status_code == 200:
|
|
soup = BeautifulSoup(page.text, 'html.parser')
|
|
h2 = soup.find_all("h2")
|
|
for title in h2:
|
|
href = title.find_all("a")[0].get("href", "/")
|
|
if href not in webpage:
|
|
o = urlparse(href)
|
|
o = o._replace(scheme="https").geturl()
|
|
webpage.append(o)
|
|
return webpage
|
|
|
|
|
|
def downloadPage(webpage, backup_dir, logger):
|
|
|
|
for i in range(0, len(webpage)):
|
|
o = urlparse(webpage[i])
|
|
path_web = o.path.split("/")
|
|
filePageWeb = path_web[len(path_web)-1]
|
|
path_web.pop(len(path_web)-1)
|
|
dir_page_web = "/".join(path_web)
|
|
mkdirPath("{0}/{1}".format(backup_dir, dir_page_web), logger)
|
|
r = requests.get(webpage[i])
|
|
if r.status_code == 200:
|
|
fileDownload = "{0}/index.html".format(backup_dir)
|
|
if len(dir_page_web) > 0 and len(filePageWeb) > 0:
|
|
fileDownload = "{0}{1}/{2}".format(backup_dir, dir_page_web, filePageWeb)
|
|
logger.info("{0}/{1} : {2}".format(i, len(webpage), fileDownload))
|
|
open(fileDownload, "wb").write(r.content)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--url", help="canblog URL to be scraping", required=True)
|
|
parser.add_argument("--dir",
|
|
default="backup",
|
|
help="backup file path")
|
|
parser.add_argument("--debug", help="Verbosity", action="store_true")
|
|
args = parser.parse_args()
|
|
logger = logging.getLogger('web_scrap')
|
|
ch = logging.StreamHandler()
|
|
|
|
if args.debug is True:
|
|
logger.setLevel(logging.DEBUG)
|
|
ch.setLevel(logging.DEBUG)
|
|
else:
|
|
logger.setLevel(logging.INFO)
|
|
ch.setLevel(logging.INFO)
|
|
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
ch.setFormatter(formatter)
|
|
logger.addHandler(ch)
|
|
|
|
o = urlparse(args.url)
|
|
o = o._replace(scheme="https")
|
|
url = o.geturl().replace(":///", "://")
|
|
script = getScriptCss(url, True, False, logger)
|
|
logger.info(script)
|
|
#webpage = getUrlPage(url, logger)
|
|
|
|
downloadPage(script, "{0}/{1}".format(args.dir, "dists/js"), logger) |