v4l3n71n/web_scrap
1
0
Fork 0
Code Issues Pull Requests Projects Releases 10 Wiki Activity

Ajout print image ajoute

Browse Source
This commit is contained in:
Valentin CZERYBA 2023-04-04 22:14:10 +02:00
parent 665f1474f2
commit ba511bc6c4
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
insert_wordpress.py
View File

@ -13,23 +13,21 @@ class WPimport:
# Public method # Public method
def fromFile(self, file): def fromFile(self, files):
with open(file, 'r') as f: for file in files.split(","):
content = f.read() with open(file, 'r') as f:
soup = BeautifulSoup(content, 'html.parser') content = f.read()
articlebody = soup.find_all("div", class_="articlebody") soup = BeautifulSoup(content, 'html.parser')
if len(articlebody) > 0: articlebody = soup.find_all("div", class_="articlebody")
self._addOrUpdatePost(soup) if len(articlebody) > 0:
else: self._addOrUpdatePost(soup)
self._addOrUpdateFeaturedMedia(soup) else:
self._addOrUpdateFeaturedMedia(soup)
# Private method # Private method
## Add or update featured media ## Add or update featured media
## Get or update featured image
def _addOrUpdateFeaturedMedia(self, soup): def _addOrUpdateFeaturedMedia(self, soup):
item_div = soup.find_all("div", {"data-edittype": "post"}) item_div = soup.find_all("div", {"data-edittype": "post"})
for i in item_div: for i in item_div:
@ -92,6 +90,7 @@ class WPimport:
headers={ 'Content-Type': img_type,'Content-Disposition' : 'attachment; filename={0}'.format(img_name)} headers={ 'Content-Type': img_type,'Content-Disposition' : 'attachment; filename={0}'.format(img_name)}
r = requests.post("http://{0}/wp-json/wp/v2/media".format(self._wordpress), auth=self._basic, headers=headers, data=data) r = requests.post("http://{0}/wp-json/wp/v2/media".format(self._wordpress), auth=self._basic, headers=headers, data=data)
if r.status_code == 201: if r.status_code == 201:
print("Ajout d'image {0}".format(img_name))
res = r.json() res = r.json()
media["id"] = res["id"] media["id"] = res["id"]
media["rendered"] = res["guid"]["rendered"] media["rendered"] = res["guid"]["rendered"]