From f250637912f8432ca9a578d66300efad32a153e3 Mon Sep 17 00:00:00 2001 From: Valentin CZERYBA Date: Wed, 29 Mar 2023 22:31:35 +0200 Subject: [PATCH] add class + wordpress --- insert_wordpress.py | 101 +++++++++++++++++++++++--------------------- 1 file changed, 54 insertions(+), 47 deletions(-) diff --git a/insert_wordpress.py b/insert_wordpress.py index babec3a..c0c4d3d 100644 --- a/insert_wordpress.py +++ b/insert_wordpress.py @@ -5,27 +5,33 @@ from requests.auth import HTTPBasicAuth from getpass import getpass import requests, os, argparse, logging, re -def insertWordpress(file, basic): - tags = [] - month = {"janvier":"01", "février": "02", "mars": "03", "avril":"04", "mai": "05", "juin": "06", "juillet": "07", "août": "08", "septembre": "09", "octobre": "10", "novembre": "11", "décembre": "12"} - - liste = ["categories", "tags"] - elements = {} - element = {} - listelement = {} - - for i in liste: - page = requests.get("http://localhost:8080/wp-json/wp/v2/{0}".format(i)) - if page.status_code == 200: - elements[i] = page.json() - element[i] = [] - listelement[i] = [] - +class WPimport: - with open(file, 'r') as f: - contents = f.read() + def __init__(self, basic, wordpress): + self.basic = basic + self.wordpress = wordpress - soup = BeautifulSoup(contents, 'html.parser') + def fromFile(self, file): + with open(file, 'r') as f: + contents = f.read() + self.insertWordpress(contents) + + def insertWordpress(self, content): + tags = [] + month = {"janvier":"01", "février": "02", "mars": "03", "avril":"04", "mai": "05", "juin": "06", "juillet": "07", "août": "08", "septembre": "09", "octobre": "10", "novembre": "11", "décembre": "12"} + liste = ["categories", "tags"] + elements = {} + element = {} + listelement = {} + + for i in liste: + page = requests.get("http://{0}/wp-json/wp/v2/{1}".format(self.wordpress,i)) + if page.status_code == 200: + elements[i] = page.json() + element[i] = [] + listelement[i] = [] + + soup = BeautifulSoup(content, 'html.parser') articletitle = soup.find_all("h2", class_="articletitle") articlebody = soup.find_all("div", class_="articlebody") articledate = soup.find_all("span", class_="articledate") @@ -63,14 +69,11 @@ def insertWordpress(file, basic): array = listelement[i].append(k["id"]) if element_exist is False: data = {"name": j} - page = requests.post("http://localhost:8080/wp-json/wp/v2/{0}".format(i), auth=basic, data=data) + page = requests.post("http://{0}/wp-json/wp/v2/{1}".format(self.wordpress, i), auth=self.basic, data=data) if page.status_code == 201: result = page.json() listelement[i].append(result["id"]) - - - title = articletitle[0].text author = articleacreator[0].text.lower() body = articlebody[0].find_all("p") @@ -86,34 +89,36 @@ def insertWordpress(file, basic): data = {"title":title, "content":bodyhtml, "status":"publish", "date": "{0}-{1}-{2}T{3}:00".format(time[2],month[time[1]],time[0], hour), "tags": listelement["tags"], "categories": listelement["categories"]} params = {"search":author} - page = requests.get("http://localhost:8080/wp-json/wp/v2/users", auth=basic, params=params) + page = requests.get("http://{0}/wp-json/wp/v2/users".format(self.wordpress), auth=self.basic, params=params) if page.status_code == 200: result = page.json() data["author"] = result[0]["id"] - - params = {"search":title} - page = requests.get("http://localhost:8080/wp-json/wp/v2/posts", auth=basic, params=params) - page_exist = True - if page.status_code: - result = page.json() - if len(result) == 0: - page_exist = False - else: - print("La page {0} existe deja".format(title)) - - if page_exist == False: - page = requests.post("http://localhost:8080/wp-json/wp/v2/posts", auth=basic, data=data) - if page.status_code == 201: + params = {"search":title} + page = requests.get("http://{0}/wp-json/wp/v2/posts".format(self.wordpress), auth=self.basic, params=params) + page_exist = True + if page.status_code: result = page.json() - print("Article ajoute : {0}".format(result["title"]["raw"])) - print(comment_post) - for i in comment_post: - data = {"post": result["id"], "content": i["content"], "date": i["date"], "author_name": i["author"]} - page = requests.post("http://localhost:8080/wp-json/wp/v2/comments", auth=basic, data=data) - print(page.status_code) - if page.status_code == 201: - print("Commentaire ajoute pour {0}".format(result["title"]["raw"])) + if len(result) == 0: + page_exist = False + else: + print("La page {0} existe deja".format(title)) + + if page_exist == False: + page = requests.post("http://{0}/wp-json/wp/v2/posts".format(self.wordpress), auth=self.basic, data=data) + if page.status_code == 201: + result = page.json() + print("Article ajoute : {0}".format(result["title"]["raw"])) + print(comment_post) + for i in comment_post: + data = {"post": result["id"], "content": i["content"], "date": i["date"], "author_name": i["author"]} + page = requests.post("http://{0}/wp-json/wp/v2/comments".format(self.wordpress), auth=self.basic, data=data) + print(page.status_code) + if page.status_code == 201: + print("Commentaire ajoute pour {0}".format(result["title"]["raw"])) + + + @@ -121,6 +126,7 @@ if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument("--user", help="wordpress user", required=True) parser.add_argument("--file", help="HTML file", required=True) + parser.add_argument("--wordpress", help="URL Wordpress", required=True) args = parser.parse_args() password = getpass() if len(password) == 0: @@ -128,4 +134,5 @@ if __name__ == '__main__': exit(1) basic = HTTPBasicAuth(args.user, password) - insertWordpress(args.file, basic) \ No newline at end of file + importWp = WPimport(basic) + importWp.fromFile(args.file) \ No newline at end of file