diff --git a/main.py b/main.py index 5dcf665..f910e72 100644 --- a/main.py +++ b/main.py @@ -15,6 +15,19 @@ def get_data(url): datas["text"] = (soup.find("body").text).replace("\\n", " ") return datas +def get_links(url): + links = [] + scraper = cfscrape.CloudflareScraper() + req = scraper.get(url) + soup = BeautifulSoup(req.text, 'lxml') + datas = soup.find_all("a") + for one_data in datas: + try: + links.append(one_data.get("href")) + except: + pass + return links + def save_data(datas): try: data = db.load("pages") diff --git a/server.py b/server.py index 274f5ef..72bc57a 100644 --- a/server.py +++ b/server.py @@ -27,6 +27,20 @@ def add_page_r(): text = "Главная страница | Добавить страницу

" + data["uid"] + " - "+ data["data"]["title"] + "" + "

" return html_page.format(title=title, text=text) +@app.route('/add_pagem', methods=('GET', 'POST')) +def add_pagem_r(): + urls = request.form['url'] + datas = main.get_data(urls) + data = main.save_data(datas) + html_page = open("templates/blanks.html", "r").read() + if data["code"] == 0: + title = "Страница присутсвует в каталоге" + text = "Главная страница | Добавить страницу

" + data["uid"] + " - "+ data["data"]["title"] + "" + "

" + else: + title = "Страница добавлена в каталог" + text = "Главная страница | Добавить страницу

" + data["uid"] + " - "+ data["data"]["title"] + "" + "

" + return html_page.format(title=title, text=text) + @app.route('/remove_page') def remove_page(): return render_template('remove_page.html') diff --git a/templates/add_page.html b/templates/add_page.html index 3ba0a6c..4946397 100644 --- a/templates/add_page.html +++ b/templates/add_page.html @@ -17,11 +17,17 @@
- Главная страница | Удалить страницу + Главная страница | Удалить страницу
+

Добавить дону страницу:

-
+ +

Или все ссылки с страницы:

+
+ + +